main.go 2.1 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899
  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. httpClient, err := client.NewClientWithResponses(url)
  45. if err != nil {
  46. slog.Error("Failed to create client", "error", err)
  47. os.Exit(1)
  48. }
  49. // Create main context for the application
  50. ctx, cancel := context.WithCancel(context.Background())
  51. defer cancel()
  52. app_, err := app.New(ctx, version, appInfo, httpClient)
  53. if err != nil {
  54. panic(err)
  55. }
  56. program := tea.NewProgram(
  57. tui.NewModel(app_),
  58. tea.WithAltScreen(),
  59. tea.WithKeyboardEnhancements(),
  60. tea.WithMouseCellMotion(),
  61. )
  62. eventClient, err := client.NewClient(url)
  63. if err != nil {
  64. slog.Error("Failed to create event client", "error", err)
  65. os.Exit(1)
  66. }
  67. evts, err := eventClient.Event(ctx)
  68. if err != nil {
  69. slog.Error("Failed to subscribe to events", "error", err)
  70. os.Exit(1)
  71. }
  72. go func() {
  73. for item := range evts {
  74. program.Send(item)
  75. }
  76. }()
  77. // Run the TUI
  78. result, err := program.Run()
  79. if err != nil {
  80. slog.Error("TUI error", "error", err)
  81. }
  82. slog.Info("TUI exited", "result", result)
  83. }