mirror of
https://github.com/schollz/cowyo.git
synced 2023-08-10 21:13:00 +03:00
bba9d5b454
Former-commit-id: e8f547f5d313841f5d7d4d948fc41647e133d6be [formerly eab73aa394d77bbdc282424608474169b7c59d32] [formerly 390884efb34984f8e12bab5539d22708bfac9bfe [formerly 39b0a914769e18e09e945d12b4a4cf8ea76d6f4d [formerlyade7a635c4]]] Former-commit-id: c6b97dfebd3e67ae41995ee2c7460fe42ba6fe9b [formerly 730e316ee79a607e336c32ffb9f35778bd257e64] Former-commit-id: 24302f964c296a31cff2d75d0d406ab3e199fe21 Former-commit-id:c9bbc821be
78 lines
2.1 KiB
Go
78 lines
2.1 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 cowyo
|
|
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
|
|
}
|
|
|
|
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.ServerCRT, "crt", "", "location of ssl crt")
|
|
flag.StringVar(&RuntimeArgs.ServerKey, "key", "", "location of ssl key")
|
|
flag.CommandLine.Usage = func() {
|
|
fmt.Println(`cowyo: a websocket notepad
|
|
run this to start the server and then visit localhost at the port you specify
|
|
(see parameters).
|
|
Example: 'cowyo localhost'
|
|
Example: 'cowyo -p :8080 localhost'
|
|
Example: 'cowyo -db /var/lib/cowyo/db.bolt localhost'
|
|
Example: 'cowyo -p :8080 -crt ssl/server.crt -key ssl/server.key localhost'
|
|
Options:`)
|
|
flag.CommandLine.PrintDefaults()
|
|
}
|
|
flag.Parse()
|
|
RuntimeArgs.ExternalIP = flag.Arg(0)
|
|
if RuntimeArgs.ExternalIP == "" {
|
|
log.Fatal("You need to specify the external IP address")
|
|
}
|
|
Open(RuntimeArgs.DatabaseLocation)
|
|
defer Close()
|
|
|
|
// Default page
|
|
p := CowyoData{"about", about_page, []string{}, []string{}}
|
|
p.save(about_page)
|
|
|
|
//var q CowyoData
|
|
//q.load("SpikySeaSlug")
|
|
//rebuildTexts(q)
|
|
|
|
r := gin.Default()
|
|
r.LoadHTMLGlob(path.Join(path.Dir(executableFile), "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)
|
|
}
|
|
}
|