yw1573
04e1045c4f
Docker 容器化部署是长期运行的服务，移除关闭服务按钮： - 删除前端 header.ts 中的关闭按钮 - 删除 main.go 中的 /api/shutdown 接口 - 删除 router.go 中的 quit 路由和处理函数 Co-Authored-By: AI
137 lines
3.7 KiB
Go
137 lines
3.7 KiB
Go
package main
|
|
|
|
import (
|
|
"database/sql"
|
|
"embed"
|
|
"fmt"
|
|
"io/fs"
|
|
"log"
|
|
"net/http"
|
|
|
|
"bilidown/internal/router"
|
|
"bilidown/internal/util"
|
|
|
|
_ "modernc.org/sqlite"
|
|
)
|
|
|
|
//go:embed all:static
|
|
var staticFiles embed.FS
|
|
|
|
const (
|
|
HTTP_PORT = 8098 // HTTP 服务器端口
|
|
HTTP_HOST = "" // HTTP 服务器主机
|
|
VERSION = "v2.1.1" // 软件版本号
|
|
)
|
|
|
|
var urlLocal = fmt.Sprintf("http://127.0.0.1:%d", HTTP_PORT)
|
|
var ffmpegAvailable bool
|
|
|
|
func main() {
|
|
ffmpegAvailable = checkFFmpeg()
|
|
mustInitTables()
|
|
mustRunServer()
|
|
fmt.Printf("Bilidown %s server running at %s\n", VERSION, urlLocal)
|
|
if !ffmpegAvailable {
|
|
fmt.Println("Warning: FFmpeg is not installed. Video download will not work.")
|
|
}
|
|
select {} // 保持运行
|
|
}
|
|
|
|
// checkFFmpeg 检测 ffmpeg 的安装情况，返回是否可用
|
|
func checkFFmpeg() bool {
|
|
_, err := util.GetFFmpegPath()
|
|
return err == nil
|
|
}
|
|
|
|
// mustRunServer 配置和启动 HTTP 服务器
|
|
func mustRunServer() {
|
|
// 从嵌入的文件系统中获取静态文件
|
|
staticFS, err := fs.Sub(staticFiles, "static")
|
|
if err != nil {
|
|
log.Fatal("Failed to load static files:", err)
|
|
}
|
|
// 前端静态文件
|
|
http.Handle("/", http.FileServer(http.FS(staticFS)))
|
|
// 后端 API 接口
|
|
apiRouter := router.API()
|
|
// 添加 FFmpeg 状态检查接口
|
|
http.HandleFunc("/api/checkFFmpeg", func(w http.ResponseWriter, r *http.Request) {
|
|
util.Res{Success: true, Data: map[string]bool{"available": ffmpegAvailable}}.Write(w)
|
|
})
|
|
http.Handle("/api/", http.StripPrefix("/api", apiRouter))
|
|
// 启动 HTTP 服务器
|
|
go func() {
|
|
err := http.ListenAndServe(fmt.Sprintf("%s:%d", HTTP_HOST, HTTP_PORT), nil)
|
|
if err != nil {
|
|
log.Fatal("http.ListenAndServe:", err)
|
|
}
|
|
}()
|
|
}
|
|
|
|
// mustInitTables 初始化数据表
|
|
func mustInitTables() {
|
|
db := util.MustGetDB()
|
|
defer db.Close()
|
|
|
|
if _, err := db.Exec(`CREATE TABLE IF NOT EXISTS "field" (
|
|
"name" TEXT PRIMARY KEY NOT NULL,
|
|
"value" TEXT
|
|
)`); err != nil {
|
|
log.Fatalln("create table field:", err)
|
|
}
|
|
|
|
if _, err := db.Exec(`CREATE TABLE IF NOT EXISTS "log" (
|
|
"id" integer NOT NULL PRIMARY KEY AUTOINCREMENT,
|
|
"content" TEXT NOT NULL,
|
|
"create_at" text NOT NULL DEFAULT CURRENT_TIMESTAMP
|
|
)`); err != nil {
|
|
log.Fatalln("create table log:", err)
|
|
}
|
|
|
|
if _, err := db.Exec(`CREATE TABLE IF NOT EXISTS "task" (
|
|
"id" integer NOT NULL PRIMARY KEY AUTOINCREMENT,
|
|
"bvid" text NOT NULL,
|
|
"cid" integer NOT NULL,
|
|
"format" integer NOT NULL,
|
|
"title" text NOT NULL,
|
|
"owner" text NOT NULL,
|
|
"cover" text NOT NULL,
|
|
"status" text NOT NULL,
|
|
"folder" text NOT NULL,
|
|
"duration" integer NOT NULL,
|
|
"download_type" text NOT NULL DEFAULT 'merge',
|
|
"create_at" text NOT NULL DEFAULT CURRENT_TIMESTAMP
|
|
)`); err != nil {
|
|
log.Fatalln("create table task:", err)
|
|
}
|
|
|
|
if _, err := util.GetCurrentFolder(db); err != nil {
|
|
log.Fatalln("util.GetCurrentFolder:", err)
|
|
}
|
|
|
|
if err := initHistoryTask(db); err != nil {
|
|
log.Fatalln("initHistoryTask:", err)
|
|
}
|
|
|
|
// 添加可能缺失的列（用于数据库迁移）
|
|
if err := addMissingColumns(db); err != nil {
|
|
log.Fatalln("addMissingColumns:", err)
|
|
}
|
|
}
|
|
|
|
// addMissingColumns 添加可能缺失的列（用于数据库迁移）
|
|
func addMissingColumns(db *sql.DB) error {
|
|
util.SqliteLock.Lock()
|
|
_, _ = db.Exec(`ALTER TABLE "task" ADD COLUMN "download_type" TEXT DEFAULT 'merge'`)
|
|
_, _ = db.Exec(`UPDATE "task" SET "download_type" = 'merge' WHERE "download_type" IS NULL`)
|
|
util.SqliteLock.Unlock()
|
|
return nil
|
|
}
|
|
|
|
// initHistoryTask 将上一次程序运行时未完成的任务状态变为 error
|
|
func initHistoryTask(db *sql.DB) error {
|
|
util.SqliteLock.Lock()
|
|
_, err := db.Exec(`UPDATE "task" SET "status" = 'error' WHERE "status" IN ('waiting', 'running')`)
|
|
util.SqliteLock.Unlock()
|
|
return err
|
|
} |