wails v2从夸夸到劝退

// Quit the application
func Quit(ctx context.Context) {
  if ctx == nil {
    log.Fatalf("Error calling 'runtime.Quit': %s", contextError)
  }
  appFrontend := getFrontend(ctx)
  appFrontend.Quit()
}

//\wails\v2@v2.11.0\internal\frontend\desktop\windows\frontend.go
func (f *Frontend) Quit() {
  if f.frontendOptions.OnBeforeClose != nil && f.frontendOptions.OnBeforeClose(f.ctx) {
    return
  }
  // Exit must be called on the Main-Thread. It calls PostQuitMessage which sends the WM_QUIT message to the thread's
  // message queue and our message queue runs on the Main-Thread.
  f.mainWindow.Invoke(winc.Exit)
}

OnBeforeClose: func(ctx context.Context) bool {
  // 如果是从托盘强制退出，直接允许关闭
  if forceQuit {
    return false
  }
  if config.CloseToTray {
    runtime.WindowHide(ctx)
    return true
  }
  return false
},

OnStartup: func(ctx context.Context) {
  appCtx = ctx
  // .......

  configService.Init(ctx, db, config)
  // 设置安全退出回调
  configService.SetQuitHandler(func() {
    forceQuit = true
    runtime.Quit(ctx)
  })

  //......

  // 在 Wails 启动后初始化系统托盘
  systrayQuit = make(chan struct{})
  go systray.Run(onSystrayReady, onSystrayExit)
},

OnShutdown: func(ctx context.Context) {
  // 关闭系统托盘
  if systrayQuit != nil {
    systray.Quit()
    <-systrayQuit // 等待 systray 完全退出
  }

  //......

  // 保存配置
  if err := appconf.SaveConfig(config); err != nil {
    appLogger.Error("failed to save config: " + err.Error())
  }
}

// 系统托盘初始化
func onSystrayReady() {
  systray.SetIcon(icon)
  systray.SetTitle("LunaBox")
  systray.SetTooltip("LunaBox")

  // 点击托盘图标时显示窗口
  systray.SetOnClick(func(menu systray.IMenu) {
    if appCtx != nil {
      runtime.WindowShow(appCtx)
    }
  })

  // 双击托盘图标时也显示窗口
  systray.SetOnDClick(func(menu systray.IMenu) {
    if appCtx != nil {
      runtime.WindowShow(appCtx)
    }
  })

  mShow := systray.AddMenuItem("显示主窗口", "显示 LunaBox 主窗口")
  systray.AddSeparator()
  mQuit := systray.AddMenuItem("退出", "退出 LunaBox")

  mShow.Click(func() {
    if appCtx != nil {
      runtime.WindowShow(appCtx)
    }
  })

  mQuit.Click(func() {
    // 通过托盘退出时，设置强制退出标志，绕过 OnBeforeClose 的最小化逻辑
    forceQuit = true
    if appCtx != nil {
      runtime.Quit(appCtx)
    }
  })
}

func onSystrayExit() {
  if systrayQuit != nil {
    close(systrayQuit)
  }
}

// main.go
AssetServer: &assetserver.Options{
  Assets: assets,
  Middleware: func(next http.Handler) http.Handler {
    return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
      // 跨域处理
      w.Header().Set("Access-Control-Allow-Origin", "*")
      w.Header().Set("Access-Control-Allow-Methods", "GET, POST, PUT, DELETE, OPTIONS")
      w.Header().Set("Access-Control-Allow-Headers", "Content-Type, Authorization")

      if r.Method == "OPTIONS" {
        w.WriteHeader(http.StatusOK)
        return
      }

      if strings.HasPrefix(r.URL.Path, "/local/") {
        localFileHandler.ServeHTTP(w, r)
        return
      }

      next.ServeHTTP(w, r)
    })
  },
},

//internal/utils/local_file_handler.go

type LocalFileHandler struct {
  appDir string
}

func NewLocalFileHandler() (*LocalFileHandler, error) {
  appDir, err := GetDataDir()
  if err != nil {
    return nil, err
  }
  return &LocalFileHandler{
    appDir: appDir,
  }, nil
}

// ServeHTTP 实现 http.Handler 接口
func (h *LocalFileHandler) ServeHTTP(w http.ResponseWriter, r *http.Request) {
  // 只处理 /local/ 开头的请求
  if !strings.HasPrefix(r.URL.Path, "/local/") {
    http.NotFound(w, r)
    return
  }

  // 移除 /local/ 前缀
  relativePath := strings.TrimPrefix(r.URL.Path, "/local/")

  // 构建完整路径
  fullPath := filepath.Join(h.appDir, relativePath)

  // 清理路径
  cleanPath := filepath.Clean(fullPath)

  // 检查文件是否存在
  if _, err := os.Stat(cleanPath); os.IsNotExist(err) {
    http.NotFound(w, r)
    return
  }

  // 设置缓存头
  w.Header().Set("Cache-Control", "public, max-age=31536000")

  // 提供文件服务
  http.ServeFile(w, r, cleanPath)
}