Changeset 71 in code for trunk/cmd

Timestamp:

Apr 17, 2025, 6:18:24 PM (7 weeks ago)

Author:

yakumo.izuru

Message:

今では重荷がなくなったので...

Location:

trunk/cmd

Files:

• mai/main.go (modified) (7 diffs)
• mai/parseflags.go (modified) (1 diff)
• mai/readconf.go (modified) (1 diff)
• simplytranslate (deleted)

trunk/cmd/mai/main.go

@@ -13 +13 @@
         "syscall"
 
-        "marisa.chaotic.ninja/mai"
-        "marisa.chaotic.ninja/mai/engines"
+        "mahou-no-mori.yakumo.dev/mai"
+        "mahou-no-mori.yakumo.dev/mai/engines"
 
         "github.com/gofiber/fiber/v2"
@@ -25 +25 @@
 var (
         configfile string
-        groupname string
-        username string
+        verbose bool
 )
 var conf struct {
-        danmaku int
+        group string
         listen string
-        staticpath string
-        tmplpath string
+        requests int
+        static string
+        templates string
+        user string
 }
 func MaiSkipLimiter(c *fiber.Ctx) bool {
         // Paths listed here are not considered for rate limiting
         path := c.Path()
-        return strings.HasPrefix(path, "/static") ||
-                strings.HasPrefix(path, "/docs")
+        return strings.HasPrefix(path, "/static")
 }
 func main() {
         parseFlags()
 
+        conf.listen = "127.0.0.1:5000"
+        conf.requests = 5
+        conf.static = "./static"
+        conf.templates = "./views"
+
         if configfile != "" {
+                if verbose {
+                        log.Printf("Reading configuration from %s", configfile)
+                }
                 readConf(configfile)
         }
 
-        // Default settings
-        conf.danmaku = 10
-        conf.listen = "127.0.0.1:5000"
-        conf.staticpath = "./static"
-        conf.tmplpath = "./views"
-
-        if username != "" {
-                uid, gid, err := usergroupids(username, groupname)
+        if conf.user != "" {
+                uid, gid, err := usergroupids(conf.user, conf.group)
                 if err != nil {
                         log.Fatal(err)
@@ -62 +64 @@
         }
         
-        engine := html.New(conf.tmplpath, ".html")
+        engine := html.New(conf.templates, ".html")
         engine.AddFunc("inc", func(i int) int { return i + 1 })
 
@@ -86 +88 @@
         server.Use(favicon.New(
                 favicon.Config{
-                        File: conf.staticpath + "/favicon.ico",
+                        File: conf.static + "/favicon.ico",
                 },
         ))
@@ -101 +103 @@
         server.Use(limiter.New(limiter.Config{
                 Next: MaiSkipLimiter,
-                Max: conf.danmaku,
+                Max: conf.requests,
                 Expiration: 30 * time.Second,
                 LimiterMiddleware: limiter.SlidingWindow{},
@@ -297 +299 @@
 
         server.Get("/toomanyrequests", func(c *fiber.Ctx) error {
-                return c.SendFile(conf.tmplpath + "/429.html")
+                return c.SendFile(conf.templates + "/429.html")
                 return c.SendStatus(429)
         })
@@ -326 +328 @@
                 return c.Redirect("/")
         })
-        server.Static("/static", conf.staticpath, fiber.Static{
-                Compress: true,
+        server.Static("/static", conf.static, fiber.Static{
                 ByteRange: true,
                 Browse: true,
         })
 
-        server.Static("/docs", "./docs", fiber.Static{})
-
         server.Listen(conf.listen)
 }

trunk/cmd/mai/parseflags.go

@@ -8 +8 @@
 func parseFlags() {
         flag.StringVar(&configfile, "f", "", "Configuration file")
-        flag.StringVar(&username, "u", "", "Sets the user to which privilege dropping is done")
-        flag.StringVar(&groupname, "g", "", "Sets the group to which privilege dropping is done")
+        flag.BoolVar(&verbose, "v", false, "Be verbose")
         flag.Parse()
 }

trunk/cmd/mai/readconf.go

@@ -11 +11 @@
                 return err                                              
         }
-        conf.danmaku, _ = cfg.Section("mai").Key("danmaku").Int()
-        conf.listen = cfg.Section("mai").Key("listen").String()
-        conf.staticpath = cfg.Section("mai").Key("static").String()
-        conf.tmplpath = cfg.Section("mai").Key("templates").String()
 
+        conf.group = cfg.Section("mai").Key("group").String()
+        conf.listen = cfg.Section("http").Key("listen").String()
+        conf.requests, _ = cfg.Section("http").Key("requests").Int()
+        conf.static = cfg.Section("paths").Key("static").String()
+        conf.templates = cfg.Section("paths").Key("templates").String()
+        conf.user = cfg.Section("mai").Key("user").String()
         return nil
 }