main.go

package main

import (
	"flag"
	"log"
	"os"
	"time"

	"github.com/gin-contrib/cors"
	"github.com/securityfirst/umbrella-api/utils"

	"github.com/securityfirst/tent"
	"github.com/securityfirst/tent/auth"
	"github.com/securityfirst/tent/repo"

	"github.com/gin-gonic/gin"
	"github.com/tmilewski/goenv"
)

var isProduction bool

var conf = auth.Config{
	OAuthHost: "http://127.0.0.1:8080",
	Login:     auth.HandleConf{"/auth/login", "/"},
	Logout:    auth.HandleConf{"/auth/logout", "/"},
	Callback:  auth.HandleConf{"/auth/callback", "/"},
}

func init() {
	err := goenv.Load()
	utils.CheckErr(err)
	envArg := flag.String("env", "development", "Environment")
	flag.Parse()
	if *envArg == "production" {
		isProduction = true
	}
	conf.ID = os.Getenv("GITHUB_ID")
	conf.Secret = os.Getenv("GITHUB_SECRET")
	if conf.ID == "" || conf.Secret == "" {
		log.Println("GITHUB_ID/GITHUB_SECRET not found. Please check your environment")
		os.Exit(1)
	}
	if host := os.Getenv("APP_HOST"); host != "" {
		conf.OAuthHost = host
	}
	conf.State = os.Getenv("APP_SECRET")
	log.SetFlags(log.Ltime | log.Lshortfile)
}

func main() {
	um := getUmbrella()
	go func() {
		tick := time.Tick(time.Hour)
		for {
			log.Println("Feed update - Starting...")
			if err := um.UpdateFeeds(); err != nil {
				log.Println("Feed update - error:", err)
				continue
			}
			log.Println("Feed update - complete")
			<-tick
		}
	}()
	r := gin.Default()
	r.Use(cors.New(cors.Config{
		AllowOrigins:     []string{"https://api.secfirst.org"},
		AllowMethods:     []string{"GET", "PUT", "POST", "DELETE"},
		AllowHeaders:     []string{"Origin", "Authorization", "Content-Type"},
		ExposeHeaders:    []string{"Content-Length", "Access-Control-Allow-Origin"},
		AllowCredentials: true,
		MaxAge:           12 * time.Hour,
	}))

	v1 := r.Group("/v1")
	{
		v1.GET("/feed", um.getFeed)
		v1.GET("/segments", um.getSegments)
		v1.GET("/check_items", um.Auth(false), um.getCheckItems)
		v1.GET("/categories", um.Auth(false), um.getCategories)
		v1.GET("/categories/:id", um.Auth(false), um.getCategory)
		v1.GET("/languages", um.Auth(false), um.getLanguages)
	}
	v2 := r.Group("/v2")
	v3 := r.Group("/v3")
	{
		r, err := repo.New("securityfirst", "tent-content", "difficulty")
		if err != nil {
			log.Fatalf("Repo error: %s", err)
		}
		o := tent.New(r)
		// No authentication
		v2.GET("/feed", um.getFeed)
		v3.GET("/feed", um.getFeed)
		// Autentication
		cfg := auth.Config{
			ID:        conf.ID,
			Secret:    conf.Secret,
			OAuthHost: conf.OAuthHost,
			Host:      conf.Host,
			State:     conf.State,
			Login:     auth.HandleConf{conf.Login.Endpoint, conf.Login.Redirect},
			Logout:    auth.HandleConf{conf.Logout.Endpoint, conf.Logout.Redirect},
			Callback:  auth.HandleConf{conf.Callback.Endpoint, conf.Callback.Redirect},
		}
		o.Register(v2, cfg)
		o.Register(v3, cfg)
	}
	r.Run(":8080")

}