handler.js

const AWS = require('aws-sdk');
const { v4: uuidv4 } = require('uuid');

let options = {};
if (process.env.IS_OFFLINE) {
  options = {
    region: 'localhost',
    endpoint: 'http://localhost:8000',
  };
}

const dynamoDb = new AWS.DynamoDB.DocumentClient({
    region: 'localhost',
    endpoint: 'http://localhost:8000',
    accessKeyId: 'fakeMyKeyId',
    secretAccessKey: 'fakeSecretAccessKey',
  });

module.exports.createOrganization = async (event) => {
  try {
    const data = JSON.parse(event.body);

    if (!data.name || typeof data.name !== 'string') {
      return {
        statusCode: 400,
        headers: { 'Content-Type': 'application/json' },
        body: JSON.stringify({ message: 'Назва організації обов\'язкова' })
      };
    }

    const existingOrgParams = {
      TableName: process.env.ORGANIZATIONS_TABLE,
      IndexName: 'nameIndex',
      KeyConditionExpression: '#name = :name',
      ExpressionAttributeNames: {
        '#name': 'name'
      },
      ExpressionAttributeValues: {
        ':name': data.name
      }
    };

    const existingOrg = await dynamoDb.query(existingOrgParams).promise();

    if (existingOrg.Items && existingOrg.Items.length > 0) {
      return {
        statusCode: 409,
        headers: { 'Content-Type': 'application/json' },
        body: JSON.stringify({ message: 'Організація з такою назвою вже існує' })
      };
    }

    const orgId = uuidv4();
    const timestamp = new Date().getTime();

    const params = {
      TableName: process.env.ORGANIZATIONS_TABLE,
      Item: {
        orgId,
        name: data.name,
        description: data.description || '',
        createdAt: timestamp,
        updatedAt: timestamp
      }
    };

    await dynamoDb.put(params).promise();

    return {
      statusCode: 201,
      headers: { 'Content-Type': 'application/json' },
      body: JSON.stringify(params.Item)
    };
  } catch (error) {
    console.error('Помилка створення організації:', error);
    return {
      statusCode: 500,
      headers: { 'Content-Type': 'application/json' },
      body: JSON.stringify({ message: 'Не вдалося створити організацію', error: error.message })
    };
  }
};

module.exports.updateOrganization = async (event) => {
  try {
    const data = JSON.parse(event.body);

    if (!data.orgId || typeof data.orgId !== 'string') {
      return {
        statusCode: 400,
        headers: { 'Content-Type': 'application/json' },
        body: JSON.stringify({ message: 'ID організації обов\'язковий' })
      };
    }

    const orgParams = {
      TableName: process.env.ORGANIZATIONS_TABLE,
      Key: {
        orgId: data.orgId
      }
    };

    const org = await dynamoDb.get(orgParams).promise();

    if (!org.Item) {
      return {
        statusCode: 404,
        headers: { 'Content-Type': 'application/json' },
        body: JSON.stringify({ message: 'Організацію не знайдено' })
      };
    }

    if (data.name && data.name !== org.Item.name) {
      const existingOrgParams = {
        TableName: process.env.ORGANIZATIONS_TABLE,
        IndexName: 'nameIndex',
        KeyConditionExpression: '#name = :name',
        ExpressionAttributeNames: {
          '#name': 'name'
        },
        ExpressionAttributeValues: {
          ':name': data.name
        }
      };

      const existingOrg = await dynamoDb.query(existingOrgParams).promise();

      if (existingOrg.Items && existingOrg.Items.length > 0) {
        return {
          statusCode: 409,
          headers: { 'Content-Type': 'application/json' },
          body: JSON.stringify({ message: 'Організація з такою назвою вже існує' })
        };
      }
    }

    const timestamp = new Date().getTime();

    const updateParams = {
      TableName: process.env.ORGANIZATIONS_TABLE,
      Key: {
        orgId: data.orgId
      },
      ExpressionAttributeNames: {
        '#name': 'name',
        '#desc': 'description',
        '#updatedAt': 'updatedAt'
      },
      ExpressionAttributeValues: {
        ':name': data.name || org.Item.name,
        ':description': data.description || org.Item.description,
        ':updatedAt': timestamp
      },
      UpdateExpression: 'SET #name = :name, #desc = :description, #updatedAt = :updatedAt',
      ReturnValues: 'ALL_NEW'
    };

    const result = await dynamoDb.update(updateParams).promise();

    return {
      statusCode: 200,
      headers: { 'Content-Type': 'application/json' },
      body: JSON.stringify(result.Attributes)
    };
  } catch (error) {
    console.error('Помилка оновлення організації:', error);
    return {
      statusCode: 500,
      headers: { 'Content-Type': 'application/json' },
      body: JSON.stringify({ message: 'Не вдалося оновити організацію', error: error.message })
    };
  }
};

