feat(serverless-cognito-user-pool): add support for username attribut…

…es and schema (#1160)
webiny · Jul 27, 2020 · fc9bebb · fc9bebb
1 parent a019b25
commit fc9bebb
Showing 1 changed file with 14 additions and 5 deletions.
diff --git a/packages/serverless-aws-cognito-user-pool/serverless.js b/packages/serverless-aws-cognito-user-pool/serverless.js
@@ -17,16 +17,24 @@ class ServerlessAwsCognito extends Component {
             this.context.instance.debug("Input was not changed, no action required.");
             return this.state.output;
         }
-
+        let schemaConfig = []
         const {
             region = "us-east-1",
             name,
             tags = {},
             appClients = [],
-            allowSignup = false
+            allowSignup = false,
+            schema
         } = inputs;
         const passwordPolicy = Object.assign({}, defaultPasswordPolicy, inputs.passwordPolicy);
-
+        const usernameAttributes = inputs.usernameAttributes ? inputs.usernameAttributes : ["email"]
+        if(schema){
+            if(Array.isArray(schema)){
+                schemaConfig = [...schema]
+            }else{
+                throw Error(`Schema attributes needs to be passed in array.`);
+            }
+        }
         const cognito = new Cognito({ region });
         this.state.output = this.state.output || { appClients: [] };
 
@@ -139,13 +147,14 @@ class ServerlessAwsCognito extends Component {
                         Mutable: true,
                         Name: "given_name",
                         Required: true
-                    }
+                    },
+                    ...schemaConfig
                 ],
                 UserPoolAddOns: {
                     AdvancedSecurityMode: "OFF" /* required */
                 },
                 UserPoolTags: Object.assign({}, tags),
-                UsernameAttributes: ["email"],
+                UsernameAttributes: [...usernameAttributes],
                 VerificationMessageTemplate: {
                     DefaultEmailOption: "CONFIRM_WITH_CODE"
                 }