add support for single file

Author: ttt43ttt

cmd/start.go

@@ -4,6 +4,7 @@ import (
 	"fmt"
 	"net/http"
 	"os"
+	"path/filepath"
 	"strconv"
 
 	"github.com/mdp/qrterminal/v3"
@@ -18,7 +19,7 @@ var showQrCode bool
 func init() {
 	rootCmd.AddCommand(startCmd)
 	startCmd.Flags().StringVarP(&folder, "directory", "d", "", "Path of the web folder")
-	startCmd.Flags().StringVarP(&filePath, "file", "f", "", "Path of a file (for hosting a single file)")
+	startCmd.Flags().StringVarP(&filePath, "file", "f", "", "Path of a file (for sharing a single file)")
 	startCmd.Flags().BoolVar(&showQrCode, "qrcode", false, "Show QR code of server URL")
 	startCmd.Flags().IntVarP(&port, "port", "p", 8080, "Port number")
 }
@@ -30,18 +31,22 @@ var startCmd = &cobra.Command{
 	Args: cobra.MinimumNArgs(0),
 
 	Run: func(cmd *cobra.Command, args []string) {
-
 		if filePath != "" {
+			http.HandleFunc("/", func(res http.ResponseWriter, req *http.Request) {
+				res.Header().Add("Content-Disposition", "attachment;filename=\""+filepath.Base(filePath)+"\"")
+				http.ServeFile(res, req, filePath)
+			})
+		} else {
+			fileHandler := http.FileServer(http.Dir(folder))
+			http.Handle("/", fileHandler)
 		}
 
-		fileHandler := http.FileServer(http.Dir(folder))
-
 		ips, err := externalIPs()
 		if err != nil {
 			panic(err)
 		}
 
-		fmt.Println("Http server is running at: ")
+		fmt.Println("Http server is running at:")
 		fmt.Println("http://localhost:" + strconv.Itoa(port))
 		for _, ip := range ips {
 			url := "http://" + ip + ":" + strconv.Itoa(port)
@@ -51,7 +56,7 @@ var startCmd = &cobra.Command{
 			}
 		}
 
-		err = http.ListenAndServe(fmt.Sprint(":", port), fileHandler)
+		err = http.ListenAndServe(fmt.Sprint(":", port), nil)
 		fmt.Println("bye~")
 		if err != nil {
 			panic(err)