alaska/server.go

75 lines
1.6 KiB
Go
Raw Normal View History

2020-08-20 13:49:00 +02:00
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")))
}