1
0
mirror of https://github.com/schollz/cowyo.git synced 2023-08-10 21:13:00 +03:00
cowyo/main.go
Zack Scholl 654d8163b1 Renaming
Former-commit-id: ea07c6f1b4ad1fe779161e8c63b68175e0f20774 [formerly 830cdbb6e964f2adc6780c40c6472457573e550f] [formerly a46a5f25adeda78033b824cbf76878d0a5a45247 [formerly 40909bd84c4eab75dc11465a12d2971b30f2d87e [formerly fdc6d22cae]]]
Former-commit-id: d6df2121622424298556a07eedb3f699abcf8d5b [formerly 4c4fac131114fc5ae07cd8e6be8229988ba96bf1]
Former-commit-id: e34de2d461f8cb277b09ad69e27497797ea39305
Former-commit-id: 866dc6b32d
2016-02-10 09:22:28 -05:00

82 lines
2.4 KiB
Go

package main
import (
"flag"
"fmt"
"log"
"path"
"runtime"
"github.com/gin-gonic/gin"
)
// AllowedIPs is a white/black list of
// IP addresses allowed to access awwkoala
var AllowedIPs = map[string]bool{
"192.168.1.13": true,
"192.168.1.12": true,
"192.168.1.2": true,
}
// RuntimeArgs contains all runtime
// arguments available
var RuntimeArgs struct {
ExternalIP string
Port string
DatabaseLocation string
ServerCRT string
ServerKey string
SourcePath string
AdminKey string
}
func main() {
_, executableFile, _, _ := runtime.Caller(0) // get full path of this file
databaseFile := path.Join(path.Dir(executableFile), "data.db")
flag.StringVar(&RuntimeArgs.Port, "p", ":12312", "port to bind")
flag.StringVar(&RuntimeArgs.DatabaseLocation, "db", databaseFile, "location of database file")
flag.StringVar(&RuntimeArgs.AdminKey, "a", "", "key to access admin priveleges")
flag.StringVar(&RuntimeArgs.ServerCRT, "crt", "", "location of ssl crt")
flag.StringVar(&RuntimeArgs.ServerKey, "key", "", "location of ssl key")
flag.CommandLine.Usage = func() {
fmt.Println(`AwwKoala: A Websocket Wiki and Kind Of A List Application
run this to start the server and then visit localhost at the port you specify
(see parameters).
Example: 'awwkoala localhost'
Example: 'awwkoala -p :8080 localhost:8080'
Example: 'awwkoala -db /var/lib/awwkoala/db.bolt localhost:12312'
Example: 'awwkoala -p :8080 -crt ssl/server.crt -key ssl/server.key localhost:8080'
Options:`)
flag.CommandLine.PrintDefaults()
}
flag.Parse()
RuntimeArgs.ExternalIP = flag.Arg(0)
if RuntimeArgs.ExternalIP == "" {
log.Fatal("You need to specify the external IP address")
}
RuntimeArgs.SourcePath = path.Dir(executableFile)
Open(RuntimeArgs.DatabaseLocation)
defer Close()
// Default page
p := WikiData{"about", aboutPageText, []string{}, []string{}}
p.save(aboutPageText)
// var q WikiData
// q.load("about")
// fmt.Println(getImportantVersions(q))
r := gin.Default()
r.LoadHTMLGlob(path.Join(RuntimeArgs.SourcePath, "templates/*"))
r.GET("/", newNote)
r.GET("/:title", editNote)
r.GET("/:title/*option", everythingElse)
r.DELETE("/listitem", deleteListItem)
if RuntimeArgs.ServerCRT != "" && RuntimeArgs.ServerKey != "" {
r.RunTLS(RuntimeArgs.Port, RuntimeArgs.ServerCRT, RuntimeArgs.ServerKey)
} else {
log.Println("No crt/key found, running non-https")
r.Run(RuntimeArgs.Port)
}
}