FilesExpand file tree

main

/

gui_main.go

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

216 lines (199 loc) · 6.79 KB

main

/

gui_main.go

Copy path

Top

File metadata and controls

• Code
• Blame

216 lines (199 loc) · 6.79 KB

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

29

30

31

32

33

34

35

36

37

38

39

40

41

42

43

44

45

46

47

48

49

50

51

52

53

54

55

56

57

58

59

60

61

62

63

64

65

66

67

68

69

70

71

72

73

74

75

76

77

78

79

80

81

82

83

84

85

86

87

88

89

90

91

92

93

94

95

96

97

98

99

100

101

102

103

104

105

106

107

108

109

110

111

112

113

114

115

116

117

118

119

120

121

122

123

124

125

126

127

128

129

130

131

132

133

134

135

136

137

138

139

140

141

142

143

144

145

146

147

148

149

150

151

152

153

154

155

156

157

158

159

160

161

162

163

164

165

166

167

168

169

170

171

172

173

174

175

176

177

178

179

180

181

182

183

184

185

186

187

188

189

190

191

192

193

194

195

196

197

198

199

200

201

202

203

204

205

206

207

208

209

210

211

212

213

214

215

216

//go:build wailsgui

package main

import (

"context"

"encoding/json"

"fmt"

"io/fs"

"log"

"os"

"path/filepath"

"time"

"github.com/wailsapp/wails/v2"

"github.com/wailsapp/wails/v2/pkg/options"

"github.com/wailsapp/wails/v2/pkg/options/assetserver"

"github.com/wailsapp/wails/v2/pkg/options/windows"

wailsruntime "github.com/wailsapp/wails/v2/pkg/runtime"

"grok_switch/internal/agentbridge"

grokconfig "grok_switch/internal/config"

"grok_switch/internal/cpamint"

"grok_switch/internal/crash"

"grok_switch/internal/grokauth"

"grok_switch/internal/grokpool"

"grok_switch/internal/mcpserver"

"grok_switch/internal/paths"

"grok_switch/internal/profiles"

"grok_switch/internal/registrar"

"grok_switch/internal/remoteaccess"

"grok_switch/internal/server"

"grok_switch/internal/settings"

"grok_switch/internal/singleinstance"

"grok_switch/internal/switcher"

"grok_switch/internal/updatecheck"

)

