mirror of
https://github.com/schollz/cowyo.git
synced 2023-08-10 21:13:00 +03:00
e5cb53cf20
Former-commit-id: 0a0d82ebdcfe25a260b8a5ee7627dcb35c3b4da1 [formerly cc3c56e55c8d693b71273f45d61d896ffa95c452] [formerly 9f0f4545ffafe52055b824bf19cad7e0a13e7737 [formerly 8315ce08728423d204b84b6f2ef014fae56e0bb7 [formerly5ce388764a
]]] Former-commit-id: 2de411fcf47c6f180fdb00b5f0bbcc979e6f4b9f [formerly 116af09214ccc929ef1a164b9bf84429c88d34a1] Former-commit-id: 7b50bab330b8f2e0320140631d947e292a2d78db Former-commit-id:d86413147d
124 lines
3.8 KiB
Go
124 lines
3.8 KiB
Go
package main
|
|
|
|
import (
|
|
"flag"
|
|
"fmt"
|
|
"io/ioutil"
|
|
"os"
|
|
"path"
|
|
|
|
"github.com/boltdb/bolt"
|
|
"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 {
|
|
WikiName string
|
|
ExternalIP string
|
|
Port string
|
|
DatabaseLocation string
|
|
ServerCRT string
|
|
ServerKey string
|
|
SourcePath string
|
|
AdminKey string
|
|
Socket string
|
|
}
|
|
var VersionNum string
|
|
|
|
func main() {
|
|
VersionNum = "0.91"
|
|
// _, executableFile, _, _ := runtime.Caller(0) // get full path of this file
|
|
cwd, _ := os.Getwd()
|
|
databaseFile := path.Join(cwd, "data.db")
|
|
flag.StringVar(&RuntimeArgs.Port, "p", ":8003", "port to bind")
|
|
flag.StringVar(&RuntimeArgs.DatabaseLocation, "db", databaseFile, "location of database file")
|
|
flag.StringVar(&RuntimeArgs.AdminKey, "a", RandStringBytesMaskImprSrc(50), "key to access admin priveleges")
|
|
flag.StringVar(&RuntimeArgs.ServerCRT, "crt", "", "location of ssl crt")
|
|
flag.StringVar(&RuntimeArgs.ServerKey, "key", "", "location of ssl key")
|
|
flag.StringVar(&RuntimeArgs.WikiName, "w", "AwwKoala", "custom name for wiki")
|
|
dumpDataset := flag.Bool("dump", false, "flag to dump all data to 'dump' directory")
|
|
flag.CommandLine.Usage = func() {
|
|
fmt.Println(`AwwKoala (version ` + VersionNum + `): 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 yourserver.com'
|
|
Example: 'awwkoala -p :8080 localhost:8080'
|
|
Example: 'awwkoala -db /var/lib/awwkoala/db.bolt localhost:8003'
|
|
Example: 'awwkoala -p :8080 -crt ssl/server.crt -key ssl/server.key localhost:8080'
|
|
Options:`)
|
|
flag.CommandLine.PrintDefaults()
|
|
}
|
|
flag.Parse()
|
|
|
|
if *dumpDataset {
|
|
fmt.Println("Dumping data to 'dump' folder...")
|
|
Open(RuntimeArgs.DatabaseLocation)
|
|
dumpEverything()
|
|
Close()
|
|
os.Exit(1)
|
|
}
|
|
|
|
RuntimeArgs.ExternalIP = flag.Arg(0)
|
|
if RuntimeArgs.ExternalIP == "" {
|
|
RuntimeArgs.ExternalIP = GetLocalIP() + RuntimeArgs.Port
|
|
}
|
|
RuntimeArgs.SourcePath = cwd
|
|
|
|
Open(RuntimeArgs.DatabaseLocation)
|
|
defer Close()
|
|
|
|
// create programdata bucket
|
|
err := db.Update(func(tx *bolt.Tx) error {
|
|
_, err := tx.CreateBucketIfNotExists([]byte("programdata"))
|
|
if err != nil {
|
|
return fmt.Errorf("create bucket: %s", err)
|
|
}
|
|
return err
|
|
})
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
// Default page
|
|
aboutFile, _ := ioutil.ReadFile(path.Join(RuntimeArgs.SourcePath, "templates/aboutpage.md"))
|
|
p := WikiData{"about", "", []string{}, []string{}}
|
|
p.save(string(aboutFile))
|
|
|
|
// var q WikiData
|
|
// q.load("about")
|
|
// fmt.Println(getImportantVersions(q))
|
|
|
|
r := gin.Default()
|
|
r.LoadHTMLGlob(path.Join(RuntimeArgs.SourcePath, "templates/*"))
|
|
r.GET("/", newNote)
|
|
r.HEAD("/", func(c *gin.Context) { c.Status(200) })
|
|
r.GET("/:title", editNote)
|
|
r.GET("/:title/*option", everythingElse)
|
|
r.DELETE("/listitem", deleteListItem)
|
|
r.DELETE("/deletepage", deletePage)
|
|
if RuntimeArgs.ServerCRT != "" && RuntimeArgs.ServerKey != "" {
|
|
RuntimeArgs.Socket = "wss"
|
|
fmt.Println("--------------------------")
|
|
fmt.Println("AwwKoala (version " + VersionNum + ") is up and running on https://" + RuntimeArgs.ExternalIP)
|
|
fmt.Println("Admin key: " + RuntimeArgs.AdminKey)
|
|
fmt.Println("--------------------------")
|
|
r.RunTLS(RuntimeArgs.Port, RuntimeArgs.ServerCRT, RuntimeArgs.ServerKey)
|
|
} else {
|
|
RuntimeArgs.Socket = "ws"
|
|
fmt.Println("--------------------------")
|
|
fmt.Println("AwwKoala (version " + VersionNum + ") is up and running on http://" + RuntimeArgs.ExternalIP)
|
|
fmt.Println("Admin key: " + RuntimeArgs.AdminKey)
|
|
fmt.Println("--------------------------")
|
|
r.Run(RuntimeArgs.Port)
|
|
}
|
|
}
|