pastebin/main.go

package main

import (
	"log"
	"time"

	"github.com/boltdb/bolt"
	"github.com/namsral/flag"
)

var (
	db  *bolt.DB
	cfg Config
)

func main() {
	var (
		config string
		dbpath string
		bind   string
		fqdn   string
		expiry time.Duration
	)

	flag.StringVar(&config, "config", "", "config file")
	flag.StringVar(&dbpath, "dbpth", "urls.db", "Database path")
	flag.StringVar(&bind, "bind", "0.0.0.0:8000", "[int]:<port> to bind to")
	flag.StringVar(&fqdn, "fqdn", "localhost", "FQDN for public access")
	flag.DurationVar(&expiry, "expiry", 5*time.Minute, "expiry time for pastes")
	flag.Parse()

	if expiry.Seconds() < 60 {
		log.Fatalf("expiry of %s is too small", expiry)
	}

	// TODO: Abstract the Config and Handlers better
	cfg.fqdn = fqdn
	cfg.expiry = expiry

	var err error
	db, err = bolt.Open(dbpath, 0600, nil)
	if err != nil {
		log.Fatal(err)
	}
	defer db.Close()

	NewServer(bind, cfg).ListenAndServe()
}