Warp/apps/desktop/cmd/warp-desktop/main.go

// Command warp-desktop is the Warp desktop client: the web client in a
// native window (Wails v3), talking to the homeserver directly.
//
//	warp-desktop [-homeserver https://matrix.example.org]
package main

import (
	_ "embed"
	"flag"
	"fmt"
	"log/slog"
	"os"
	"path/filepath"
	"runtime"

	"github.com/wailsapp/wails/v3/pkg/application"

	"github.com/dariuszjaniak/warp/apps/desktop/internal/shell"
	"github.com/dariuszjaniak/warp/apps/web/client"
)

// Set with -ldflags "-X main.version=...".
var version = "0.0.0-dev"

//go:embed appicon.png
var icon []byte

// appID names the app to the OS: the single-instance lock, and the
// directory the settings live in on Linux and macOS.
const appID = "io.github.dariuszjaniak.warp"

func main() {
	if err := run(); err != nil {
		fmt.Fprintln(os.Stderr, "warp-desktop:", err)
		os.Exit(1)
	}
}

func run() error {
	homeserver := flag.String("homeserver", os.Getenv("WARP_DESKTOP_HOMESERVER"),
		"homeserver base URL for this run, instead of the one chosen in the app (WARP_DESKTOP_HOMESERVER)")
	showVersion := flag.Bool("version", false, "print the version and exit")
	flag.Parse()

	if *showVersion {
		fmt.Println("warp-desktop", version)
		return nil
	}

	logger := slog.New(slog.NewTextHandler(os.Stderr, nil))
	slog.SetDefault(logger)

	dirs, err := dataDirs()
	if err != nil {
		return err
	}

	files, built := client.FS()
	if !built {
		logger.Warn("the web client was not built into this binary; serving a placeholder page")
		files = nil
	}

	var app *application.App
	sh, err := shell.New(shell.Options{
		Store:      shell.NewStore(filepath.Join(dirs.config, "desktop.json")),
		Homeserver: *homeserver,
		Client:     files,
		Version:    version,
		Logger:     logger,
		OpenURL:    func(u string) error { return app.Browser.OpenURL(u) },
	})
	if err != nil {
		return err
	}

	var window *application.WebviewWindow
	app = application.New(application.Options{
		Name:        "Warp",
		Description: "A Matrix client",
		Icon:        icon,
		Logger:      logger,
		Assets:      application.AssetOptions{Handler: sh},
		// The client's sessions and encryption keys are in the webview's
		// storage; two instances on one store would both sync and both
		// use the same one-time keys.
		SingleInstance: &application.SingleInstanceOptions{
			UniqueID: appID,
			OnSecondInstanceLaunch: func(application.SecondInstanceData) {
				if window != nil {
					window.Restore()
					window.Show()
					window.Focus()
				}
			},
		},
		Mac: application.MacOptions{
			ApplicationShouldTerminateAfterLastWindowClosed: true,
		},
		Windows: application.WindowsOptions{
			WebviewUserDataPath:   dirs.webview,
			AdditionalBrowserArgs: devBrowserArgs(),
		},
	})

	window = app.Window.NewWithOptions(application.WebviewWindowOptions{
		Name:      "main",
		Title:     "Warp",
		Width:     1200,
		Height:    800,
		MinWidth:  360,
		MinHeight: 480,
		URL:       "/",
	})

	app.Menu.SetApplicationMenu(appMenu(app, window))

	return app.Run()
}

func appMenu(app *application.App, window *application.WebviewWindow) *application.Menu {
	menu := app.NewMenu()
	if runtime.GOOS == "darwin" {
		menu.AddRole(application.AppMenu)
	}
	file := menu.AddSubmenu("File")
	file.Add("Change homeserver…").OnClick(func(*application.Context) {
		window.SetURL(shell.SetupPath)
	})
	file.AddSeparator()
	if runtime.GOOS == "darwin" {
		file.AddRole(application.CloseWindow)
	} else {
		file.AddRole(application.Quit)
	}
	menu.AddRole(application.EditMenu)
	menu.AddRole(application.ViewMenu)
	menu.AddRole(application.WindowMenu)
	return menu
}

type dirs struct {
	config  string // settings
	webview string // the webview's storage, on Windows
}

func dataDirs() (dirs, error) {
	cfg, err := os.UserConfigDir()
	if err != nil {
		return dirs{}, err
	}
	d := dirs{config: filepath.Join(cfg, "Warp")}
	if runtime.GOOS == "windows" {
		// %LocalAppData%: the webview's storage is large and machine-local,
		// so it doesn't belong in the roaming profile with the settings.
		local, err := os.UserCacheDir()
		if err != nil {
			return dirs{}, err
		}
		d.webview = filepath.Join(local, "Warp", "WebView2")
	}
	return d, nil
}