main.go

package main

import (
	"fmt"

	"github.com/gin-contrib/cors"
	"github.com/gin-contrib/static"
	"github.com/gin-gonic/gin"
	"github.com/tianhai82/stock-timing/authen"
	"github.com/tianhai82/stock-timing/cloudtask"
	"github.com/tianhai82/stock-timing/cron"
	"github.com/tianhai82/stock-timing/rpcs"
)

const domain = "https://stock-timing.web.app"

func main() {
	fmt.Println("Stock Timing starting")
	r := gin.Default()
	domains := []string{domain}
	if gin.Mode() == gin.DebugMode {
		r.Use(static.Serve("/", static.LocalFile("./public", false)))
		domains = append(domains, "http://localhost:8080")
	}

	r.Use(cors.New(cors.Config{
		AllowOrigins:     domains,
		AllowCredentials: true,
		AllowHeaders:     []string{"Authorization"},
		AllowMethods:     []string{"GET", "POST", "DELETE"},
	}))
	rpcsRouter := r.Group("/rpc")
	rpcs.AddEtoroRpcs(rpcsRouter)
	authRouter := rpcsRouter.Group("/auth", authen.AuthCheck)
	rpcs.AddSubscriptionRpcs(authRouter)

	tdaRouter := rpcsRouter.Group("/tda", authen.TdaAuth)
	rpcs.AddTdaRpcs(tdaRouter)

	tigerRouter := rpcsRouter.Group("/tiger")
	rpcs.AddTigerRpcs(tigerRouter)

	cronRouter := r.Group("/cron")
	if gin.Mode() != gin.DebugMode {
		cronRouter.Use(func(c *gin.Context) {
			cronHeader := c.GetHeader("X-Appengine-Cron")
			if cronHeader == "" {
				c.Abort()
			}
		})
	}
	cron.AddCronJobs(cronRouter)

	taskRouter := r.Group("/task")
	if gin.Mode() != gin.DebugMode {
		taskRouter.Use(func(c *gin.Context) {
			taskHeader := c.GetHeader("X-Appengine-Taskname")
			if taskHeader == "" {
				c.Abort()
			}
		})
	}
	cloudtask.AddCloudTasks(taskRouter)

	err := r.Run()
	if err != nil {
		fmt.Println(err)
	}
}