All files / roles list.js

92.68% Statements 38/41
83.33% Branches 20/24
100% Functions 5/5
92.1% Lines 35/38

Press n or j to go to the next uncovered block, b, p or k for the previous block.

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 1481x           1x 1x   1x 1x   1x 13x 13x 13x   13x 2x                     11x             11x 2x                     9x   9x 1x                     8x     8x   8x                     8x                             8x   7x     7x 7x 7x 1x       1x         1x 1x     7x         7x       7x                         1x 1x                    
const { DynamoDBClient } = require('@aws-sdk/client-dynamodb')
const {
  DynamoDBDocumentClient,
  QueryCommand,
  GetCommand,
  BatchGetCommand
} = require('@aws-sdk/lib-dynamodb')
const { permissionChecker } = require('../utils/PermissionChecker')
 
const client = new DynamoDBClient({})
const docClient = DynamoDBDocumentClient.from(client)
 
exports.handler = async (event) => {
  try {
    const flowId = event.pathParameters?.flowId
    const userId = event.requestContext?.authorizer?.claims?.sub
 
    if (!userId || !flowId) {
      return {
        statusCode: 401,
        headers: {
          'Content-Type': 'application/json',
          'Access-Control-Allow-Origin': '*'
        },
        body: JSON.stringify({ error: 'Unauthorized' })
      }
    }
 
    // Check if user is member of this flow
    const membershipCheck = await docClient.send(
      new GetCommand({
        TableName: process.env.FLOW_MEMBERSHIPS_TABLE_NAME,
        Key: { userId, flowId }
      })
    )
 
    if (!membershipCheck.Item || membershipCheck.Item.status !== 'active') {
      return {
        statusCode: 403,
        headers: {
          'Content-Type': 'application/json',
          'Access-Control-Allow-Origin': '*'
        },
        body: JSON.stringify({ error: 'Access denied - not a member of this flow' })
      }
    }
 
    // Check if user has permission to list roles
    const permissionCheck = await permissionChecker.hasPermission(userId, flowId, 'roles:list')
 
    if (!permissionCheck.hasPermission) {
      return {
        statusCode: 403,
        headers: {
          'Content-Type': 'application/json',
          'Access-Control-Allow-Origin': '*'
        },
        body: JSON.stringify({ error: 'Forbidden - Insufficient permissions' })
      }
    }
 
    // Pagination support
    const limit = event.queryStringParameters?.limit
      ? parseInt(event.queryStringParameters.limit)
      : 30
    const nextToken = event.queryStringParameters?.nextToken
 
    const queryParams = {
      TableName: process.env.ROLES_TABLE_NAME,
      KeyConditionExpression: 'PK = :pk AND begins_with(SK, :sk)',
      ExpressionAttributeValues: {
        ':pk': `FLOW#${flowId}`,
        ':sk': 'ROLE#'
      },
      Limit: limit
    }
 
    // Add pagination token if provided
    Iif (nextToken) {
      try {
        queryParams.ExclusiveStartKey = JSON.parse(Buffer.from(nextToken, 'base64').toString())
      } catch {
        return {
          statusCode: 400,
          headers: {
            'Content-Type': 'application/json',
            'Access-Control-Allow-Origin': '*'
          },
          body: JSON.stringify({ error: 'Invalid nextToken' })
        }
      }
    }
 
    const result = await docClient.send(new QueryCommand(queryParams))
 
    const roles = result.Items || []
 
    // Enrich roles with creator names
    const creatorIds = [...new Set(roles.map((r) => r.createdBy).filter(Boolean))]
    let userNameMap = {}
    if (creatorIds.length > 0) {
      const usersResult = await docClient.send(
        new BatchGetCommand({
          RequestItems: {
            [process.env.USERS_TABLE_NAME]: {
              Keys: creatorIds.map((id) => ({ id }))
            }
          }
        })
      )
      const users = usersResult.Responses?.[process.env.USERS_TABLE_NAME] || []
      userNameMap = Object.fromEntries(users.map((u) => [u.id, u.name]))
    }
 
    const enrichedRoles = roles.map((role) => ({
      ...role,
      createdByName: userNameMap[role.createdBy] || null
    }))
 
    const responseNextToken = result.LastEvaluatedKey
      ? Buffer.from(JSON.stringify(result.LastEvaluatedKey)).toString('base64')
      : null
 
    return {
      statusCode: 200,
      headers: {
        'Content-Type': 'application/json',
        'Access-Control-Allow-Origin': '*'
      },
      body: JSON.stringify({
        items: enrichedRoles,
        nextToken: responseNextToken,
        hasMore: !!responseNextToken
      })
    }
  } catch (error) {
    console.error('Error listing roles:', error)
    return {
      statusCode: 500,
      headers: {
        'Content-Type': 'application/json',
        'Access-Control-Allow-Origin': '*'
      },
      body: JSON.stringify({ error: 'Failed to list roles' })
    }
  }
}