package httpserver import ( "b612.me/starlog" "b612.me/staros" "context" "encoding/json" "github.com/spf13/cobra" "os" "os/signal" "strings" ) var s HttpServer var daemon bool var hooks string func init() { Cmd.Flags().StringVarP(&hooks, "hook", "H", "", "fileget hook for modify") Cmd.Flags().StringVarP(&s.port, "port", "p", "80", "监听端口") Cmd.Flags().StringVarP(&s.addr, "ip", "i", "0.0.0.0", "监听ip") Cmd.Flags().StringVarP(&s.envPath, "folder", "f", "./", "本地文件地址") Cmd.Flags().StringVarP(&s.uploadFolder, "upload", "u", "", "文件上传文件夹路径") Cmd.Flags().BoolVarP(&daemon, "daemon", "d", false, "以后台进程运行") Cmd.Flags().StringVarP(&s.basicAuthUser, "auth", "a", "", "HTTP BASIC AUTH认证(用户名:密码)") Cmd.Flags().StringVarP(&s.indexFile, "index", "n", "", "Index文件名,如index.html") Cmd.Flags().StringVarP(&s.logpath, "log", "l", "", "log地址") Cmd.Flags().StringVarP(&s.cert, "ssl-cert", "c", "", "TLS证书路径") Cmd.Flags().StringVarP(&s.key, "ssl-key", "k", "", "TLS密钥路径") Cmd.Flags().BoolVarP(&s.disableMIME, "disablemime", "m", false, "停止解析MIME,全部按下载文件处理") Cmd.Flags().StringSliceVarP(&s.protectAuthPage, "protect-page", "P", []string{}, "Basic Auth 开启白名单") Cmd.Flags().StringVar(&s.page401, "401", "", "自定义401页面地址") Cmd.Flags().StringVar(&s.page403, "403", "", "自定义403页面地址") Cmd.Flags().StringVar(&s.page404, "404", "", "自定义404页面地址") Cmd.Flags().Bool("daeapplied", false, "") Cmd.Flags().MarkHidden("daeapplied") } var Cmd = &cobra.Command{ Use: "http", Short: "HTTP文件服务器(HTTP File Browser Server)", Long: `HTTP文件服务器(HTTP File Browser Server)`, Run: func(cmd *cobra.Command, args []string) { if hooks != "" { if !staros.Exists(hooks) { starlog.Criticalln("hook file not exists") os.Exit(2) } data, err := os.ReadFile(hooks) if err != nil { starlog.Criticalln("read hook file error", err) os.Exit(3) } var hk []ServerHook err = json.Unmarshal(data, &hk) if err != nil { starlog.Errorln("Unmarshal hook Json Failed", err) os.Exit(4) } s.hooks = hk } apply, _ := cmd.Flags().GetBool("daeapplied") if daemon && !apply { nArgs := append(os.Args[1:], "--daeapplied") pid, err := staros.Daemon(os.Args[0], nArgs...) if err != nil { starlog.Criticalln("Daemon Error:", err) os.Exit(1) } starlog.StdPrintf([]starlog.Attr{starlog.FgGreen}, "Success,PID=%v\n", pid) return } sig := make(chan os.Signal) signal.Notify(sig, os.Kill, os.Interrupt) ctx, fn := context.WithCancel(context.Background()) go func() { starlog.Infoln("Recv Signal", <-sig) fn() }() if s.basicAuthUser != "" { tmp := strings.SplitN(s.basicAuthUser, ":", 2) if len(tmp) != 2 { starlog.Errorln("basic Auth should have user and password") os.Exit(2) } s.basicAuthUser = strings.TrimSpace(tmp[0]) s.basicAuthPwd = tmp[1] } err := s.Run(ctx) if err != nil { starlog.Errorln("Http Server Closed by Errors", err) os.Exit(1) } starlog.Infoln("Http Server Closed Normally") }, }