1
0
mirror of https://github.com/muety/wakapi.git synced 2023-08-10 21:12:56 +03:00
wakapi/main.go

117 lines
2.8 KiB
Go
Raw Normal View History

2019-05-05 23:36:49 +03:00
package main
import (
2019-05-16 23:51:11 +03:00
"fmt"
2019-05-06 01:58:01 +03:00
"log"
2019-05-05 23:36:49 +03:00
"net/http"
"os"
"strconv"
"time"
2019-05-06 00:23:54 +03:00
"github.com/codegangsta/negroni"
"github.com/gorilla/mux"
2019-05-11 18:49:56 +03:00
"github.com/jinzhu/gorm"
2019-05-06 01:58:01 +03:00
"github.com/joho/godotenv"
2019-05-16 23:51:11 +03:00
ini "gopkg.in/ini.v1"
2019-05-06 00:23:54 +03:00
2019-05-06 01:40:41 +03:00
"github.com/n1try/wakapi/middlewares"
2019-05-05 23:36:49 +03:00
"github.com/n1try/wakapi/models"
2019-05-06 01:40:41 +03:00
"github.com/n1try/wakapi/routes"
2019-05-05 23:36:49 +03:00
"github.com/n1try/wakapi/services"
2019-05-11 18:49:56 +03:00
"github.com/n1try/wakapi/utils"
_ "github.com/jinzhu/gorm/dialects/mysql"
2019-05-05 23:36:49 +03:00
)
2019-05-16 23:53:03 +03:00
func readConfig() *models.Config {
2019-05-11 18:57:58 +03:00
if err := godotenv.Load(); err != nil {
2019-05-06 01:58:01 +03:00
log.Fatal(err)
}
dbUser, valid := os.LookupEnv("WAKAPI_DB_USER")
dbPassword, valid := os.LookupEnv("WAKAPI_DB_PASSWORD")
dbHost, valid := os.LookupEnv("WAKAPI_DB_HOST")
dbName, valid := os.LookupEnv("WAKAPI_DB_NAME")
2019-05-06 01:47:38 +03:00
2019-05-06 01:58:01 +03:00
if !valid {
2019-05-16 23:51:11 +03:00
log.Fatal("Environment variables missing.")
2019-05-06 01:58:01 +03:00
}
2019-05-06 01:47:38 +03:00
2019-05-16 23:51:11 +03:00
cfg, err := ini.Load("config.ini")
if err != nil {
log.Fatal(fmt.Sprintf("Fail to read file: %v", err))
}
port, err := strconv.Atoi(os.Getenv("PORT"))
if err != nil {
port = cfg.Section("server").Key("port").MustInt()
}
2019-05-16 23:51:11 +03:00
2019-05-16 23:53:03 +03:00
return &models.Config{
Port: port,
DbHost: dbHost,
DbUser: dbUser,
DbPassword: dbPassword,
DbName: dbName,
DbDialect: "mysql",
2019-05-05 23:36:49 +03:00
}
}
func main() {
// Read Config
2019-05-06 01:47:38 +03:00
config := readConfig()
2019-05-05 23:36:49 +03:00
2019-05-11 18:49:56 +03:00
// Connect to database
2019-05-16 23:53:03 +03:00
db, err := gorm.Open(config.DbDialect, utils.MakeConnectionString(config))
db.LogMode(true)
2019-05-05 23:36:49 +03:00
if err != nil {
2019-05-11 18:49:56 +03:00
// log.Fatal("Could not connect to database.")
log.Fatal(err)
2019-05-05 23:36:49 +03:00
}
2019-05-11 18:49:56 +03:00
defer db.Close()
// Migrate database schema
db.AutoMigrate(&models.User{})
db.AutoMigrate(&models.Heartbeat{}).AddForeignKey("user_id", "users(id)", "RESTRICT", "RESTRICT")
2019-05-05 23:36:49 +03:00
2019-05-06 01:40:41 +03:00
// Services
2019-05-16 23:53:03 +03:00
heartbeatSrvc := &services.HeartbeatService{config, db}
userSrvc := &services.UserService{config, db}
summarySrvc := &services.SummaryService{config, db, heartbeatSrvc}
2019-05-06 01:40:41 +03:00
// Handlers
heartbeatHandler := &routes.HeartbeatHandler{HeartbeatSrvc: heartbeatSrvc}
summaryHandler := &routes.SummaryHandler{SummarySrvc: summarySrvc}
2019-05-06 01:40:41 +03:00
// Middlewares
authenticate := &middlewares.AuthenticateMiddleware{UserSrvc: userSrvc}
2019-05-05 23:36:49 +03:00
2019-05-06 00:23:54 +03:00
// Setup Routing
router := mux.NewRouter()
apiRouter := mux.NewRouter().PathPrefix("/api").Subrouter()
// API Routes
heartbeats := apiRouter.Path("/heartbeat").Subrouter()
2019-05-06 01:40:41 +03:00
heartbeats.Methods("POST").HandlerFunc(heartbeatHandler.Post)
2019-05-06 00:23:54 +03:00
aggreagations := apiRouter.Path("/summary").Subrouter()
aggreagations.Methods("GET").HandlerFunc(summaryHandler.Get)
2019-05-06 00:23:54 +03:00
// Sub-Routes Setup
router.PathPrefix("/api").Handler(negroni.Classic().With(
2019-05-06 01:40:41 +03:00
negroni.HandlerFunc(authenticate.Handle),
2019-05-06 00:23:54 +03:00
negroni.Wrap(apiRouter),
))
2019-05-05 23:36:49 +03:00
// Listen HTTP
portString := ":" + strconv.Itoa(config.Port)
s := &http.Server{
2019-05-06 00:23:54 +03:00
Handler: router,
2019-05-05 23:36:49 +03:00
Addr: portString,
ReadTimeout: 10 * time.Second,
WriteTimeout: 10 * time.Second,
}
2019-05-06 01:58:01 +03:00
log.Printf("Listening on %+s\n", portString)
2019-05-05 23:36:49 +03:00
s.ListenAndServe()
}