feat: implement ufile sharing service

2026-09-09 20:53:39 -07:00
commit d3016b94db
19 changed files with 4022 additions and 0 deletions
@@ -0,0 +1,57 @@
+package main
+
+import (
+	"context"
+	"embed"
+	"flag"
+	"fmt"
+	"io/fs"
+	"log"
+	"net/http"
+	"os"
+	"os/signal"
+	"path/filepath"
+	"syscall"
+	"time"
+)
+
+//go:embed all:frontend/dist
+var frontendAssets embed.FS
+
+func main() {
+	dataPath := flag.String("data", "data", "文件保存目录")
+	dbPath := flag.String("db", "", "SQLite 数据库路径（默认 <data>/data.db）")
+	address := flag.String("address", "0.0.0.0:6616", "监听地址")
+	flag.Parse()
+	if *dbPath == "" {
+		*dbPath = filepath.Join(*dataPath, "data.db")
+	}
+	if err := os.MkdirAll(filepath.Dir(*dbPath), 0o750); err != nil {
+		log.Fatal(err)
+	}
+	web, err := fs.Sub(frontendAssets, "frontend/dist")
+	if err != nil {
+		log.Fatal(err)
+	}
+	app, initialPassword, err := newApp(*dbPath, *dataPath, web)
+	if err != nil {
+		log.Fatal(err)
+	}
+	if initialPassword != "" {
+		fmt.Printf("\n首次启动管理员账号：admin\n随机密码：%s\n登录后必须立即修改密码。\n\n", initialPassword)
+	}
+	ctx, stop := signal.NotifyContext(context.Background(), syscall.SIGINT, syscall.SIGTERM)
+	defer stop()
+	go app.runCleanup(ctx)
+	server := &http.Server{Addr: *address, Handler: app.router(), ReadHeaderTimeout: 10 * time.Second}
+	go func() {
+		<-ctx.Done()
+		shutdownCtx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
+		defer cancel()
+		_ = server.Shutdown(shutdownCtx)
+	}()
+	log.Printf("ufile listening on %s", *address)
+	if err := server.ListenAndServe(); err != nil && err != http.ErrServerClosed {
+		log.Fatal(err)
+	}
+}