main.go 2.2 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101
  1. package main
  2. import (
  3. "context"
  4. "encoding/json"
  5. "log/slog"
  6. "os"
  7. "path/filepath"
  8. "strings"
  9. tea "github.com/charmbracelet/bubbletea/v2"
  10. "github.com/sst/opencode/internal/app"
  11. "github.com/sst/opencode/internal/tui"
  12. "github.com/sst/opencode/pkg/client"
  13. )
  14. var Version = "dev"
  15. func main() {
  16. version := Version
  17. if version != "dev" && !strings.HasPrefix(Version, "v") {
  18. version = "v" + Version
  19. }
  20. url := os.Getenv("OPENCODE_SERVER")
  21. appInfoStr := os.Getenv("OPENCODE_APP_INFO")
  22. var appInfo client.AppInfo
  23. err := json.Unmarshal([]byte(appInfoStr), &appInfo)
  24. if err != nil {
  25. slog.Error("Failed to unmarshal app info", "error", err)
  26. os.Exit(1)
  27. }
  28. logfile := filepath.Join(appInfo.Path.Data, "log", "tui.log")
  29. if _, err := os.Stat(filepath.Dir(logfile)); os.IsNotExist(err) {
  30. err := os.MkdirAll(filepath.Dir(logfile), 0755)
  31. if err != nil {
  32. slog.Error("Failed to create log directory", "error", err)
  33. os.Exit(1)
  34. }
  35. }
  36. file, err := os.Create(logfile)
  37. if err != nil {
  38. slog.Error("Failed to create log file", "error", err)
  39. os.Exit(1)
  40. }
  41. defer file.Close()
  42. logger := slog.New(slog.NewTextHandler(file, &slog.HandlerOptions{Level: slog.LevelDebug}))
  43. slog.SetDefault(logger)
  44. slog.Debug("TUI launched", "app", appInfo)
  45. httpClient, err := client.NewClientWithResponses(url)
  46. if err != nil {
  47. slog.Error("Failed to create client", "error", err)
  48. os.Exit(1)
  49. }
  50. // Create main context for the application
  51. ctx, cancel := context.WithCancel(context.Background())
  52. defer cancel()
  53. app_, err := app.New(ctx, version, appInfo, httpClient)
  54. if err != nil {
  55. panic(err)
  56. }
  57. program := tea.NewProgram(
  58. tui.NewModel(app_),
  59. tea.WithAltScreen(),
  60. tea.WithKeyboardEnhancements(),
  61. tea.WithMouseCellMotion(),
  62. )
  63. eventClient, err := client.NewClient(url)
  64. if err != nil {
  65. slog.Error("Failed to create event client", "error", err)
  66. os.Exit(1)
  67. }
  68. evts, err := eventClient.Event(ctx)
  69. if err != nil {
  70. slog.Error("Failed to subscribe to events", "error", err)
  71. os.Exit(1)
  72. }
  73. go func() {
  74. for item := range evts {
  75. program.Send(item)
  76. }
  77. }()
  78. // Run the TUI
  79. result, err := program.Run()
  80. if err != nil {
  81. slog.Error("TUI error", "error", err)
  82. }
  83. slog.Info("TUI exited", "result", result)
  84. }