Examples changes (#201)

examples/blog/function.go

@@ -0,0 +1,90 @@
+package main
+
+import (
+	"encoding/json"
+	"fmt"
+	"os"
+
+	"github.com/iron-io/functions/examples/blog/database"
+	"github.com/iron-io/functions/examples/blog/models"
+	"github.com/iron-io/functions/examples/blog/routes"
+)
+
+var noAuth = map[string]interface{}{}
+
+func main() {
+	request := fmt.Sprintf("%s %s", os.Getenv("METHOD"), os.Getenv("ROUTE"))
+
+	dbURI := os.Getenv("CONFIG_DB")
+	if dbURI == "" {
+		dbURI = "127.0.0.1/blog"
+	}
+	db := database.New(dbURI)
+
+	if created := createUser(db); created {
+		return
+	}
+
+	// PUBLIC REQUESTS
+	switch request {
+	case "GET /posts":
+		route.HandlePostList(db, noAuth)
+		return
+	case "GET /posts/:id":
+		route.HandlePostRead(db, noAuth)
+		return
+	}
+
+	// GETTING TOKEN
+	if os.Getenv("ROUTE") == "/token" {
+		route.HandleToken(db)
+		return
+	}
+
+	// AUTHENTICATION
+	auth, valid := route.Authentication()
+	if !valid {
+		route.SendError("Invalid authentication")
+		return
+	}
+
+	// AUTHENTICATED ONLY REQUESTS
+	if request == "POST /posts" {
+		route.HandlePostCreate(db, auth)
+		return
+	}
+
+	route.SendError("Not found")
+}
+
+func createUser(db *database.Database) bool {
+	env := os.Getenv("NEWUSER")
+
+	if env == "" {
+		return false
+	}
+
+	var user *models.User
+	err := json.Unmarshal([]byte(env), &user)
+	if err != nil {
+		fmt.Println(err)
+		return true
+	}
+
+	if user.Username == "" || user.NewPassword == "" {
+		fmt.Println("missing username or password")
+		return true
+	}
+
+	user.Password = []byte(user.NewPassword)
+	user.NewPassword = ""
+
+	user, err = db.SaveUser(user)
+	if err != nil {
+		fmt.Println("couldn't create user")
+	} else {
+		fmt.Println("user created")
+	}
+
+	return true
+}