Something went wrong. Try again.
[READ-ONLY] Mirror of https://github.com/openstatusHQ/cli. OpenStatus CLI www.openstatus.dev
Something went wrong. Try again.
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320package monitors
import ( "context" "fmt" "net" "net/http" "os" "strconv"
"buf.build/gen/go/openstatus/api/connectrpc/gosimple/openstatus/monitor/v1/monitorv1connect" monitorv1 "buf.build/gen/go/openstatus/api/protocolbuffers/go/openstatus/monitor/v1" "github.com/urfave/cli/v3" "sigs.k8s.io/yaml"
"github.com/openstatusHQ/cli/internal/auth" output "github.com/openstatusHQ/cli/internal/cli" "github.com/openstatusHQ/cli/internal/config")
// ExportMonitor exports all monitors to a YAML file using the SDKfunc ExportMonitor(ctx context.Context, client monitorv1connect.MonitorServiceClient, path string) error { resp, err := client.ListMonitors(ctx, &monitorv1.ListMonitorsRequest{}) if err != nil { return fmt.Errorf("failed to list monitors: %w", err) }
t := map[string]config.Monitor{}
for _, monitor := range resp.GetHttpMonitors() { t[monitor.GetId()] = convertHTTPMonitorToConfig(monitor) }
for _, monitor := range resp.GetTcpMonitors() { t[monitor.GetId()] = convertTCPMonitorToConfig(monitor) }
configYAML, err := yaml.Marshal(&t) if err != nil { return err }
lock := make(map[string]config.Lock, len(t)) for id, monitor := range t { i, err := strconv.Atoi(id) if err != nil { return fmt.Errorf("invalid monitor ID %q: %w", id, err) } lock[id] = config.Lock{ ID: i, Monitor: monitor, } }
lockYAML, err := yaml.Marshal(&lock) if err != nil { return fmt.Errorf("failed to marshal lock file: %w", err) }
file, err := os.OpenFile(path, os.O_RDWR|os.O_CREATE|os.O_TRUNC, 0o600) if err != nil { return err } defer file.Close()
if _, err := file.WriteString("# yaml-language-server: $schema=https://www.openstatus.dev/schema.json\n\n"); err != nil { return err } if _, err := file.Write(configYAML); err != nil { return err }
lockFile, err := os.OpenFile("openstatus.lock", os.O_RDWR|os.O_CREATE|os.O_TRUNC, 0o600) if err != nil { return fmt.Errorf("failed to create lock file: %w", err) } defer lockFile.Close()
if _, err := lockFile.Write(lockYAML); err != nil { return fmt.Errorf("failed to write lock file: %w", err) }
return nil}
// convertHTTPMonitorToConfig converts an SDK HTTPMonitor to config.Monitorfunc convertHTTPMonitorToConfig(m *monitorv1.HTTPMonitor) config.Monitor { regions := make([]config.Region, len(m.GetRegions())) for i, r := range m.GetRegions() { regions[i] = config.Region(regionToString(r)) }
headers := make(map[string]string) for _, h := range m.GetHeaders() { if h.GetKey() != "" { headers[h.GetKey()] = h.GetValue() } }
var assertions []config.Assertion for _, a := range m.GetStatusCodeAssertions() { assertions = append(assertions, config.Assertion{ Kind: config.StatusCode, Target: int(a.GetTarget()), Compare: convertNumberComparator(a.GetComparator()), }) } for _, a := range m.GetBodyAssertions() { assertions = append(assertions, config.Assertion{ Kind: config.TextBody, Target: a.GetTarget(), Compare: convertStringComparator(a.GetComparator()), }) } for _, a := range m.GetHeaderAssertions() { assertions = append(assertions, config.Assertion{ Kind: config.Header, Target: a.GetTarget(), Compare: convertStringComparator(a.GetComparator()), Key: a.GetKey(), }) }
return config.Monitor{ Name: m.GetName(), Description: m.GetDescription(), Active: m.GetActive(), Public: m.GetPublic(), Frequency: convertPeriodicity(m.GetPeriodicity()), DegradedAfter: m.GetDegradedAt(), Timeout: m.GetTimeout(), Retry: m.GetRetry(), Kind: config.HTTP, Regions: regions, Assertions: assertions, Request: config.Request{ URL: m.GetUrl(), Method: convertHTTPMethod(m.GetMethod()), Body: m.GetBody(), Headers: headers, }, }}
// convertTCPMonitorToConfig converts an SDK TCPMonitor to config.Monitorfunc convertTCPMonitorToConfig(m *monitorv1.TCPMonitor) config.Monitor { regions := make([]config.Region, len(m.GetRegions())) for i, r := range m.GetRegions() { regions[i] = config.Region(regionToString(r)) }
// Parse host:port from URI using net.SplitHostPort for IPv6 support uri := m.GetUri() host, portStr, err := net.SplitHostPort(uri) var port int64 if err != nil { // Fallback: treat the whole URI as host if no port separator found host = uri } else { p, _ := strconv.Atoi(portStr) port = int64(p) }
return config.Monitor{ Name: m.GetName(), Description: m.GetDescription(), Active: m.GetActive(), Public: m.GetPublic(), Frequency: convertPeriodicity(m.GetPeriodicity()), DegradedAfter: m.GetDegradedAt(), Timeout: m.GetTimeout(), Retry: m.GetRetry(), Kind: config.TCP, Regions: regions, Request: config.Request{ Host: host, Port: port, }, }}
// convertPeriodicity converts SDK Periodicity enum to config.Frequency stringfunc convertPeriodicity(p monitorv1.Periodicity) config.Frequency { switch p { case monitorv1.Periodicity_PERIODICITY_30S: return config.The30S case monitorv1.Periodicity_PERIODICITY_1M: return config.The1M case monitorv1.Periodicity_PERIODICITY_5M: return config.The5M case monitorv1.Periodicity_PERIODICITY_10M: return config.The10M case monitorv1.Periodicity_PERIODICITY_30M: return config.The30M case monitorv1.Periodicity_PERIODICITY_1H: return config.The1H default: return config.The10M }}
// convertHTTPMethod converts SDK HTTPMethod enum to config.Method stringfunc convertHTTPMethod(m monitorv1.HTTPMethod) config.Method { switch m { case monitorv1.HTTPMethod_HTTP_METHOD_GET: return config.Get case monitorv1.HTTPMethod_HTTP_METHOD_POST: return config.Post case monitorv1.HTTPMethod_HTTP_METHOD_PUT: return config.Put case monitorv1.HTTPMethod_HTTP_METHOD_PATCH: return config.Patch case monitorv1.HTTPMethod_HTTP_METHOD_DELETE: return config.Delete case monitorv1.HTTPMethod_HTTP_METHOD_HEAD: return config.Head case monitorv1.HTTPMethod_HTTP_METHOD_OPTIONS: return config.Options default: return config.Get }}
// convertNumberComparator converts SDK NumberComparator to config.Comparefunc convertNumberComparator(c monitorv1.NumberComparator) config.Compare { switch c { case monitorv1.NumberComparator_NUMBER_COMPARATOR_EQUAL: return config.Eq case monitorv1.NumberComparator_NUMBER_COMPARATOR_NOT_EQUAL: return config.NotEq case monitorv1.NumberComparator_NUMBER_COMPARATOR_GREATER_THAN: return config.Gt case monitorv1.NumberComparator_NUMBER_COMPARATOR_GREATER_THAN_OR_EQUAL: return config.Gte case monitorv1.NumberComparator_NUMBER_COMPARATOR_LESS_THAN: return config.Lt case monitorv1.NumberComparator_NUMBER_COMPARATOR_LESS_THAN_OR_EQUAL: return config.LTE default: return config.Eq }}
// convertStringComparator converts SDK StringComparator to config.Comparefunc convertStringComparator(c monitorv1.StringComparator) config.Compare { switch c { case monitorv1.StringComparator_STRING_COMPARATOR_EQUAL: return config.Eq case monitorv1.StringComparator_STRING_COMPARATOR_NOT_EQUAL: return config.NotEq case monitorv1.StringComparator_STRING_COMPARATOR_CONTAINS: return config.Contains case monitorv1.StringComparator_STRING_COMPARATOR_NOT_CONTAINS: return config.NotContains case monitorv1.StringComparator_STRING_COMPARATOR_EMPTY: return config.Empty case monitorv1.StringComparator_STRING_COMPARATOR_NOT_EMPTY: return config.NotEmpty case monitorv1.StringComparator_STRING_COMPARATOR_GREATER_THAN: return config.Gt case monitorv1.StringComparator_STRING_COMPARATOR_GREATER_THAN_OR_EQUAL: return config.Gte case monitorv1.StringComparator_STRING_COMPARATOR_LESS_THAN: return config.Lt case monitorv1.StringComparator_STRING_COMPARATOR_LESS_THAN_OR_EQUAL: return config.LTE default: return config.Eq }}
// ExportMonitorWithHTTPClient is a convenience function that creates a client and exports monitorsfunc ExportMonitorWithHTTPClient(ctx context.Context, httpClient *http.Client, apiKey string, path string) error { client := NewMonitorClientWithHTTPClient(httpClient, apiKey) return ExportMonitor(ctx, client, path)}
func GetMonitorImportCmd() *cli.Command { monitorImportCmd := cli.Command{ Name: "import", Usage: "Import all your monitors", UsageText: `openstatus monitors import openstatus monitors import --output monitors.yaml`, Description: "Import all your monitors from your workspace to a YAML file; it will also create a lock file to manage your monitors with 'apply'.\n\nDEPRECATED: the monitors.yaml config file is deprecated for import. Prefer managing monitors declaratively with 'openstatus monitors apply'.", Action: func(ctx context.Context, cmd *cli.Command) error { apiKey, err := auth.ResolveAccessToken(cmd) if err != nil { return cli.Exit(err.Error(), 1) } fmt.Fprintln(os.Stderr, "Warning: the monitors.yaml config file is deprecated for import. Prefer managing monitors declaratively with 'openstatus monitors apply'.") s := output.StartSpinner("Importing monitors...") client := NewMonitorClient(apiKey) err = ExportMonitor(ctx, client, cmd.String("output")) output.StopSpinner(s) if err != nil { return cli.Exit(err.Error(), 1) } fmt.Printf("Monitors successfully imported to: %s\n", cmd.String("output")) fmt.Println("Run 'openstatus monitors apply' to sync changes") return nil }, Flags: []cli.Flag{ &cli.StringFlag{ Name: "access-token", Usage: "OpenStatus API Access Token", Aliases: []string{"t"}, Sources: cli.EnvVars("OPENSTATUS_API_TOKEN"), }, &cli.StringFlag{ Name: "output", Usage: "The output file name ", DefaultText: "openstatus.yaml", Value: "openstatus.yaml", Aliases: []string{"o"}, }, }, } return &monitorImportCmd}