rushlink/rushlink.go

74 lines
1.8 KiB
Go
Raw Normal View History

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-08-25 21:33:56 +02:00
bolt "go.etcd.io/bbolt"
)
type ParsedArguments struct {
databaseName string
}
var appConfig ParsedArguments
var db *bolt.DB
func main() {
// Parse the arguments and construct the ParsedArguments
appConfigRef, err := parseArguments()
if err != nil {
log.Fatal(err)
}
appConfig = *appConfigRef
// Open the bolt database
db, err = bolt.Open(appConfig.databaseName, 0666, &bolt.Options{Timeout: 1 * time.Second})
if err != nil {
decoratedErr := errors.Wrapf(err, "failed to open database at '%v'", appConfig.databaseName)
log.Fatal(decoratedErr)
}
defer db.Close()
err = db.Update(migrateDatabase)
if err != nil {
decoratedErr := errors.Wrapf(err, "failed to migrate database")
log.Fatal(decoratedErr)
}
2019-08-29 23:40:24 +02:00
// Export prometheus metrics
go startMetricsServer()
2019-08-25 21:33:56 +02:00
// Initialize Gorilla router
router := mux.NewRouter()
router.HandleFunc("/", indexGetHandler).Methods("GET")
router.HandleFunc("/", indexPostHandler).Methods("POST")
router.HandleFunc("/{key:[A-Za-z0-9-_]{4,}}", pasteGetHandler).Methods("GET")
router.HandleFunc("/{key:[A-Za-z0-9-_]{4,}}/nr", pasteGetHandlerNoRedirect).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
}