fyp/main.go

package main

import (
	"database/sql"
	"fmt"

	_ "github.com/lib/pq"

	. "git.andr3h3nriqu3s.com/andr3/fyp/logic/models"
	models_utils "git.andr3h3nriqu3s.com/andr3/fyp/logic/models/utils"
	. "git.andr3h3nriqu3s.com/andr3/fyp/logic/utils"
)

const (
	host     = "localhost"
	port     = 5432
	user     = "postgres"
	password = "verysafepassword"
	dbname   = "aistuff"
)

func main() {
	psqlInfo := fmt.Sprintf("host=%s port=%d user=%s "+
		"password=%s dbname=%s sslmode=disable",
		host, port, user, password, dbname)

	db, err := sql.Open("postgres", psqlInfo)
	if err != nil {
		panic(err)
	}
	defer db.Close()
	fmt.Println("Starting server on :8000!")

    //TODO check if file structure exists to save data
	handle := NewHandler(db)

    _, err = db.Exec("update models set status=$1 where status=$2", models_utils.FAILED_TRAINING, models_utils.TRAINING);
    if err != nil {
        panic(err)
    }


	// TODO Handle this in other way
    handle.StaticFiles("/styles/", ".css", "text/css");
    handle.StaticFiles("/js/", ".js", "text/javascript");
    handle.ReadFiles("/imgs/", "views", ".png", "image/png;");
    handle.ReadFiles("/savedData/", ".", ".png", "image/png;");

	handle.GetHTML("/", AnswerTemplate("index.html", nil, 0))

    usersEndpints(db, handle)
    HandleModels(handle)

	handle.Startup()
}