mirror of https://github.com/muety/wakapi.git
73 lines
1.5 KiB
Go
73 lines
1.5 KiB
Go
package main
|
|
|
|
import (
|
|
"database/sql"
|
|
"flag"
|
|
"fmt"
|
|
"net/http"
|
|
"os"
|
|
"strconv"
|
|
"time"
|
|
|
|
"github.com/codegangsta/negroni"
|
|
"github.com/gorilla/mux"
|
|
|
|
_ "github.com/go-sql-driver/mysql"
|
|
"github.com/n1try/wakapi/models"
|
|
"github.com/n1try/wakapi/services"
|
|
)
|
|
|
|
var HeartbeatSrvc services.HeartbeatService
|
|
|
|
func getConfig() models.Config {
|
|
portPtr := flag.Int("port", 8080, "Port for the webserver to listen on")
|
|
flag.Parse()
|
|
return models.Config{
|
|
Port: *portPtr,
|
|
}
|
|
}
|
|
|
|
func main() {
|
|
// Read Config
|
|
config := getConfig()
|
|
|
|
// Connect Database
|
|
db, _ := sql.Open("mysql", "fakatime_user:eB2zyLt2heqWj5Y9@tcp(muetsch.io:3306)/fakatime")
|
|
defer db.Close()
|
|
err := db.Ping()
|
|
if err != nil {
|
|
fmt.Println("Could not connect to database.")
|
|
os.Exit(1)
|
|
}
|
|
|
|
// Init Services
|
|
HeartbeatSrvc = services.HeartbeatService{db}
|
|
|
|
// Setup Routing
|
|
router := mux.NewRouter()
|
|
apiRouter := mux.NewRouter().PathPrefix("/api").Subrouter()
|
|
n := negroni.Classic()
|
|
n.UseHandler(router)
|
|
|
|
// API Routes
|
|
heartbeats := apiRouter.Path("/heartbeat").Subrouter()
|
|
heartbeats.Methods("POST").HandlerFunc(HeartbeatHandler)
|
|
|
|
// Sub-Routes Setup
|
|
router.PathPrefix("/api").Handler(negroni.Classic().With(
|
|
negroni.HandlerFunc(AuthenticateMiddleware),
|
|
negroni.Wrap(apiRouter),
|
|
))
|
|
|
|
// Listen HTTP
|
|
portString := ":" + strconv.Itoa(config.Port)
|
|
s := &http.Server{
|
|
Handler: router,
|
|
Addr: portString,
|
|
ReadTimeout: 10 * time.Second,
|
|
WriteTimeout: 10 * time.Second,
|
|
}
|
|
fmt.Printf("Listening on %+s\n", portString)
|
|
s.ListenAndServe()
|
|
}
|