package main

import (
	"net/http"

	"git.postblue.info/chris/alaska/models"
	"github.com/gin-gonic/contrib/static"
	"github.com/gin-gonic/gin"
	"github.com/jinzhu/gorm"
	_ "github.com/jinzhu/gorm/dialects/sqlite"
	"github.com/prometheus/common/log"
)

func main() {
	// database
	db, err := gorm.Open("sqlite3", "certs.db")
	if err != nil {
		log.Fatal(err)
	}
	db.AutoMigrate(&models.Certificate{})
	defer db.Close()

	// router
	router := gin.Default()
	router.Use(static.Serve("/", static.LocalFile("./views", true)))
	api := router.Group("/api")
	{
		api.GET("/", func(c *gin.Context) {
			c.JSON(http.StatusOK, gin.H{
				"message": "pong",
			})
		})
	}
	api.GET("/certs", CertHandler)
	api.GET("/certs/:name", GetCert)
	router.Run(":3000")
}

func getCerts() []models.Certificate {
	db, err := gorm.Open("sqlite3", "certs.db")
	if err != nil {
		log.Fatal(err)
	}
	defer db.Close()
	res := []models.Certificate{}
	if err := db.Limit(100).Find(&res).Error; err != nil {
		log.Error(err)
	}
	return res
}

func searchCert(search string) []models.Certificate {
	db, err := gorm.Open("sqlite3", "certs.db")
	if err != nil {
		log.Fatal(err)
	}
	defer db.Close()
	res := []models.Certificate{}
	if err := db.Where(&models.Certificate{CN: search}).Or(&models.Certificate{Fingerprint: search}).Find(&res).Error; err != nil {
		log.Warn(err)
	}
	return res
}

func CertHandler(c *gin.Context) {
	getCerts()
	c.Header("Content-Type", "application/json")
	c.JSON(http.StatusOK, getCerts())
}

func GetCert(c *gin.Context) {
	c.Header("Content-Type", "application/json")
	c.JSON(http.StatusOK, searchCert(c.Param("name")))
}