/
api.go
73 lines (57 loc) · 1.32 KB
/
api.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
package api
import (
"log"
"github.com/gin-contrib/cors"
"github.com/gin-gonic/gin"
"github.com/millken/tcpwder/config"
)
/* gin app */
var app *gin.Engine
/**
* Initialize module
*/
func init() {
gin.SetMode(gin.ReleaseMode)
}
/**
* Starts REST API server
*/
func Start(cfg config.ApiConfig) {
if !cfg.Enabled {
log.Printf("[INFO] API disabled")
return
}
log.Printf("[INFO] Starting up API")
app = gin.New()
if cfg.Cors {
corsConfig := cors.DefaultConfig()
corsConfig.AllowAllOrigins = true
corsConfig.AllowCredentials = true
corsConfig.AllowMethods = []string{"PUT", "POST", "DELETE", "GET", "OPTIONS"}
corsConfig.AllowHeaders = []string{"Origin", "Authorization"}
app.Use(cors.New(corsConfig))
log.Printf("[INFO] API CORS enabled")
}
r := app.Group("/")
if cfg.BasicAuth != nil {
log.Printf("[INFO] Using HTTP Basic Auth")
r.Use(gin.BasicAuth(gin.Accounts{
cfg.BasicAuth.Login: cfg.BasicAuth.Password,
}))
}
/* attach endpoints */
attachRoot(r)
attachServers(r)
var err error
/* start rest api server */
if cfg.Tls != nil {
log.Printf("[INFO] Starting HTTPS server %s", cfg.Bind)
err = app.RunTLS(cfg.Bind, cfg.Tls.CertPath, cfg.Tls.KeyPath)
} else {
log.Printf("[INFO] Starting HTTP server %s", cfg.Bind)
err = app.Run(cfg.Bind)
}
if err != nil {
log.Fatal(err)
}
}