package application

import (
	"github.com/wailsapp/wails/v2/internal/app"
	"github.com/wailsapp/wails/v2/internal/signal"
	"github.com/wailsapp/wails/v2/pkg/menu"
	"github.com/wailsapp/wails/v2/pkg/options"
)

// Application is the main Wails application
type Application struct {
	application *app.App
	options     *options.App

	// running flag
	running bool
}

// NewWithOptions creates a new Application with the given options
func NewWithOptions(options *options.App) *Application {
	if options == nil {
		return New()
	}
	return &Application{
		options: options,
	}
}

// New creates a new Application with the default options
func New() *Application {
	return &Application{
		options: &options.App{},
	}
}

// SetApplicationMenu sets the application menu
func (a *Application) SetApplicationMenu(appMenu *menu.Menu) {
	if a.running {
		a.application.SetApplicationMenu(appMenu)
		return
	}

	a.options.Menu = appMenu
}

// Run starts the application
func (a *Application) Run() error {

	err := applicationInit()
	if err != nil {
		return err
	}

	application, err := app.CreateApp(a.options)
	if err != nil {
		return err
	}

	a.application = application

	// Control-C handlers
	signal.OnShutdown(func() {
		a.application.Shutdown()
	})
	signal.Start()

	a.running = true

	return a.application.Run()
}

// Quit will shut down the application
func (a *Application) Quit() {
	a.application.Shutdown()
}