package main

import (
	"crypto/sha256"
	"database/sql"
	"fmt"
	"io/ioutil"
	"log"
	"os"
	"os/exec"
	"path/filepath"
	"time"

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

// TODO: change these to be cmd args instead!
const db_file = "/home/lmas/projects/ss13_se/src/db.sqlite3"

// Dir to save new graphs in
const save_dir = "/home/lmas/projects/ss13_se/src/static/graphs"

// How far back in time the graphs will go
var last_week = time.Now().AddDate(0, 0, -7)

func checkerror(err error) {
	if err != nil {
		log.Fatal(err)
	}
}

func main() {
	// open a db connection
	db, err := sql.Open("sqlite3", db_file)
	checkerror(err)
	defer db.Close()

	// loop over each server in db
	rows, err := db.Query("select id, title from gameservers_server")
	checkerror(err)
	defer rows.Close()

	var (
		id    int
		title string
	)
	for rows.Next() {
		err := rows.Scan(&id, &title)
		checkerror(err)
		creategraphs(db, id, title)
	}
	err = rows.Err()
	checkerror(err)
}

func creategraphs(db *sql.DB, id int, title string) {
	// create a tmp file
	ifile, err := ioutil.TempFile("", "graph")
	checkerror(err)
	defer ifile.Close()
	ifilename := ifile.Name()

	// Make sure we have somewhere to save the stored graphs in
	err = os.MkdirAll(save_dir, 0777)
	checkerror(err)
	hash := fmt.Sprintf("%x", sha256.Sum256([]byte(title)))
	ofilename := filepath.Join(save_dir, hash)

	// get the server's data and write it to the file
	rows, err := db.Query("select created,players from gameservers_serverhistory where server_id = ? and created >= ? order by created asc", id, last_week)
	checkerror(err)

	var (
		created time.Time
		players int
	)
	for rows.Next() {
		err := rows.Scan(&created, &players)
		checkerror(err)
		_, err = ifile.WriteString(fmt.Sprintf("%d, %d\n", created.Unix(), players))
		checkerror(err)
	}
	err = rows.Err()
	checkerror(err)

	// run the plotter against the data file
	err = exec.Command("./plotter.sh", ifilename, ofilename).Run()
	checkerror(err)

	// close and remove the tmp file
	ifile.Close()
	os.Remove(ifilename)
}