diff options
| author | Přemysl Eric Janouch <p@janouch.name> | 2022-10-06 00:44:09 +0200 | 
|---|---|---|
| committer | Přemysl Eric Janouch <p@janouch.name> | 2022-10-08 17:00:55 +0200 | 
| commit | 5ebc35ad11989328c0c417230805b4ff9ed7a045 (patch) | |
| tree | f448ff5bb4a8c7b7728f46f41e3a300ebbeb564b /hswc/main.go | |
| parent | c3cc608570aa473d8777496e857e69637507da25 (diff) | |
| download | haven-5ebc35ad11989328c0c417230805b4ff9ed7a045.tar.gz haven-5ebc35ad11989328c0c417230805b4ff9ed7a045.tar.xz haven-5ebc35ad11989328c0c417230805b4ff9ed7a045.zip  | |
Add a basic tool for collecting web comments
Diffstat (limited to 'hswc/main.go')
| -rw-r--r-- | hswc/main.go | 68 | 
1 files changed, 68 insertions, 0 deletions
diff --git a/hswc/main.go b/hswc/main.go new file mode 100644 index 0000000..99aa3c6 --- /dev/null +++ b/hswc/main.go @@ -0,0 +1,68 @@ +package main + +import ( +	"fmt" +	"log" +	"net/http" +	"os" +	"regexp" +	"strings" +	"sync" +	"time" +) + +var ( +	long = regexp.MustCompile(`(.{72}\S*)\s+`) +	file *os.File +	m    sync.Mutex +) + +func wrap(s string) string { +	return strings.ReplaceAll(long.ReplaceAllString( +		strings.ReplaceAll(s, "\r", ""), "$1\n"), "\n", "\n ") +} + +func handler(w http.ResponseWriter, r *http.Request) { +	defer r.Body.Close() +	if err := r.ParseForm(); err != nil { +		w.WriteHeader(http.StatusBadRequest) +		return +	} +	text := r.FormValue("text") +	if len(text) > 64<<10 { +		w.WriteHeader(http.StatusBadRequest) +		return +	} + +	m.Lock() +	defer m.Unlock() + +	if s, err := file.Stat(); err != nil { +		log.Fatalln(err) +	} else if s.Size()+int64(len(text)) > 64<<20 { +		w.WriteHeader(http.StatusInternalServerError) +	} else { +		fmt.Fprintf(file, "%s %s\n %s\n", +			time.Now().Local().Format(time.RFC1123), r.RequestURI, wrap(text)) +		if err := file.Sync(); err != nil { +			log.Fatalln(err) +		} + +		fmt.Fprintln(w, "Saved.") +	} +} + +func main() { +	if len(os.Args) != 3 { +		log.Fatalf("Usage: %s BIND DB\n", os.Args[0]) +	} + +	var err error +	if file, err = os.OpenFile( +		os.Args[2], os.O_CREATE|os.O_APPEND|os.O_WRONLY, 0644); err != nil { +		log.Fatalln(err) +	} + +	http.HandleFunc("/", handler) +	log.Fatalln(http.ListenAndServe(os.Args[1], nil)) +}  | 
