Files
fn-serverless/examples/blog/function.go
C Cirello a61c4dab78 Assert License (#224)
* license: assert license for Go code
* license: add in shell scripts
* license: assert license for Ruby code
* license: assert license to individual cases
* license: assert license to Dockerfile
2016-11-05 23:33:07 +01:00

105 lines
2.1 KiB
Go

// Copyright 2016 Iron.io
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
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
}