summaryrefslogtreecommitdiff
path: root/main.go
blob: e06c63fdea9c8c836e86c5307fd93ba713824b02 (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
package main

import (
	"delayed.link/pages"
	_ "delayed.link/pages"
	"delayed.link/storage"
	"delayed.link/storage/sqlite"
	"github.com/spf13/pflag"
	"log"
	"net"
	"net/http"
	"os"
	"os/signal"
	"syscall"
)

var (
	port = pflag.StringP("port", "p", "8880", "port to listen on")
	host = pflag.StringP("host", "h", "0.0.0.0", "host to listen on")
)

func main() {
	pflag.Parse()

	pages.SetupLocal()

	http.HandleFunc("GET /{$}", pages.Land)
	http.HandleFunc("POST /{$}", pages.Create)
	http.HandleFunc("GET /l/{id}", pages.Get)

	storage.Current = sqlite.New()
	defer storage.Current.Close()
	err := http.ListenAndServe(net.JoinHostPort(*host, *port), http.DefaultServeMux)
	log.Printf("Started listening on %v:%v\n", *host, *port)
	ch := make(chan os.Signal, 1)
	signal.Notify(ch, syscall.SIGINT, syscall.SIGTERM)
	<-ch
	if err != nil {
		log.Panic(err)
	}

}