advlight.go

package main

import (
	"flag"
	"log"
	"net/http"
	"os"

	"github.com/blit/advlight/config"
	"github.com/blit/advlight/tickets"

	"github.com/blit/advlight/views"
	"github.com/go-chi/chi"
	"github.com/go-chi/chi/middleware"
	_ "github.com/lib/pq" // required for database/sql
)

func main() {
	flag.BoolVar(&tickets.CAPTCHADisabled, "nocaptcha", false, "disabled captcha")
	flag.Parse()
	runServer()
	//addTickets()
}

func addTickets() {
	slots, err := tickets.Repo.GetSlotsStats()
	if err != nil {
		log.Panicln(err)
	}
	for _, s := range slots {
		if s.EventCode != "" {
			continue
		}
		if s.AvailableTickets == 0 {
			tickets.Repo.CreateSlots("grace", int(s.Slot.Unix()), 50)
			log.Println(s.Slot, s.AvailableTickets)
		}
	}
}

func runServer() {
	r := chi.NewRouter()
	// A good base middleware stack
	r.Use(middleware.RequestID)
	r.Use(middleware.RealIP)
	r.Use(middleware.Logger)
	r.Use(middleware.Recoverer)
	r.Get("/favicon.ico", func(w http.ResponseWriter, r *http.Request) {
		return
	})
	r.Get("/", views.TicketIndexHandler)
	r.Post("/", views.TicketIndexHandler)

	r.Get("/admin", views.TicketAdminHandler)
	r.Post("/admin", views.TicketAdminHandler)

	r.Get("/admin/run_expired", views.TicketAdminExpiresHandler)

	r.Get("/{guestID}", views.TicketIndexHandler)
	r.Post("/{guestID}", views.TicketIndexHandler)
	r.Get("/{guestID}/ticket/{ticketID}", views.TicketShowHandler)
	r.Get("/assets/img/{imageID}", views.AssetImageHandler)
	r.Get("/ticketfaces", views.TicketFacesHandler)
	if tickets.DatabaseURL == "" {
		log.Println(os.Getenv("ADVLIGHT_DATABASE_URL"))
		log.Fatal("ADVLIGHT_DATABASE_URL is not set; try export ADVLIGHT_DATABASE_URL=postgres://postgres@localhost/advlight?sslmode=disable")
	}
	log.Println(tickets.HostName, tickets.DatabaseURL, "CAPTCHADisabled:", tickets.CAPTCHADisabled)
	log.Fatalln(http.ListenAndServe(config.Port, r))
}