/
main.go
124 lines (111 loc) · 3.35 KB
/
main.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
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
package main
import (
"database/sql"
"embed"
"fmt"
"log"
"net/http"
"os"
"github.com/apexskier/httpauth"
"github.com/asaskevich/govalidator"
"github.com/turnkey-commerce/go-ping-sites/config"
"github.com/turnkey-commerce/go-ping-sites/controllers"
"github.com/turnkey-commerce/go-ping-sites/database"
"github.com/turnkey-commerce/go-ping-sites/notifier"
"github.com/turnkey-commerce/go-ping-sites/pinger"
)
var (
authBackend httpauth.SqlAuthBackend
authBackendFile = "go-ping-sites-auth.db"
roles map[string]httpauth.Role
authorizer httpauth.Authorizer
// Version is the Semver version of the application and should be passed in the build.
version = "No Version provided"
)
//go:embed public/*
var publicFiles embed.FS
//go:embed templates/*
var templateFiles embed.FS
func main() {
var err error
// Setup the main db.
var db *sql.DB
err = pinger.CreatePingerLog("", false)
if err != nil {
fatalError("Failed to initialize logging:", err)
}
startLog("Starting go-ping-sites version " + version + ", website on port " +
config.Settings.Website.HTTPPort + "...")
db, err = database.InitializeDB("go-ping-sites.db", "db-seed.toml")
if err != nil {
fatalError("Failed to initialize database:", err)
}
defer db.Close()
// Validate the required config for the website.
_, err = govalidator.ValidateStruct(config.Settings.Website)
if err != nil {
fatalError("Error in config.toml configuration:", err)
}
// Setup the auth
// create the authorization backend
authBackend, err = createAuthBackendFile()
if err != nil {
fatalError("Failed to create Auth Backend: ", err)
}
cookieKey := []byte(config.Settings.Website.CookieKey)
secureCookie := config.Settings.Website.SecureHTTPS
roles = getRoles()
authorizer, err = httpauth.NewAuthorizer(authBackend, cookieKey, "user", roles)
createDefaultUser()
// Start the Pinger
p := pinger.NewPinger(db, pinger.GetSites, pinger.RequestURL, notifier.SendEmail, notifier.SendSms)
p.Start()
// Start the web server.
templates := controllers.PopulateTemplates(templateFiles)
controllers.Register(db, authorizer, authBackend, roles, templates, p, version, cookieKey, secureCookie, publicFiles)
err = http.ListenAndServe(":"+config.Settings.Website.HTTPPort, nil) // set listen port
if err != nil {
fatalError("ListenAndServe: ", err)
}
}
func createDefaultUser() {
// create a default user
userName := "admin"
pwd := "adminpassword"
if _, err := authBackend.User(userName); err != nil {
defaultUser := httpauth.UserData{Username: userName, Email: "admin@localhost.com", Role: "admin"}
err = authBackend.SaveUser(defaultUser)
if err != nil {
panic(err)
}
err = authorizer.Update(nil, nil, userName, pwd, "")
if err != nil {
panic(err)
}
}
}
func startLog(message string) {
fmt.Println(message)
log.Println(message)
}
func fatalError(message string, content interface{}) {
fmt.Println(message, content)
log.Fatal(message, content)
}
func createAuthBackendFile() (s httpauth.SqlAuthBackend, err error) {
if _, err = os.Stat(authBackendFile); os.IsNotExist(err) {
_, err = os.Create(authBackendFile)
if err != nil {
return s, err
}
}
s, err = httpauth.NewSqlAuthBackend("sqlite3", authBackendFile)
return s, err
}
func getRoles() map[string]httpauth.Role {
// create some default roles
var r = make(map[string]httpauth.Role)
r["user"] = 30
r["admin"] = 80
return r
}