aboutsummaryrefslogtreecommitdiff
path: root/hswc/main.go
blob: 99aa3c69c52cce5c3a041da2f619d054cd686803 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
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))
}