package main

import (
	"context"

	"github.com/adrg/xdg"
	"gorm.io/driver/sqlite"
	"gorm.io/gorm"
	"gorm.io/gorm/logger"
	"gorm.io/gorm/clause"
)

type AppConfig struct {
	Title string
	Version string
}

type App struct {
	ctx         context.Context
	Config      AppConfig
	DB          *gorm.DB
	UserContact Contact
}

func NewApp(config AppConfig) *App {
	app := &App{Config: config}
	db, err := gorm.Open(sqlite.Open(app.GetDataFile()), &gorm.Config{
		Logger: logger.Default.LogMode(logger.Silent),
	})
	if err != nil {
		panic(err)
	}
	app.DB = db
	app.migrate()
	return app
}

func (app *App) startup(ctx context.Context) {
	app.Log("Running startup.").
		Stdout()
	defer app.Log("Startup finished.").
		Stdout()
	app.ctx = ctx
	app.EstablishMailConnections()
	return
}

func (app *App) shutdown(ctx context.Context) {
	app.Log("Running shutdown.").
		Stdout()
	defer app.Log("finished shutdown.").
		Stdout()
	CloseMailConnections()
	return
}

func (app *App) migrate() {
	app.Log("Starting migration...").
		Stdout()
	defer app.Log("finished migration.").
		Stdout()
	app.DB.AutoMigrate(&Contact{})
	app.DB.AutoMigrate(&EmailAddress{})
	app.DB.AutoMigrate(&PhoneNumber{})
	app.DB.AutoMigrate(&URL{})
	app.DB.AutoMigrate(&StreetAddress{})
	app.DB.AutoMigrate(&Date{})
	app.DB.AutoMigrate(&OtherField{})
	app.DB.Preload(clause.Associations).
		FirstOrCreate(
			&app.UserContact,
			Contact{Model: gorm.Model{ID: 1}},
		)
	app.Log("Ensuring contact.").
		Stdout()
	app.DB.AutoMigrate(&MailAccountConnection{})
	app.DB.AutoMigrate(&Conversation{})
	app.DB.AutoMigrate(&Message{})
	app.DB.AutoMigrate(&MailHeader{})
	app.DB.AutoMigrate(&LastSynced{})
	app.DB.AutoMigrate(&MessageContent{})
	app.DB.AutoMigrate(&DisplayImage{})
	return
}

func (app *App) GetTitle() string {
	return app.Config.Title
}

func (app *App) GetVersion() string {
	return app.Config.Version
}

func (app *App) GetDataFile() string {
	if file, err := xdg.DataFile(app.GetTitle() + "/data.db"); err != nil {
		panic(err)
	} else {
		return file
	}
}

func (app *App) GetMailPasswordKeyringService() string {
	title := app.GetTitle()
	return title + "-mail-password"
}