2019-08-25 21:33:56 +02:00
|
|
|
package main
|
|
|
|
|
|
|
|
import (
|
|
|
|
"flag"
|
|
|
|
"log"
|
|
|
|
"net/http"
|
|
|
|
"time"
|
|
|
|
|
2019-08-29 23:40:24 +02:00
|
|
|
"github.com/gorilla/mux"
|
|
|
|
"github.com/pkg/errors"
|
2019-09-10 17:52:45 +02:00
|
|
|
|
2019-09-15 17:43:09 +02:00
|
|
|
"gitea.hashru.nl/dsprenkels/rushlink/db"
|
|
|
|
"gitea.hashru.nl/dsprenkels/rushlink/handlers"
|
|
|
|
"gitea.hashru.nl/dsprenkels/rushlink/metrics"
|
2019-08-25 21:33:56 +02:00
|
|
|
)
|
|
|
|
|
|
|
|
type ParsedArguments struct {
|
|
|
|
databaseName string
|
|
|
|
}
|
|
|
|
|
|
|
|
var appConfig ParsedArguments
|
|
|
|
|
|
|
|
func main() {
|
|
|
|
// Parse the arguments and construct the ParsedArguments
|
|
|
|
appConfigRef, err := parseArguments()
|
|
|
|
if err != nil {
|
|
|
|
log.Fatal(err)
|
|
|
|
}
|
|
|
|
appConfig = *appConfigRef
|
|
|
|
|
2019-09-10 17:52:45 +02:00
|
|
|
db.Init(appConfig.databaseName)
|
2019-08-25 21:33:56 +02:00
|
|
|
|
2019-08-29 23:40:24 +02:00
|
|
|
// Export prometheus metrics
|
2019-09-10 17:52:45 +02:00
|
|
|
go metrics.StartMetricsServer()
|
2019-08-29 23:40:24 +02:00
|
|
|
|
2019-08-25 21:33:56 +02:00
|
|
|
// Initialize Gorilla router
|
|
|
|
router := mux.NewRouter()
|
2019-09-10 17:52:45 +02:00
|
|
|
router.HandleFunc("/", handlers.IndexGetHandler).Methods("GET")
|
|
|
|
router.HandleFunc("/", handlers.IndexPostHandler).Methods("POST")
|
|
|
|
router.HandleFunc("/{key:[A-Za-z0-9-_]{4,}}", handlers.PasteGetHandler).Methods("GET")
|
|
|
|
router.HandleFunc("/{key:[A-Za-z0-9-_]{4,}}/nr", handlers.PasteGetHandlerNoRedirect).Methods("GET")
|
|
|
|
router.HandleFunc("/{key:[A-Za-z0-9-_]{4,}}/meta", handlers.PasteGetHandlerMeta).Methods("GET")
|
2019-08-25 21:33:56 +02:00
|
|
|
|
|
|
|
// Start the server
|
|
|
|
srv := &http.Server{
|
|
|
|
Handler: router,
|
|
|
|
Addr: "127.0.0.1:8000",
|
|
|
|
WriteTimeout: 15 * time.Second,
|
|
|
|
ReadTimeout: 15 * time.Second,
|
|
|
|
}
|
|
|
|
log.Fatal(srv.ListenAndServe())
|
|
|
|
}
|
|
|
|
|
|
|
|
// Parse the input arguments and return the initialized application config struct
|
|
|
|
func parseArguments() (*ParsedArguments, error) {
|
|
|
|
config := ParsedArguments{}
|
|
|
|
|
|
|
|
flag.StringVar(&config.databaseName, "database", "", "Location of the database file")
|
|
|
|
flag.Parse()
|
|
|
|
|
|
|
|
if config.databaseName == "" {
|
|
|
|
return nil, errors.New("database not set")
|
|
|
|
}
|
|
|
|
return &config, nil
|
|
|
|
}
|