gosora/main.go

733 lines
16 KiB
Go

/*
*
* Gosora Main File
* Copyright Azareal 2016 - 2020
*
*/
// Package main contains the main initialisation logic for Gosora
package main // import "github.com/Azareal/Gosora"
import (
"bytes"
"crypto/tls"
"flag"
"fmt"
"io"
"log"
"mime"
"net/http"
"os"
"os/signal"
"runtime"
"runtime/debug"
"runtime/pprof"
"strconv"
"strings"
"syscall"
"time"
c "github.com/Azareal/Gosora/common"
co "github.com/Azareal/Gosora/common/counters"
meta "github.com/Azareal/Gosora/common/meta"
p "github.com/Azareal/Gosora/common/phrases"
_ "github.com/Azareal/Gosora/extend"
qgen "github.com/Azareal/Gosora/query_gen"
"github.com/Azareal/Gosora/routes"
"github.com/fsnotify/fsnotify"
//"github.com/lucas-clemente/quic-go/http3"
"github.com/pkg/errors"
)
var router *GenRouter
// TODO: Wrap the globals in here so we can pass pointers to them to subpackages
var globs *Globs
type Globs struct {
stmts *Stmts
}
// Temporary alias for renderTemplate
func init() {
c.RenderTemplateAlias = routes.RenderTemplate
}
func afterDBInit() (err error) {
if err := storeInit(); err != nil {
return err
}
log.Print("Exitted storeInit")
c.GzipStartEtag = "\"" + strconv.FormatInt(c.StartTime.Unix(), 10) + "-ng\""
c.StartEtag = "\"" + strconv.FormatInt(c.StartTime.Unix(), 10) + "-n\""
var uids []int
tc := c.Topics.GetCache()
if tc != nil {
log.Print("Preloading topics")
// Preload ten topics to get the wheels going
var count = 10
if tc.GetCapacity() <= 10 {
count = 2
if tc.GetCapacity() <= 2 {
count = 0
}
}
group, err := c.Groups.Get(c.GuestUser.Group)
if err != nil {
return err
}
// TODO: Use the same cached data for both the topic list and the topic fetches...
tList, _, _, err := c.TopicList.GetListByCanSee(group.CanSee, 1, 0, nil)
if err != nil {
return err
}
ctList := make([]*c.TopicsRow, len(tList))
copy(ctList, tList)
tList, _, _, err = c.TopicList.GetListByCanSee(group.CanSee, 2, 0, nil)
if err != nil {
return err
}
for _, tItem := range tList {
ctList = append(ctList, tItem)
}
tList, _, _, err = c.TopicList.GetListByCanSee(group.CanSee, 3, 0, nil)
if err != nil {
return err
}
for _, tItem := range tList {
ctList = append(ctList, tItem)
}
if count > len(ctList) {
count = len(ctList)
}
for i := 0; i < count; i++ {
_, _ = c.Topics.Get(ctList[i].ID)
}
}
uc := c.Users.GetCache()
if uc != nil {
// Preload associated users too...
for _, uid := range uids {
_, _ = c.Users.Get(uid)
}
}
log.Print("Exitted afterDBInit")
return nil
}
// Experimenting with a new error package here to try to reduce the amount of debugging we have to do
// TODO: Dynamically register these items to avoid maintaining as much code here?
func storeInit() (e error) {
acc := qgen.NewAcc()
ws := errors.WithStack
var rcache c.ReplyCache
if c.Config.ReplyCache == "static" {
rcache = c.NewMemoryReplyCache(c.Config.ReplyCacheCapacity)
}
c.Rstore, e = c.NewSQLReplyStore(acc, rcache)
if e != nil {
return ws(e)
}
c.Prstore, e = c.NewSQLProfileReplyStore(acc)
if e != nil {
return ws(e)
}
c.Likes, e = c.NewDefaultLikeStore(acc)
if e != nil {
return ws(e)
}
c.ForumActionStore, e = c.NewDefaultForumActionStore(acc)
if e != nil {
return ws(e)
}
c.Convos, e = c.NewDefaultConversationStore(acc)
if e != nil {
return ws(e)
}
c.UserBlocks, e = c.NewDefaultBlockStore(acc)
if e != nil {
return ws(e)
}
c.GroupPromotions, e = c.NewDefaultGroupPromotionStore(acc)
if e != nil {
return ws(e)
}
if e = p.InitPhrases(c.Site.Language); e != nil {
return ws(e)
}
if e = c.InitEmoji(); e != nil {
return ws(e)
}
if e = c.InitWeakPasswords(); e != nil {
return ws(e)
}
log.Print("Loading the static files.")
if e = c.Themes.LoadStaticFiles(); e != nil {
return ws(e)
}
if e = c.StaticFiles.Init(); e != nil {
return ws(e)
}
if e = c.StaticFiles.JSTmplInit(); e != nil {
return ws(e)
}
log.Print("Initialising the widgets")
c.Widgets = c.NewDefaultWidgetStore()
if e = c.InitWidgets(); e != nil {
return ws(e)
}
log.Print("Initialising the menu item list")
c.Menus = c.NewDefaultMenuStore()
if e = c.Menus.Load(1); e != nil { // 1 = the default menu
return ws(e)
}
menuHold, e := c.Menus.Get(1)
if e != nil {
return ws(e)
}
fmt.Printf("menuHold: %+v\n", menuHold)
var b bytes.Buffer
menuHold.Build(&b, &c.GuestUser, "/")
fmt.Println("menuHold output: ", string(b.Bytes()))
log.Print("Initialising the authentication system")
c.Auth, e = c.NewDefaultAuth()
if e != nil {
return ws(e)
}
log.Print("Initialising the stores")
c.WordFilters, e = c.NewDefaultWordFilterStore(acc)
if e != nil {
return ws(e)
}
c.MFAstore, e = c.NewSQLMFAStore(acc)
if e != nil {
return ws(e)
}
c.Pages, e = c.NewDefaultPageStore(acc)
if e != nil {
return ws(e)
}
c.Reports, e = c.NewDefaultReportStore(acc)
if e != nil {
return ws(e)
}
c.Emails, e = c.NewDefaultEmailStore(acc)
if e != nil {
return ws(e)
}
c.LoginLogs, e = c.NewLoginLogStore(acc)
if e != nil {
return ws(e)
}
c.RegLogs, e = c.NewRegLogStore(acc)
if e != nil {
return ws(e)
}
c.ModLogs, e = c.NewModLogStore(acc)
if e != nil {
return ws(e)
}
c.AdminLogs, e = c.NewAdminLogStore(acc)
if e != nil {
return ws(e)
}
c.IPSearch, e = c.NewDefaultIPSearcher()
if e != nil {
return ws(e)
}
if c.Config.Search == "" || c.Config.Search == "sql" {
c.RepliesSearch, e = c.NewSQLSearcher(acc)
if e != nil {
return ws(e)
}
}
c.Subscriptions, e = c.NewDefaultSubscriptionStore()
if e != nil {
return ws(e)
}
c.Attachments, e = c.NewDefaultAttachmentStore(acc)
if e != nil {
return ws(e)
}
c.Polls, e = c.NewDefaultPollStore(c.NewMemoryPollCache(100)) // TODO: Max number of polls held in cache, make this a config item
if e != nil {
return ws(e)
}
c.TopicList, e = c.NewDefaultTopicList(acc)
if e != nil {
return ws(e)
}
c.PasswordResetter, e = c.NewDefaultPasswordResetter(acc)
if e != nil {
return ws(e)
}
c.Analytics = c.NewDefaultAnalytics()
c.Activity, e = c.NewDefaultActivityStream(acc)
if e != nil {
return ws(e)
}
c.ActivityMatches, e = c.NewDefaultActivityStreamMatches(acc)
if e != nil {
return ws(e)
}
// TODO: Let the admin choose other thumbnailers, maybe ones defined in plugins
c.Thumbnailer = c.NewCaireThumbnailer()
c.Recalc, e = c.NewDefaultRecalc(acc)
if e != nil {
return ws(e)
}
log.Print("Initialising the meta store")
c.Meta, e = meta.NewDefaultMetaStore(acc)
if e != nil {
return ws(e)
}
log.Print("Initialising the view counters")
if !c.Config.DisableAnalytics {
co.GlobalViewCounter, e = co.NewGlobalViewCounter(acc)
if e != nil {
return ws(e)
}
co.AgentViewCounter, e = co.NewDefaultAgentViewCounter(acc)
if e != nil {
return ws(e)
}
co.OSViewCounter, e = co.NewDefaultOSViewCounter(acc)
if e != nil {
return ws(e)
}
co.LangViewCounter, e = co.NewDefaultLangViewCounter(acc)
if e != nil {
return ws(e)
}
if !c.Config.RefNoTrack {
co.ReferrerTracker, e = co.NewDefaultReferrerTracker()
if e != nil {
return ws(e)
}
}
co.MemoryCounter, e = co.NewMemoryCounter(acc)
if e != nil {
return ws(e)
}
co.PerfCounter, e = co.NewDefaultPerfCounter(acc)
if e != nil {
return ws(e)
}
}
co.RouteViewCounter, e = co.NewDefaultRouteViewCounter(acc)
if e != nil {
return ws(e)
}
co.PostCounter, e = co.NewPostCounter()
if e != nil {
return ws(e)
}
co.TopicCounter, e = co.NewTopicCounter()
if e != nil {
return ws(e)
}
co.TopicViewCounter, e = co.NewDefaultTopicViewCounter()
if e != nil {
return ws(e)
}
co.ForumViewCounter, e = co.NewDefaultForumViewCounter()
if e != nil {
return ws(e)
}
return nil
}
// TODO: Split this function up
func main() {
// TODO: Recover from panics
defer func() {
if r := recover(); r != nil {
log.Print(r)
debug.PrintStack()
log.Fatal("Fatal error.")
}
}()
c.StartTime = time.Now()
// TODO: Have a file for each run with the time/date the server started as the file name?
// TODO: Log panics with recover()
f, err := os.OpenFile("./logs/ops-"+strconv.FormatInt(c.StartTime.Unix(), 10)+".log", os.O_WRONLY|os.O_APPEND|os.O_CREATE, 0755)
if err != nil {
log.Fatal(err)
}
//c.LogWriter = io.MultiWriter(os.Stderr, f)
c.LogWriter = io.MultiWriter(os.Stdout, f)
c.ErrLogWriter = io.MultiWriter(os.Stderr, f)
log.SetOutput(c.LogWriter)
c.ErrLogger = log.New(c.ErrLogWriter, "", log.LstdFlags)
log.Print("Running Gosora v" + c.SoftwareVersion.String())
fmt.Println("")
// TODO: Add a flag for enabling the profiler
if false {
f, err := os.Create(c.Config.LogDir + "cpu.prof")
if err != nil {
log.Fatal(err)
}
pprof.StartCPUProfile(f)
}
err = mime.AddExtensionType(".avif", "image/avif")
if err != nil {
log.Fatal(err)
}
jsToken, err := c.GenerateSafeString(80)
if err != nil {
log.Fatal(err)
}
c.JSTokenBox.Store(jsToken)
log.Print("Loading the configuration data")
err = c.LoadConfig()
if err != nil {
log.Fatal(err)
}
log.Print("Processing configuration data")
err = c.ProcessConfig()
if err != nil {
log.Fatal(err)
}
if c.Config.DisableStdout {
c.LogWriter = f
log.SetOutput(c.LogWriter)
}
if c.Config.DisableStderr {
c.ErrLogWriter = f
c.ErrLogger = log.New(c.ErrLogWriter, "", log.LstdFlags)
}
c.Tasks = c.NewScheduledTasks()
err = c.InitTemplates()
if err != nil {
log.Fatal(err)
}
c.Themes, err = c.NewThemeList()
if err != nil {
log.Fatal(err)
}
c.TopicListThaw = c.NewSingleServerThaw()
err = InitDatabase()
if err != nil {
log.Fatal(err)
}
defer db.Close()
buildTemplates := flag.Bool("build-templates", false, "build the templates")
flag.Parse()
if *buildTemplates {
if err = c.CompileTemplates(); err != nil {
log.Fatal(err)
}
if err = c.CompileJSTemplates(); err != nil {
log.Fatal(err)
}
return
}
err = afterDBInit()
if err != nil {
log.Fatalf("%+v", err)
}
err = c.VerifyConfig()
if err != nil {
log.Fatal(err)
}
if !c.Dev.NoFsnotify {
log.Print("Initialising the file watcher")
watcher, err := fsnotify.NewWatcher()
if err != nil {
log.Fatal(err)
}
defer watcher.Close()
go func() {
defer c.EatPanics()
var ErrFileSkip = errors.New("skip mod file")
modifiedFileEvent := func(path string) error {
pathBits := strings.Split(path, "\\")
if len(pathBits) == 0 {
return nil
}
if pathBits[0] == "themes" {
var themeName string
if len(pathBits) >= 2 {
themeName = pathBits[1]
}
if len(pathBits) >= 3 && pathBits[2] == "public" {
// TODO: Handle new themes freshly plopped into the folder?
theme, ok := c.Themes[themeName]
if ok {
return theme.LoadStaticFiles()
}
}
}
return ErrFileSkip
}
// TODO: Expand this to more types of files
var err error
for {
select {
case ev := <-watcher.Events:
// TODO: Handle file deletes (and renames more graciously by removing the old version of it)
if ev.Op&fsnotify.Write == fsnotify.Write {
err = modifiedFileEvent(ev.Name)
if err != ErrFileSkip {
log.Println("modified file:", ev.Name)
} else {
err = nil
}
} else if ev.Op&fsnotify.Create == fsnotify.Create {
log.Println("new file:", ev.Name)
err = modifiedFileEvent(ev.Name)
} else {
log.Println("unknown event:", ev)
err = nil
}
if err != nil {
c.LogError(err)
}
case err = <-watcher.Errors:
c.LogWarning(err)
}
}
}()
// TODO: Keep tabs on the (non-resource) theme stuff, and the langpacks
err = watcher.Add("./public")
if err != nil {
log.Fatal(err)
}
err = watcher.Add("./templates")
if err != nil {
log.Fatal(err)
}
for _, theme := range c.Themes {
err = watcher.Add("./themes/" + theme.Name + "/public")
if err != nil {
log.Fatal(err)
}
}
}
/*if err = c.StaticFiles.GenJS(); err != nil {
c.LogError(err)
}*/
log.Print("Checking for init tasks")
if err = sched(); err != nil {
c.LogError(err)
}
log.Print("Initialising the task system")
// Thumbnailer goroutine, we only want one image being thumbnailed at a time, otherwise they might wind up consuming all the CPU time and leave no resources left to service the actual requests
// TODO: Could we expand this to attachments and other things too?
thumbChan := make(chan bool)
go c.ThumbTask(thumbChan)
if err = tickLoop(thumbChan); err != nil {
c.LogError(err)
}
go TickLoop.Loop()
// Resource Management Goroutine
go func() {
defer c.EatPanics()
uc, tc := c.Users.GetCache(), c.Topics.GetCache()
if uc == nil && tc == nil {
return
}
var lastEvictedCount int
var couldNotDealloc bool
secondTicker := time.NewTicker(time.Second)
for {
select {
case <-secondTicker.C:
// TODO: Add a LastRequested field to cached User structs to avoid evicting the same things which wind up getting loaded again anyway?
if uc != nil {
ucap := uc.GetCapacity()
if uc.Length() <= ucap || c.Users.Count() <= ucap {
couldNotDealloc = false
continue
}
lastEvictedCount = uc.DeallocOverflow(couldNotDealloc)
couldNotDealloc = (lastEvictedCount == 0)
}
}
}
}()
log.Print("Initialising the router")
router, err = NewGenRouter(&RouterConfig{
Uploads: http.FileServer(http.Dir("./uploads")),
})
if err != nil {
log.Fatal(err)
}
log.Print("Initialising the plugins")
c.InitPlugins()
log.Print("Setting up the signal handler")
sigs := make(chan os.Signal, 1)
signal.Notify(sigs, syscall.SIGINT, syscall.SIGTERM)
go func() {
defer c.EatPanics()
sig := <-sigs
log.Print("Received a signal to shutdown: ", sig)
// TODO: Gracefully shutdown the HTTP server
if e := runHook("before_shutdown_tick"); e != nil {
log.Print("before_shutdown_tick err:", e)
}
log.Print("Running shutdown tasks")
c.Tasks.Shutdown.Run()
log.Print("Ran shutdown tasks")
if e := runHook("after_shutdown_tick"); e != nil {
log.Print("after_shutdown_tick err:", e)
}
log.Print("Stopping server")
c.StoppedServer("Stopped server")
}()
// Start up the WebSocket ticks
c.WsHub.Start()
if false {
f, e := os.Create(c.Config.LogDir + "cpu.prof")
if e != nil {
log.Fatal(e)
}
pprof.StartCPUProfile(f)
}
//if profiling {
// pprof.StopCPUProfile()
//}
startServer()
args := <-c.StopServerChan
if false {
pprof.StopCPUProfile()
f, err := os.Create(c.Config.LogDir + "mem.prof")
if err != nil {
log.Fatal(err)
}
defer f.Close()
runtime.GC()
err = pprof.WriteHeapProfile(f)
if err != nil {
log.Fatal(err)
}
}
// Why did the server stop?
log.Fatal(args...)
}
func startServer() {
// We might not need the timeouts, if we're behind a reverse-proxy like Nginx
newServer := func(addr string, h http.Handler) *http.Server {
rtime := c.Config.ReadTimeout
if rtime == 0 {
rtime = 8
} else if rtime == -1 {
rtime = 0
}
wtime := c.Config.WriteTimeout
if wtime == 0 {
wtime = 10
} else if wtime == -1 {
wtime = 0
}
itime := c.Config.IdleTimeout
if itime == 0 {
itime = 120
} else if itime == -1 {
itime = 0
}
return &http.Server{
Addr: addr,
Handler: h,
ConnState: c.ConnWatch.StateChange,
ReadTimeout: time.Duration(rtime) * time.Second,
WriteTimeout: time.Duration(wtime) * time.Second,
IdleTimeout: time.Duration(itime) * time.Second,
TLSConfig: &tls.Config{
PreferServerCipherSuites: true,
CurvePreferences: []tls.CurveID{
tls.CurveP256,
tls.X25519,
},
},
}
}
// TODO: Let users run *both* HTTP and HTTPS
log.Print("Initialising the HTTP server")
/*if c.Dev.QuicPort != 0 {
sQuicPort := strconv.Itoa(c.Dev.QuicPort)
log.Print("Listening on quic port " + sQuicPort)
go func() {
defer c.EatPanics()
c.StoppedServer(http3.ListenAndServeQUIC(":"+sQuicPort, c.Config.SslFullchain, c.Config.SslPrivkey, router))
}()
}*/
if !c.Site.EnableSsl {
if c.Site.Port == "" {
c.Site.Port = "80"
}
log.Print("Listening on port " + c.Site.Port)
go func() {
defer c.EatPanics()
c.StoppedServer(newServer(":"+c.Site.Port, router).ListenAndServe())
}()
return
}
if c.Site.Port == "" {
c.Site.Port = "443"
}
if c.Site.Port == "80" || c.Site.Port == "443" {
// We should also run the server on port 80
// TODO: Redirect to port 443
go func() {
defer c.EatPanics()
log.Print("Listening on port 80")
c.StoppedServer(newServer(":80", &HTTPSRedirect{}).ListenAndServe())
}()
}
log.Printf("Listening on port %s", c.Site.Port)
go func() {
defer c.EatPanics()
c.StoppedServer(newServer(":"+c.Site.Port, router).ListenAndServeTLS(c.Config.SslFullchain, c.Config.SslPrivkey))
}()
}