package main

import (
	"context"
	"fmt"
	"log"
	"net/http"
	"os"
	"strconv"
	"sync"

	"github.com/gotk3/gotk3/glib"
	"github.com/gotk3/gotk3/gtk"
)

var (
	DEFAULT_PORT, DEFAULT_DIR string = "8080", "."
)

func serve(port, dir string, wg *sync.WaitGroup) *http.Server {
	mux := http.NewServeMux()
	mux.Handle("/", http.FileServer(http.Dir(dir)))
    server := &http.Server{Addr: ":" + port, Handler: mux}

	if port == "" { port = DEFAULT_PORT }
	if dir == "" { dir = DEFAULT_DIR }

    go func() {
        defer wg.Done()

        if err := server.ListenAndServe(); err != http.ErrServerClosed {
			// TODO: show error in GTK error window
            log.Printf("Error while running HTTP server: %v\n", err)
        }
    }()

    return server
}


func main() {
	app, _ := gtk.ApplicationNew("net.mikunonaka.gofileserver", glib.APPLICATION_FLAGS_NONE)
	app.Connect("activate", func() { onActivate(app) })
	app.Run(os.Args)
}

func onActivate(app *gtk.Application) {
	win, _ := gtk.ApplicationWindowNew(app)
	win.SetTitle("HTTP File Server")
	win.SetDefaultSize(400, 400)

	box, _ := gtk.BoxNew(gtk.ORIENTATION_VERTICAL, 0)

	dirBox, _ := gtk.BoxNew(gtk.ORIENTATION_HORIZONTAL, 0)
	dirLabel, _ := gtk.LabelNew("Directory to serve: ")
	dirBox.PackStart(dirLabel, false, false, 5)
	dirInput, _ := gtk.EntryNew()
	dirInput.SetText(DEFAULT_DIR)
	dirBox.PackEnd(dirInput, false, false, 5)

	browseButton, _ := gtk.FileChooserButtonNew("Browse", gtk.FILE_CHOOSER_ACTION_SELECT_FOLDER)
	dirBox.PackEnd(browseButton, false, false, 5)
	box.PackStart(dirBox, false, false, 5)

	portBox, _ := gtk.BoxNew(gtk.ORIENTATION_HORIZONTAL, 0)
	portLabel, _ := gtk.LabelNew("Port:")
	portBox.PackStart(portLabel, false, false, 5)
	portInput, _ := gtk.EntryNew()
	portInput.SetText(DEFAULT_PORT)
	portBox.PackEnd(portInput, false, false, 5)
	box.PackStart(portBox, false, false, 5)

	buttonBox, _ := gtk.BoxNew(gtk.ORIENTATION_HORIZONTAL, 0)
	buttonSwitch, _ := gtk.ToggleButtonNew()
	buttonSwitch.SetLabel("Start")
	buttonBox.PackStart(buttonSwitch, true, true, 0)
	box.PackStart(buttonBox, false, false, 5)

	statusLabel, _ := gtk.LabelNew("")
	box.PackStart(statusLabel, false, true, 10)

	browseButton.Connect("selection-changed", func() {
		dirInput.SetText(browseButton.GetFilename())
	})

	var on = false
	var server *http.Server
	buttonSwitch.Connect("toggled", func() {
		// TODO: validate dir and port
		port, _ := portInput.GetText()
		dir, _ := dirInput.GetText()

		if on {
			server.Shutdown(context.TODO())
		} else {
			go func() {
				killServerDone := &sync.WaitGroup{}
				server = serve(port, dir, killServerDone)

				killServerDone.Add(1)
				// do this after server starts
				on = true
				buttonSwitch.SetLabel("Stop")
				statusLabel.SetText(fmt.Sprintf("Serving directory '%s' on PORT %s", dir, port))
				dirInput.SetEditable(false)
				dirInput.SetCanFocus(false)
				portInput.SetEditable(false)
				portInput.SetCanFocus(false)

				killServerDone.Wait()
				// do this after server shuts down
				on = false
				buttonSwitch.SetLabel("Start")
				statusLabel.SetText("")
				dirInput.SetEditable(true)
				dirInput.SetCanFocus(true)
				portInput.SetEditable(true)
				portInput.SetCanFocus(true)
			}()
		}
	})

	portInput.Connect("insert-text", func(_ *gtk.Entry, input string) {
		if _, err := strconv.Atoi(input); err != nil {
			// if input is not numeric, don't insert
			portInput.StopEmission("insert-text")
		} else {
			// stop if port number is invalid / out of range
			currentInput, _ := portInput.GetText()
			nextInput := currentInput + input
			n, _ := strconv.Atoi(nextInput)
			if n < 0 || n > 65535 {
				portInput.StopEmission("insert-text")
			}
		}
	})

	win.Add(box)
	win.ShowAll()
}