buster/main.go at master · zood/buster · GitHub

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
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
package main

import (
	"flag"
	"fmt"
	"log"
	"net/http"
	"path/filepath"
	"time"

	"github.com/gorilla/mux"
	"zood.xyz/buster/cors"
	"zood.xyz/buster/email"
	"zood.xyz/buster/mailgun"
	"zood.xyz/buster/resources"
)

func main() {
	log.SetFlags(log.Ldate | log.Ltime | log.Lshortfile)

	resPath := flag.String("resources", "", "Path to resources directory")
	mailgunApiKey := flag.String("mailgun-api-key", "", "Mailgun API key")
	mailgunDomain := flag.String("mailgun-domain", "", "Mailgun domain")
	devMode := flag.Bool("dev", false, "true makes the server reload templates on every request. default is false.")
	port := flag.Int("port", 1313, "Port to listen on")
	flag.Parse()

	if *resPath == "" {
		log.Fatal("Resources path is empty")
	}

	if !*devMode {
		if *mailgunApiKey == "" {
			log.Fatal("mailgun api key is missing")
		}
		if *mailgunDomain == "" {
			log.Fatal("mailgun domain is missing")
		}
	}

	rsrcs, err := resources.New(*resPath)
	if err != nil {
		log.Fatalf("Failed to load resources: %v", err)
	}
	rsrcs.DevMode = *devMode

	r := mux.NewRouter().StrictSlash(true)
	r.PathPrefix("/css/").Handler(http.StripPrefix("/css/", http.FileServer(http.Dir(filepath.Join(*resPath, "css")))))
	r.PathPrefix("/images/").Handler(http.StripPrefix("/images/", http.FileServer(http.Dir(filepath.Join(*resPath, "images")))))
	r.PathPrefix("/fonts/").Handler(http.StripPrefix("/fonts/", http.FileServer(http.Dir(filepath.Join(*resPath, "fonts")))))
	r.PathPrefix("/static/").Handler(http.StripPrefix("/static/", http.FileServer(http.Dir(filepath.Join(*resPath, "static")))))

	r.HandleFunc("/", homeHandler).Methods(http.MethodGet)
	r.HandleFunc("/privacy", privacyHandler).Methods(http.MethodGet)
	r.HandleFunc("/privacy/mobile-apps", mobileAppsPrivacyHandler).Methods(http.MethodGet)
	r.HandleFunc("/about", aboutHandler).Methods(http.MethodGet)
	r.HandleFunc("/contact", contactHandler).Methods(http.MethodGet)

	// blog
	r.HandleFunc("/blog", blogHomeHandler).Methods(http.MethodGet)
	r.HandleFunc("/blog/archive", blogArchivesHandler).Methods(http.MethodGet)
	r.HandleFunc("/blog/{id:[0-9]+}", blogPostHandler).Methods(http.MethodGet)
	r.HandleFunc("/blog/{id:[0-9]+}/{slug}", blogPostHandler).Methods(http.MethodGet)

	r.HandleFunc("/verify-email", verifyEmailHandler).Methods(http.MethodGet)
	r.HandleFunc("/disavow-email", disavowEmailHandler).Methods(http.MethodGet)

	r.NotFoundHandler = http.HandlerFunc(notFoundHandler)
	r.MethodNotAllowedHandler = http.HandlerFunc(notFoundHandler)

	var emailer email.SendEmailer
	if *mailgunApiKey == "" || *mailgunDomain == "" {
		emailer = email.NewMock()
	} else {
		emailer = mailgun.New(*mailgunApiKey, *mailgunDomain, false)
	}
	r.Use(busterMiddleware{rsrcs: rsrcs, sendEmailer: emailer}.Middleware, cors.Middleware)

	// playground()
	var hostAddress string
	if *devMode {
		hostAddress = fmt.Sprintf(":%d", *port)
	} else {
		hostAddress = fmt.Sprintf("127.0.0.1:%d", *port)
	}
	server := http.Server{
		Addr:         hostAddress,
		Handler:      r,
		ReadTimeout:  5 * time.Second,
		WriteTimeout: 10 * time.Second,
		IdleTimeout:  120 * time.Second,
	}

	log.Printf("Starting app on port %d…", *port)
	log.Fatal(server.ListenAndServe())
}