func main() {

// MCP 子进程：由 Grok Agent 以 `grok_switch_gui mcp` 拉起，暴露生图工具。

if len(os.Args) > 1 && os.Args[1] == "mcp" {

os.Exit(mcpserver.Run(os.Args[2:]))

}

defer crash.RecoverMainThread()

resolved, err := paths.Resolve()

if err != nil {

guiFatal(err)

}

crash.Setup(resolved.LogFile)

if err := resolved.Ensure(); err != nil {

guiFatal(err)

}

settingsStore := settings.NewStore(resolved.SettingsFile)

instanceLock, alreadyRunning, err := singleinstance.Acquire(resolved.DataDir)

if err != nil {

guiFatal(fmt.Errorf("创建单实例锁失败: %w", err))

}

if alreadyRunning {

url, findErr := waitForExistingInstanceURL(settingsStore, resolved.DataDir, 3*time.Second)

if findErr != nil {

guiFatal(fmt.Errorf("连接正在运行的 grok_switch 失败: %w", findErr))

}

if err := runWailsWindow(url, resolved.DataDir); err != nil {

guiFatal(err)

}

return

}

defer instanceLock.Close()

exePath, err := os.Executable()

if err != nil {

guiFatal(err)

}

exePath, _ = filepath.Abs(exePath)

profileStore := profiles.NewStore(resolved.ProfilesFile)

grokAuthStore := grokauth.NewStore(resolved.GrokAuthFile)

grokPool, err := grokpool.NewManager(resolved.GrokPoolDir)

if err != nil {

guiFatal(err)

}

if err := grokAuthStore.SetProxyURL(grokPool.Status().Settings.ProxyURL); err != nil {

guiFatal(err)

}

if singleStatus, statusErr := grokAuthStore.Status(); statusErr == nil && singleStatus.Configured {

if raw, readErr := os.ReadFile(resolved.GrokAuthFile); readErr == nil {

if _, migrateErr := grokPool.Ensure([]grokpool.ImportFile{{Name: "legacy-grok-auth.json", Content: string(raw)}}); migrateErr != nil {

crash.Logf("migrate legacy Grok auth into pool: %v", migrateErr)

}

}

}

grokPool.Start()

defer grokPool.Close()

registrarService, err := registrar.NewService(resolved.DataDir)

if err != nil {

guiFatal(err)

}

defer registrarService.Close()

sw := &switcher.Switcher{

ConfigPath: resolved.GrokConfig,

BackupsDir: resolved.BackupsDir,

Profiles: profileStore,

}

if err := sw.EnsureDefaultProfile(); err != nil {

crash.Logf("default profile import skipped: %v", err)

}

currentSettings, err := settingsStore.Get()

if err != nil {

guiFatal(err)

}

agent := agentbridge.New(resolved.GrokHome, filepath.Join(resolved.DataDir, "agent.log"))

agent.SetDefaultCwd(currentSettings.AgentDefaultCwd)

defer agent.Stop()

updateState := updatecheck.NewPreferenceStore(resolved.UpdateStateFile)

appServer := &server.Server{

Paths: resolved,

Profiles: profileStore,

Settings: settingsStore,

RemoteAccess: remoteaccess.NewStore(resolved.RemoteAccessFile),

GrokAuth: grokAuthStore,

GrokPool: grokPool,

CpaMint: cpamint.NewService(),

Registrar: registrarService,

Switcher: sw,

Agent: agent,

Assets: assets,

ExePath: exePath,

Version: version,

UpdateChecker: updatecheck.New(version, "grok_switch_gui.exe"),

UpdateState: updateState,

}

httpServer, port, err := appServer.Listen(currentSettings.Port)

if err != nil {

guiFatal(err)

}

startUpdateNotification(appServer.UpdateChecker, updateState, "grok_switch GUI")

if crashFile := resolved.LogFile; crashFile != "" {

if f, ferr := os.OpenFile(crashFile, os.O_CREATE|os.O_APPEND|os.O_WRONLY, 0o644); ferr == nil {

httpServer.ErrorLog = log.New(f, "http: ", log.LstdFlags)

defer f.Close()

}

}

defer func() {

ctx, cancel := context.WithTimeout(context.Background(), 3*time.Second)

defer cancel()

_ = httpServer.Shutdown(ctx)

}()

url := fmt.Sprintf("http://127.0.0.1:%d", port)

// 生图能力全局开关：开启时通过 Grok CLI 原生 [mcp_servers] 注册生图

// MCP 服务器；关闭时移除注册，模型回到无生图工具的原始状态。

if currentSettings.ImageGenEnabled {

if err := grokconfig.EnsureMcpServerToFile(resolved.GrokConfig, grokconfig.McpServerConfig{

Name: "image_generator",

Command: exePath,

Args: []string{"mcp"},

Env: map[string]string{"GROK_SWITCH_BASE_URL": url},

}); err != nil {

crash.Logf("register mcp server failed: %v", err)

}

} else if err := grokconfig.RemoveMcpServerToFile(resolved.GrokConfig, "image_generator"); err != nil {

crash.Logf("remove mcp server failed: %v", err)

}

if err := runWailsWindow(url, resolved.DataDir); err != nil {

guiFatal(err)

}

}

func runWailsWindow(url, dataDir string) error {

loadingAssets, err := fs.Sub(assets, "gui")

if err != nil {

return fmt.Errorf("加载 GUI 启动画面失败: %w", err)

}

targetJSON, _ := json.Marshal(url)

redirectScript := fmt.Sprintf(`if (window.location.origin !== new URL(%s).origin) { window.location.replace(%s); }`, targetJSON, targetJSON)

icon, _ := assets.ReadFile("assets/icon.ico")

trayController := newGUITrayController(url, icon)

trayController.register()

defer trayController.shutdown()

return wails.Run(&options.App{

Title: "grok_switch GUI",

Width: 1280,

Height: 820,

MinWidth: 960,

MinHeight: 640,

BackgroundColour: &options.RGBA{R: 247, G: 247, B: 245, A: 255},

AssetServer: &assetserver.Options{Assets: loadingAssets},

OnStartup: trayController.startup,

OnDomReady: func(ctx context.Context) { wailsruntime.WindowExecJS(ctx, redirectScript) },

OnBeforeClose: trayController.beforeClose,

OnShutdown: func(context.Context) { trayController.shutdown() },

WindowStartState: options.Normal,

HideWindowOnClose: false,

Windows: &windows.Options{

Theme: windows.Light,

WebviewUserDataPath: filepath.Join(dataDir, "wails-webview2"),

ResizeDebounceMS: 12,

Messages: &windows.Messages{

Webview2NotInstalled: "需要 Microsoft Edge WebView2 Runtime 才能运行 grok_switch GUI。",

Error: "grok_switch GUI 启动失败",

},

},

})

}

func guiFatal(err error) {

crash.ReportFatal(err)

os.Exit(1)

}