Initial release

1 files changed, 123 insertions, 0 deletions

diff --git a/main.go b/main.go
new file mode 100644
index 0000000..841f801
--- /dev/null
+++ b/main.go
@@ -0,0 +1,123 @@
+package main
+
+import (
+	"flag"
+	"fmt"
+	"os"
+	"runtime"
+	"runtime/pprof"
+	"time"
+
+	"fyne.io/fyne/v2"
+	"fyne.io/fyne/v2/app"
+	"fyne.io/fyne/v2/dialog"
+
+	"github.com/shiv/assets"
+	"github.com/shiv/internal/events"
+	"github.com/shiv/internal/logger"
+	"github.com/shiv/internal/plugin"
+	"github.com/shiv/internal/proxy"
+	"github.com/shiv/internal/store"
+	"github.com/shiv/internal/ui"
+)
+
+func main() {
+	verbose := flag.Bool("verbose", false, "enable verbose debug logging")
+	profile := flag.Bool("profile", false, "enable memory profiling")
+	flag.Parse()
+
+	if *profile {
+		go func() {
+			for {
+				time.Sleep(30 * time.Second)
+				runtime.GC()
+				f, err := os.Create("heap.prof")
+				if err != nil {
+					println("err:", err.Error())
+					continue
+				}
+				pprof.WriteHeapProfile(f)
+				f.Close()
+
+				var m runtime.MemStats
+				runtime.ReadMemStats(&m)
+				println("heap:", m.HeapAlloc)
+			}
+		}()
+	}
+	logger.Init(*verbose)
+
+	fyneApp := app.NewWithID("net.shivapp")
+	fyneApp.SetIcon(fyne.NewStaticResource("logo.png", assets.Logo))
+
+	var openProject func(string, fyne.Window)
+	openProject = func(projectPath string, launchWin fyne.Window) {
+		if projectPath == "" {
+			fyneApp.Quit()
+			return
+		}
+
+		projectStore, err := store.Open(projectPath)
+		if err != nil {
+			fmt.Fprintf(os.Stderr, "[shiv] fatal: %v\n", err)
+			fyneApp.Quit()
+			return
+		}
+
+		bus := events.NewBus()
+		bus.Register(projectStore.Intercept)
+		bus.Register(projectStore)
+
+		prefs := fyneApp.Preferences()
+
+		pluginsDir := prefs.StringWithFallback("plugins_dir", fyneApp.Storage().RootURI().Path()+"/plugins")
+		engine, err := plugin.NewEngine(pluginsDir, projectStore, bus)
+		if err != nil {
+			logger.Error("plugin engine: %v", err)
+		} else if engine != nil {
+			bus.Register(engine)
+		}
+
+		proxyAddr := fmt.Sprintf("%s:%d",
+			prefs.StringWithFallback("proxy_host", "127.0.0.1"),
+			prefs.IntWithFallback("proxy_port", 9090),
+		)
+		proxyServer, err := proxy.New(proxyAddr, bus)
+		if err != nil {
+			fmt.Fprintf(os.Stderr, "[shiv] fatal: %v\n", err)
+			projectStore.Close()
+			fyneApp.Quit()
+			return
+		}
+
+		if proxyServer.CA().Fresh() && launchWin != nil {
+			go func() {
+				msg, err := proxyServer.CA().InstallCA()
+				fyne.Do(func() {
+					if err != nil {
+						dialog.ShowError(fmt.Errorf("CA install failed — import it manually from your system config dir:\n%v", err), launchWin)
+					} else {
+						dialog.ShowInformation("CA Installed", msg+"\n\nRestart your browser for changes to take effect.", launchWin)
+					}
+				})
+			}()
+		}
+
+		if prefs.BoolWithFallback("proxy_enabled", true) {
+			bus.EmitProxyRestart(events.ProxyRestartEvent{Addr: proxyAddr})
+		}
+
+		ui.ShowMainWindow(fyneApp, projectStore, bus, launchWin, func(newPath string) {
+			bus.EmitProxyStop(events.ProxyStopEvent{})
+			if engine != nil {
+				engine.Close()
+			}
+			projectStore.Close()
+			openProject(newPath, nil)
+		})
+	}
+
+	ui.ShowLaunchScreen(fyneApp, openProject)
+
+	fyneApp.Run()
+}