main.go

package main

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

	"github.com/devOpifex/skeef/app"
	"github.com/devOpifex/skeef/db"
	"github.com/golangcollege/sessions"

	// sqlite3 driver
	_ "github.com/mattn/go-sqlite3"
)

var session *sessions.Session
var secret = []byte("u46IpCV8y5Vlur8YvODJEhgOY8m9JVE5")

func main() {

	reset := flag.Bool("reset", false, "Reset the application and redo the first time setup.")
	addr := flag.String("addr", ":8080", "Address on which to serve the skeef.")
	flag.Parse()

	if *reset {
		db.Remove()
		return
	}

	infoLog := log.New(os.Stdout, "INFO\t", log.Ldate|log.Ltime)
	errorLog := log.New(os.Stderr, "ERROR\t", log.Ldate|log.Ltime)

	session = sessions.New(secret)
	session.Lifetime = 12 * time.Hour
	session.SameSite = http.SameSiteStrictMode

	pool := app.NewPool()

	app := &app.Application{
		InfoLog:  infoLog,
		ErrorLog: errorLog,
		Session:  session,
		Addr:     *addr,
		Pool:     pool,
	}

	app.Quit = make(chan struct{})

	// websocket pool
	go app.StartPool()

	firstrun := false
	if !db.Exists() {
		firstrun = true
		err := db.Create()
		fmt.Println("Visit the homepage to setup skeef!")

		if err != nil {
			errorLog.Fatal("Could not create database")
			return
		}

	}

	// connect
	app.Database.Con = db.Connect()
	defer app.Database.Con.Close()

	if firstrun {
		err := app.Database.CreateTableUser()
		if err != nil {
			db.Remove()
			errorLog.Fatal("Could not create users table")
			return
		}

		err = app.Database.CreateTableTwitterApp()
		if err != nil {
			db.Remove()
			errorLog.Fatal("Could not create twitter app table")
			return
		}

		err = app.Database.CreateTableStreams()
		if err != nil {
			db.Remove()
			errorLog.Fatal("Could not create streams table")
			return
		}
	}

	srv := &http.Server{
		Addr:     *addr,
		ErrorLog: errorLog,
		Handler:  app.Handlers(),
	}

	infoLog.Printf("Listening on port%s", *addr)
	err := srv.ListenAndServe()
	log.Fatal(err)
}