2
0

main.go 2.2 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102
  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. json.Unmarshal([]byte(appInfoStr), &appInfo)
  24. httpClient, err := client.NewClientWithResponses(url)
  25. if err != nil {
  26. slog.Error("Failed to create client", "error", err)
  27. os.Exit(1)
  28. }
  29. // Create main context for the application
  30. ctx, cancel := context.WithCancel(context.Background())
  31. defer cancel()
  32. app_, err := app.New(ctx, version, appInfo, httpClient)
  33. if err != nil {
  34. panic(err)
  35. }
  36. program := tea.NewProgram(
  37. tui.NewModel(app_),
  38. tea.WithAltScreen(),
  39. tea.WithKeyboardEnhancements(),
  40. // tea.WithMouseCellMotion(),
  41. )
  42. eventClient, err := client.NewClient(url)
  43. if err != nil {
  44. slog.Error("Failed to create event client", "error", err)
  45. os.Exit(1)
  46. }
  47. evts, err := eventClient.Event(ctx)
  48. if err != nil {
  49. slog.Error("Failed to subscribe to events", "error", err)
  50. os.Exit(1)
  51. }
  52. go func() {
  53. for item := range evts {
  54. program.Send(item)
  55. }
  56. }()
  57. go func() {
  58. paths, err := httpClient.PostPathGetWithResponse(context.Background())
  59. if err != nil {
  60. panic(err)
  61. }
  62. logfile := filepath.Join(paths.JSON200.Data, "log", "tui.log")
  63. if _, err := os.Stat(filepath.Dir(logfile)); os.IsNotExist(err) {
  64. err := os.MkdirAll(filepath.Dir(logfile), 0755)
  65. if err != nil {
  66. slog.Error("Failed to create log directory", "error", err)
  67. os.Exit(1)
  68. }
  69. }
  70. file, err := os.Create(logfile)
  71. if err != nil {
  72. slog.Error("Failed to create log file", "error", err)
  73. os.Exit(1)
  74. }
  75. defer file.Close()
  76. logger := slog.New(slog.NewTextHandler(file, &slog.HandlerOptions{Level: slog.LevelDebug}))
  77. slog.SetDefault(logger)
  78. }()
  79. // Run the TUI
  80. result, err := program.Run()
  81. if err != nil {
  82. slog.Error("TUI error", "error", err)
  83. // return fmt.Errorf("TUI error: %v", err)
  84. }
  85. slog.Info("TUI exited", "result", result)
  86. }