-
Notifications
You must be signed in to change notification settings - Fork 0
/
delete_user_vehicle.js
56 lines (49 loc) · 1.57 KB
/
delete_user_vehicle.js
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
const aws = require('aws-sdk');
const dynamoDB = new aws.DynamoDB.DocumentClient();
let VehiclesTableName = process.env.dynamodb_vehicles_table_name;
let UsersTableName = process.env.dynamodb_users_table_name;
function stagingOverwrite() {
UsersTableName = `Dev_${process.env.dynamodb_users_table_name}`;
VehiclesTableName = `Dev_${process.env.dynamodb_vehicles_table_name}`;
}
async function getUserVehicle(userId, vehicleId) {
const params = {
TableName: UsersTableName,
Key: {
user_id: userId
},
ProjectionExpression: 'vehicles'
};
const data = await dynamoDB.get(params).promise();
if (data && data.Item && data.Item.vehicles) {
const filteredVehicleArray = data.Item.vehicles.filter((v) => v === vehicleId);
return filteredVehicleArray.length > 0 ? filteredVehicleArray[0] : null;
}
return null;
}
async function deleteUserVehicle(vehicleId) {
const params = {
TableName: VehiclesTableName,
Key: {
vehicle_id: vehicleId
}
};
const data = await dynamoDB.delete(params).promise();
return data;
}
exports.handler = async (event) => {
if (event.requestContext.stage === 'staging') { stagingOverwrite(); }
const vehicleId = event.pathParameters.vehicle;
const userId = event.requestContext.authorizer.user_id;
const userVehicle = await getUserVehicle(userId, vehicleId);
if (userVehicle) {
await deleteUserVehicle(vehicleId);
}
const message = {
action: 'delete', success: true, resource: 'vehicle', resource_id: vehicleId
};
const result = {
body: JSON.stringify(message)
};
return result;
};