fn-serverless/examples/blog/function.go

package main

import (
	"encoding/json"
	"fmt"
	"os"

	"gitlab.oracledx.com/odx/functions/examples/blog/database"
	"gitlab.oracledx.com/odx/functions/examples/blog/models"
	"gitlab.oracledx.com/odx/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("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
}