aws-amplify-react-iot-pub-sub-using-lambda/amplify/backend/function/amplifyiotlambda/src/app.js at e493039581d2aff0faa3949086deead20a2c5385 · aws-samples/aws-amplify-react-iot-pub-sub-using-lambda · GitHub

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
148
149
150
151
152
153
154
155
156
157
/* Amplify Params - DO NOT EDIT
	ENV
	REGION
Amplify Params - DO NOT EDIT */
var express = require('express')
var bodyParser = require('body-parser')
var awsServerlessExpressMiddleware = require('aws-serverless-express/middleware')

// declare a new express app
var app = express()
app.use(bodyParser.json())
app.use(awsServerlessExpressMiddleware.eventContext())

// Enable CORS for all methods
app.use(function(req, res, next) {
  res.header("Access-Control-Allow-Origin", "*")
  res.header("Access-Control-Allow-Headers", "*")
  next()
});

var aws = require('aws-sdk')

/**********************
 * Example get method *
 **********************/

app.get('/item', function(req, res) {
  // Add your code here
  res.json({success: 'get call succeed!', url: req.url});
});

app.get('/item/*', function(req, res) {
  // Add your code here
  res.json({success: 'get call succeed!', url: req.url});
});

/****************************
* Example post method *
****************************/

app.post('/amplifyiotlambdaapi', function(req, res) {

  const iot = new aws.Iot({region: 'us-east-1', apiVersion: '2015-05-28', endpoint: 'https://iot.us-east-1.amazonaws.com'});

  const policyName = "amplifyiotlambdaapipolicy";
  const id = req.apiGateway.event.requestContext.identity.cognitoIdentityId;

  console.log("Create policy and attach with cognito identity id: " + id);
  var params = {policyName: policyName};
  iot.getPolicy(params , function(err, data) {
        if (err) {
             var policy = {"Version": "2012-10-17", "Statement": [{"Effect": "Allow", "Action": ["*"],"Resource": ["*"]}]};
             var policyDoc = JSON.stringify(policy);

             console.log("Creating policy: " + policyName + " with doc: " + policyDoc);

             var params = {
                     policyName: policyName,
                     policyDocument: policyDoc
                     };

             iot.createPolicy(params , function(err, data) {
                 if (err) {
                      //console.error(err);
                      if (err.code !== 'ResourceAlreadyExistsException') {
                         console.log(err);
                         res.json({error: err, url: req.url, body: req.body});
                      }
                 }
                 else {
                    console.log("CreatePolicy response=" + data);
                    var params = {policyName: policyName, target: id};

                    console.log("Attach IoT Policy: " + policyName + " with cognito identity id: " + id);
                    iot.attachPolicy(params, function(err, data) {
                         if (err) {
                               //console.error(err);
                               if (err.code !== 'ResourceAlreadyExistsException') {
                                  console.log(err);
                                  res.json({error: err, url: req.url, body: req.body});
                               }
                          }
                         else  {
                            console.log(data);
                            res.json({success: 'Create and attach policy call succeed!', url: req.url, body: req.body});
                         }
                     });
                 }
             });
        }
        else {
           console.log("Policy " + policyName + " already exists..");

           var params = {policyName: policyName, target: id};

           console.log("Attach IoT Policy: " + policyName + " with cognito identity id: " + id);
           iot.attachPolicy(params, function(err, data) {
                if (err) {
                      //console.error(err);
                      if (err.code !== 'ResourceAlreadyExistsException') {
                         console.log(err);
                         res.json({error: err, url: req.url, body: req.body});
                      }
                 }
                else  {
                   console.log(data);
                   res.json({success: 'Create and attach policy call succeed!', url: req.url, body: req.body});
                }
            });
        }
    });
});

app.post('/item', function(req, res) {
  // Add your code here
  res.json({success: 'post call succeed!', url: req.url, body: req.body})
});

app.post('/item/*', function(req, res) {
  // Add your code here
  res.json({success: 'post call succeed!', url: req.url, body: req.body})
});

/****************************
* Example put method *
****************************/

app.put('/item', function(req, res) {
  // Add your code here
  res.json({success: 'put call succeed!', url: req.url, body: req.body})
});

app.put('/item/*', function(req, res) {
  // Add your code here
  res.json({success: 'put call succeed!', url: req.url, body: req.body})
});

/****************************
* Example delete method *
****************************/

app.delete('/item', function(req, res) {
  // Add your code here
  res.json({success: 'delete call succeed!', url: req.url});
});

app.delete('/item/*', function(req, res) {
  // Add your code here
  res.json({success: 'delete call succeed!', url: req.url});
});

app.listen(3000, function() {
    console.log("App started")
});


module.exports = app