talent/main.go

109 lines
2.9 KiB
Go

package main
import (
"encoding/base64"
"flag"
"fmt"
"net/http"
asset "talent/assets"
"time"
"github.com/gin-gonic/gin"
"github.com/rs/zerolog/log"
)
var (
flagAddr string
flagDev bool
flagMigrate bool
flagBase string
DB_HOST string
DB_USER string
DB_PASSWD string
DB_NAME string
DB_PORT int
LINE_SECRET string
LINE_TOKEN string
IBM_INSTANCE string
IBM_ASSISTANT string
IBM_API string
Version string
Build string
)
func init() {
flag.StringVar(&flagAddr, "addr", "0.0.0.0:8000", "address to listen on.")
flag.StringVar(&flagBase, "base", "https://bot.ework.tw", "base url.")
flag.BoolVar(&flagDev, "dev", false, "is dev mode")
flag.BoolVar(&flagMigrate, "migrate", false, "migrate database")
flag.StringVar(&DB_HOST, "db-host", "postgres.ework.tw", "postgress db host.")
flag.StringVar(&DB_USER, "db-user", "talent", "postgress db user.")
flag.StringVar(&DB_PASSWD, "db-passwd", "show", "postgress db pqssword.")
flag.StringVar(&DB_NAME, "db-name", "talent", "postgress db name.")
flag.IntVar(&DB_PORT, "db-port", 5432, "postgress db port.")
flag.StringVar(&LINE_SECRET, "line-secret", "cf406aa7577569ba2b211bc04f51630e", "line channel secret.")
flag.StringVar(&LINE_TOKEN, "line-token", "IhP4dLM91boFQlRzHa/Iv9el2xXm2X6ByiDuWQXm6ndQL1LA+yA+O8x2OeuJkpSWlf5IE8cwkA+Mca18EjO42Q6vRm0T5cRdTLkO+42SX9HAx2GdJnhWu+S4IplOt38YktfmpodCyk6bXDBJp9YVwgdB04t89/1O/w1cDnyilFU=", "channel access token")
flag.StringVar(&IBM_INSTANCE, "ibm-instance", "d9329221-40fb-4911-ba9f-69098349b8ab", "watson instance id")
flag.StringVar(&IBM_ASSISTANT, "ibm-assistant", "13cb9242-722a-4644-ba9b-7a761ffe807e", "watson assistant id")
flag.StringVar(&IBM_API, "ibm-api", "tLHR-Fhob0UideN1HUQdQ5grtl338X_VrrT31yy8FbKB", "watson api token")
}
func main() {
tz, err := time.LoadLocation("Asia/Taipei")
if err == nil {
time.Local = tz
}
flag.Parse()
setupLog()
setupServer()
connectToDB()
if flagMigrate {
migrateDB()
}
setupLine()
IBM_API = fmt.Sprintf("Basic %v", base64.URLEncoding.EncodeToString([]byte(fmt.Sprintf("apikey:%v", IBM_API))))
// setup route
server.GET("/", func(c *gin.Context) { OK(c, "ok") })
server.GET("/check", func(c *gin.Context) {
c.Writer.Write(check_web)
})
server.GET("/profile", func(c *gin.Context) {
c.Writer.Write(profile_web)
})
server.GET("/claim", func(c *gin.Context) {
c.Writer.Write(claim_web)
})
server.GET("/product", func(c *gin.Context) {
c.Writer.Write(product_web)
})
// setup api
api := server.Group("api")
api.POST("/email", ApiUpdateEmail)
api.POST("/attach/:oid", ApiPostReceipt)
// setup bot webhook
hook := server.Group("bot")
hook.POST("/", webhook)
// setup static files
server.StaticFS("static", http.FS(asset.FS))
server.Static("attachment", "attachment")
// start server
log.Info().Msgf("serving on %v", flagAddr)
if err := server.Run(flagAddr); err != nil {
panic(err)
}
}