package main

import (
	"fmt"
	"io"
	"log"
	DB "log101/konulu-konum-backend/db"
	"log101/konulu-konum-backend/models"
	"strings"

	"os"

	"github.com/dchest/uniuri"
	"github.com/gofiber/fiber/v2"
	"github.com/gofiber/fiber/v2/middleware/cors"
	"github.com/gofiber/fiber/v2/middleware/logger"
	"github.com/h2non/bimg"
	"github.com/joho/godotenv"
	"gorm.io/gorm"
)

func main() {
	// Load environment variables
	err := godotenv.Load()
	if err != nil {
		log.Println("Error loading .env file")
	}

	// initialize db
	DB.InitDB()
	db := DB.GetDB()

	app := fiber.New()
	app.Use(logger.New())
	app.Use(cors.New(cors.Config{
		AllowOrigins: "http://localhost:4321",
	}))
	app.Static("/images", "./public")

	app.Get("/", func(c *fiber.Ctx) error {
		return c.SendString("Sample")
	})

	app.Post("/api/location", func(c *fiber.Ctx) error {
		if form, err := c.MultipartForm(); err == nil {
			// Get form values
			author := form.Value["author"][0]
			description := form.Value["description"][0]
			geolocation := fmt.Sprintf("[%s]", form.Value["geolocation"][0])

			file := form.File["selected-photo"][0]
			fmt.Println(file.Filename, file.Size, file.Header["Content-Type"][0])

			newFile, err := file.Open()
			if err != nil {
				fmt.Fprintln(os.Stderr, err)
			}
			defer newFile.Close()

			data, err := io.ReadAll(newFile)
			if err != nil {
				fmt.Fprintln(os.Stderr, err)
			}

			// Compress image
			newImage, err := bimg.NewImage(data).Convert(bimg.WEBP)
			if err != nil {
				fmt.Fprintln(os.Stderr, err)
			}

			imageName := strings.Split(file.Filename, ".")[0]
			imagePath := fmt.Sprintf("./public/%s.webp", imageName)
			imageURL := fmt.Sprintf("%s.webp", imageName)
			bimg.Write(imagePath, newImage)

			// Generate public uri for the image
			chars := uniuri.StdChars[26:52]
			randomUri := uniuri.NewLenChars(10, chars)
			imageUri := fmt.Sprintf("%s-%s-%s", randomUri[0:3], randomUri[3:7], randomUri[7:])

			db.Create(&models.KonuluKonum{URI: imageUri, ImageURL: imageURL, Coordinates: geolocation, AuthorName: author, Description: description, UnlockedCounter: 0})

			return c.JSON(fiber.Map{
				"url": imageUri,
			})
		}

		return c.SendStatus(fiber.StatusBadRequest)
	})

	app.Get("/api/location/:locationUri", func(c *fiber.Ctx) error {
		uri := c.Params("locationUri")
		if len(uri) == 0 {
			return c.SendStatus(fiber.StatusBadRequest)
		}

		var konuluKonum models.KonuluKonum
		rows := db.Where("URI = ?", uri).First(&konuluKonum)
		if rows.Error != nil {
			return c.SendStatus(fiber.StatusInternalServerError)
		}

		return c.JSON(fiber.Map{
			"url":              konuluKonum.URI,
			"blob_url":         konuluKonum.ImageURL,
			"loc":              konuluKonum.Coordinates,
			"author":           konuluKonum.AuthorName,
			"description":      konuluKonum.Description,
			"unlocked_counter": konuluKonum.UnlockedCounter,
		})
	})

	app.Patch("/api/location/increment/:locationUri", func(c *fiber.Ctx) error {
		uri := c.Params("locationUri")
		if len(uri) == 0 {
			return c.SendStatus(fiber.StatusBadRequest)
		}

		var konuluKonum models.KonuluKonum
		rows := db.Where("URI = ?", uri).First(&konuluKonum)
		if rows.Error != nil {
			return c.SendStatus(fiber.StatusNotFound)
		}

		rows = db.Model(&konuluKonum).Where("uri = ?", uri).UpdateColumn("unlocked_counter", gorm.Expr("unlocked_counter + 1"))
		if rows.Error != nil {
			c.SendStatus(fiber.StatusInternalServerError)
		}

		return c.JSON(fiber.Map{
			"counter": konuluKonum.ID,
		})
	})

	app.Listen(":3000")

}