all repos — fsrv @ 60a9765e785e277e7e697dcc73855da94aab74b7

filehost server for x.icyphox.sh 

Init fsrv
Anirudh Oppiliappan x@icyphox.sh
Sun, 06 Jun 2021 22:33:30 +0530
commit

60a9765e785e277e7e697dcc73855da94aab74b7

4 files changed, 97 insertions(+), 0 deletions(-)

jump to
A .gitignore 
@@ -0,0 +1,2 @@ 
+fsrv
+uploads/
A go.mod 
@@ -0,0 +1,3 @@ 
+module git.icyphox.sh/fsrv
+
+go 1.16
A index.html 
@@ -0,0 +1,19 @@ 
+<!DOCTYPE html>
+<html lang="en">
+  <head>
+    <meta charset="UTF-8" />
+    <meta name="viewport" content="width=device-width, initial-scale=1.0" />
+    <meta http-equiv="X-UA-Compatible" content="ie=edge" />
+    <title>Document</title>
+  </head>
+  <body>
+    <form
+      enctype="multipart/form-data"
+      action="http://localhost:9393/"
+      method="post"
+    >
+      <input type="file" name="file" />
+      <input type="submit" value="upload" />
+    </form>
+  </body>
+</html>
A main.go 
@@ -0,0 +1,73 @@ 
+package main
+
+import (
+	"flag"
+	"fmt"
+	"io"
+	"log"
+	"math/rand"
+	"net/http"
+	"os"
+	"path/filepath"
+	"time"
+)
+
+type settings struct {
+	url     string
+	port    string
+	namelen int
+	key     string
+}
+
+func randName(n int) string {
+	letters := []rune("abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ1234567890~-")
+	b := make([]rune, n)
+	for i := range b {
+		b[i] = letters[rand.Intn(len(letters))]
+	}
+	return string(b)
+}
+
+func (s *settings) uploadFile(w http.ResponseWriter, r *http.Request) {
+	r.ParseMultipartForm(10 << 20)
+	file, handler, err := r.FormFile("file")
+	if err != nil {
+		log.Println(err)
+		return
+	}
+	defer file.Close()
+	log.Printf("file: %+v\t%+v bytes", handler.Filename, handler.Size)
+
+	ext := filepath.Ext(handler.Filename)
+	fileBytes, err := io.ReadAll(file)
+	if err != nil {
+		log.Println(err)
+	}
+
+	newFile := "uploads/" + randName(5) + ext
+	os.WriteFile(newFile, fileBytes, 0644)
+	log.Printf("wrote: %+v", newFile)
+
+	fileUrl := s.url + "/" + newFile
+	fmt.Fprintf(w, "%v", fileUrl)
+}
+
+func (s *settings) readSettings() {
+	flag.StringVar(&s.url, "url", "localhost", "url for fsrv to serve files")
+	flag.StringVar(&s.port, "port", "9393", "port to listen on")
+	flag.IntVar(&s.namelen, "namelen", 5, "length of random filename")
+	flag.StringVar(&s.key, "key", "secret", "secret key; generate this yourself")
+	flag.Parse()
+}
+
+func main() {
+	rand.Seed(time.Now().UnixNano())
+
+	st := settings{}
+	st.readSettings()
+
+	http.HandleFunc("/", st.uploadFile)
+
+	log.Println("listening on " + st.port)
+	http.ListenAndServe(":"+st.port, nil)
+}