module.exports.createUser = async (event) => {
  try {
    const data = JSON.parse(event.body);
    const orgId = event.pathParameters.orgId;

    if (!data.name || typeof data.name !== 'string') {
      return {
        statusCode: 400,
        headers: { 'Content-Type': 'application/json' },
        body: JSON.stringify({ message: 'Ім\'я користувача обов\'язкове' })
      };
    }

    if (!data.email || typeof data.email !== 'string') {
      return {
        statusCode: 400,
        headers: { 'Content-Type': 'application/json' },
        body: JSON.stringify({ message: 'Email користувача обов\'язковий' })
      };
    }

    const orgParams = {
      TableName: process.env.ORGANIZATIONS_TABLE,
      Key: {
        orgId
      }
    };

    const org = await dynamoDb.get(orgParams).promise();

    if (!org.Item) {
      return {
        statusCode: 404,
        headers: { 'Content-Type': 'application/json' },
        body: JSON.stringify({ message: 'Організацію не знайдено' })
      };
    }

    const existingUserParams = {
      TableName: process.env.USERS_TABLE,
      IndexName: 'emailIndex',
      KeyConditionExpression: '#email = :email',
      ExpressionAttributeNames: {
        '#email': 'email'
      },
      ExpressionAttributeValues: {
        ':email': data.email
      }
    };

    const existingUser = await dynamoDb.query(existingUserParams).promise();

    if (existingUser.Items && existingUser.Items.length > 0) {
      return {
        statusCode: 409,
        headers: { 'Content-Type': 'application/json' },
        body: JSON.stringify({ message: 'Користувач з таким email вже існує' })
      };
    }

    const userId = uuidv4();
    const timestamp = new Date().getTime();

    const params = {
      TableName: process.env.USERS_TABLE,
      Item: {
        userId,
        orgId,
        name: data.name,
        email: data.email,
        createdAt: timestamp,
        updatedAt: timestamp
      }
    };

    await dynamoDb.put(params).promise();

    return {
      statusCode: 201,
      headers: { 'Content-Type': 'application/json' },
      body: JSON.stringify(params.Item)
    };
  } catch (error) {
    console.error('Помилка створення користувача:', error);
    return {
      statusCode: 500,
      headers: { 'Content-Type': 'application/json' },
      body: JSON.stringify({ message: 'Не вдалося створити користувача', error: error.message })
    };
  }
};

module.exports.updateUser = async (event) => {
  try {
    const data = JSON.parse(event.body);
    const orgId = event.pathParameters.orgId;

    if (!data.userId || typeof data.userId !== 'string') {
      return {
        statusCode: 400,
        headers: { 'Content-Type': 'application/json' },
        body: JSON.stringify({ message: 'ID користувача обов\'язковий' })
      };
    }

    const orgParams = {
      TableName: process.env.ORGANIZATIONS_TABLE,
      Key: {
        orgId
      }
    };

    const org = await dynamoDb.get(orgParams).promise();

    if (!org.Item) {
      return {
        statusCode: 404,
        headers: { 'Content-Type': 'application/json' },
        body: JSON.stringify({ message: 'Організацію не знайдено' })
      };
    }

    const userParams = {
      TableName: process.env.USERS_TABLE,
      Key: {
        userId: data.userId
      }
    };

    const user = await dynamoDb.get(userParams).promise();

    if (!user.Item) {
      return {
        statusCode: 404,
        headers: { 'Content-Type': 'application/json' },
        body: JSON.stringify({ message: 'Користувача не знайдено' })
      };
    }

    if (user.Item.orgId !== orgId) {
      return {
        statusCode: 403,
        headers: { 'Content-Type': 'application/json' },
        body: JSON.stringify({ message: 'Користувач не належить до вказаної організації' })
      };
    }

    if (data.email && data.email !== user.Item.email) {
      const existingUserParams = {
        TableName: process.env.USERS_TABLE,
        IndexName: 'emailIndex',
        KeyConditionExpression: '#email = :email',
        ExpressionAttributeNames: {
          '#email': 'email'
        },
        ExpressionAttributeValues: {
          ':email': data.email
        }
      };

      const existingUser = await dynamoDb.query(existingUserParams).promise();

      if (existingUser.Items && existingUser.Items.length > 0) {
        return {
          statusCode: 409,
          headers: { 'Content-Type': 'application/json' },
          body: JSON.stringify({ message: 'Користувач з таким email вже існує' })
        };
      }
    }

    const timestamp = new Date().getTime();

    const updateParams = {
      TableName: process.env.USERS_TABLE,
      Key: {
        userId: data.userId
      },
      ExpressionAttributeNames: {
        '#name': 'name',
        '#email': 'email',
        '#updatedAt': 'updatedAt'
      },
      ExpressionAttributeValues: {
        ':name': data.name || user.Item.name,
        ':email': data.email || user.Item.email,
        ':updatedAt': timestamp
      },
      UpdateExpression: 'SET #name = :name, #email = :email, #updatedAt = :updatedAt',
      ReturnValues: 'ALL_NEW'
    };

    const result = await dynamoDb.update(updateParams).promise();

    return {
      statusCode: 200,
      headers: { 'Content-Type': 'application/json' },
      body: JSON.stringify(result.Attributes)
    };
  } catch (error) {
    console.error('Помилка оновлення користувача:', error);
    return {
      statusCode: 500,
      headers: { 'Content-Type': 'application/json' },
      body: JSON.stringify({ message: 'Не вдалося оновити користувача', error: error.message })
    };
  }
};