refactoring

main.go

@@ -5,22 +5,64 @@ import (
 	"text/template"
 )
 
+var (
+	defaultAppPath          = "src/helloworld/app"
+	allowedAPIProtocols     = []string{"rest", "websocket"}
+	allowedRestAPIEndpoints = []string{"regional", "edge", "private"}
+)
+
 type TmplData struct {
 	ApiProtocol        string
 	ApiEndpoints       string
 	LambdaFunctionName string
 	ApiProjectName     string
 }
 
-var allowedAPIProtocols = []string{"rest", "websocket"}
-var allowedRestAPIEndpoints = []string{"regional", "edge", "private"}
+type LanguageMapper struct {
+	Files       []string
+	TmplAppVar  string
+	TmplDepsVar string
+	AppPath     string
+	DepsPath    string
+}
 
 // Usage to-do
-var languages = map[string][]string{
-	"node":   []string{"index.js", "package.json"},
-	"java":   []string{"App.java", "pom.xml"},
-	"python": []string{"app.py", "requirements.txt"},
-	"go":     []string{"main.go"},
+var languages = map[string]LanguageMapper{
+	"node": LanguageMapper{
+		Files:       []string{"index.js", "package.json"},
+		TmplAppVar:  "nodeFunction",
+		TmplDepsVar: "packageJson",
+		AppPath:     defaultAppPath,
+		DepsPath:    defaultAppPath,
+	},
+	"java": LanguageMapper{
+		Files:       []string{"App.java", "pom.xml"},
+		TmplAppVar:  "",
+		TmplDepsVar: "",
+		AppPath:     defaultAppPath + "/com/api",
+		DepsPath:    defaultAppPath,
+	},
+	"python": LanguageMapper{
+		Files:       []string{"app.py", "requirements.txt"},
+		TmplAppVar:  "",
+		TmplDepsVar: "",
+		AppPath:     defaultAppPath,
+		DepsPath:    defaultAppPath,
+	},
+	"ruby": LanguageMapper{
+		Files:       []string{"app.rb", "Gemfile"},
+		TmplAppVar:  "",
+		TmplDepsVar: "",
+		AppPath:     defaultAppPath,
+		DepsPath:    defaultAppPath,
+	},
+	"go": LanguageMapper{
+		Files:       []string{"main.go", "go.mod"},
+		TmplAppVar:  "",
+		TmplDepsVar: "",
+		AppPath:     defaultAppPath,
+		DepsPath:    defaultAppPath,
+	},
 }
 
 func createDir(path string) error {

tmpl-nodejs.go → tmpl-languages.go

@@ -1,6 +1,8 @@
 package main
 
-var nodeFunction = `
+// node.js
+
+const nodeFunction = `
 'use strict'
 const  winston = require('winston')
 
@@ -26,7 +28,7 @@ exports.handler = function(event, context) {
 };
 `
 
-var packageJson = `
+const packageJson = `
 {
   "name": "hello",
   "version": "1.0.0",
@@ -46,3 +48,17 @@ var packageJson = `
   }
 }
 `
+
+// python
+const pythonFunction = `
+import json
+
+def lambda_handler(event, context):
+
+    dict = { "msg" : "hello world" }
+    
+    return {
+        'statusCode': 200,
+        'body': json.dumps(dict)
+    }
+`