package main

import (
	"bufio"
	"fmt"
	"log"
	"os"
	"os/signal"
	"path"
	"strconv"
	"strings"
	"syscall"

	"github.com/gin-gonic/gin"
	"github.com/robfig/cron"

	"git.1750studios.com/AniNite/SocialDragon/config"
	"git.1750studios.com/AniNite/SocialDragon/database"
	"git.1750studios.com/AniNite/SocialDragon/folder"
	"git.1750studios.com/AniNite/SocialDragon/instagram"
	"git.1750studios.com/AniNite/SocialDragon/snapchat"
	"git.1750studios.com/AniNite/SocialDragon/twitter"
)

func main() {
	config.LoadConfig(os.Getenv("HOME") + "/.socialdragon.toml")
	database.InitDb(SendUpdate)

	c := cron.New()
	if config.C.Snapchat.UserName != "" {
		c.AddFunc("@every 30s", snapchat.LoadNewSnaps)
	}
	if config.C.Instagram.Tag != "" {
		c.AddFunc("@every 30s", instagram.LoadNewInstas)
	}
	if config.C.Folder.Folder != "" {
		c.AddFunc("@every 30s", folder.LoadNewFolders)
	}
	c.Start()

	if len(config.C.Twitter.Filter) != 0 {
		go twitter.LoadNewTweets()
	}
	go setupGin()

	ch := make(chan os.Signal)
	signal.Notify(ch, syscall.SIGINT, syscall.SIGTERM)

	<-ch

	twitter.Stop()

	exportAndDeleteImages()
}

func setupGin() {
	router := gin.Default()
	router.GET("/", renderHomepage)

	router.GET("/admin", func(c *gin.Context) { c.Redirect(301, "/admin/inbox") })
	router.GET("/admin/inbox", renderAdminInbox)
	router.GET("/admin/approved", renderAdminApproved)
	router.GET("/admin/rejected", renderAdminRejected)

	router.GET("/admin/stats", stats)
	router.POST("/admin/approve/:id", approveSnap)
	router.POST("/admin/reject/:id", rejectSnap)

	router.GET("/ws", wsHandler)
	router.GET("/listSnaps", listSnaps)

	router.LoadHTMLGlob(config.C.TemplatesDirectory + "/*.html")
	router.Static("/static", config.C.AssetsDirectory)
	router.Static(config.C.ContentWebDirectory, config.C.ContentDirectory)
	router.Run(config.C.BindAddress)
}

func exportAndDeleteImages() {
	conf := askForConfirmationNo("Do you want to export and delete the images from the database?")
	if conf {
		var ITs []database.Item
		database.Db.Find(&ITs)
		for _, IT := range ITs {
			old := path.Join(path.Join(config.C.ContentDirectory, ".."), IT.Path)
			new := path.Join(path.Join(config.C.ContentDirectory, ".."), "export")
			state := "unknown"
			if IT.State == database.Approved {
				state = "approved"
			} else if IT.State == database.Rejected {
				state = "rejected"
			} else if IT.State == database.Inbox {
				state = "inbox"
			}
			new = path.Join(new, state)
			os.MkdirAll(new, 0755)
			paths := strings.Split(IT.Path, ".")
			str := strconv.FormatUint(uint64(IT.UserID), 10) + "-" + string(IT.OriginalID) + "." + paths[len(paths)-1]
			new = path.Join(new, str)
			os.Rename(old, new)
			database.Db.Delete(&IT)
		}
		os.RemoveAll(config.C.ContentDirectory)
	}
}

func askForConfirmationNo(s string) bool {
	reader := bufio.NewReader(os.Stdin)
	fmt.Printf("%s [y/N]: ", s)
	response, err := reader.ReadString('\n')
	if err != nil {
		log.Fatal(err)
	}
	response = strings.ToLower(strings.TrimSpace(response))
	if response == "y" || response == "yes" {
		return true
	}
	return false
}