ldcli/cmd/dev_server/dev_server.go at 2b7a3b1dd322891932ae5c6ffe640de1591c6db4 · launchdarkly/ldcli · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
package dev_server

import (
	"fmt"

	"github.com/spf13/cobra"
	"github.com/spf13/viper"

	cmdAnalytics "github.com/launchdarkly/ldcli/cmd/analytics"
	"github.com/launchdarkly/ldcli/cmd/cliflags"
	resourcecmd "github.com/launchdarkly/ldcli/cmd/resources"
	"github.com/launchdarkly/ldcli/internal/analytics"
	"github.com/launchdarkly/ldcli/internal/dev_server"
	"github.com/launchdarkly/ldcli/internal/resources"
)

func NewDevServerCmd(client resources.Client, analyticsTrackerFn analytics.TrackerFn, ldClient dev_server.Client) *cobra.Command {
	cmd := &cobra.Command{
		Use:   "dev-server",
		Short: "Development server",
		Long:  "Start and use a local development server for overriding flag values.",
		PersistentPreRun: func(cmd *cobra.Command, args []string) {

			tracker := analyticsTrackerFn(
				viper.GetString(cliflags.AccessTokenFlag),
				viper.GetString(cliflags.BaseURIFlag),
				viper.GetBool(cliflags.AnalyticsOptOut),
			)
			tracker.SendCommandRunEvent(cmdAnalytics.CmdRunEventProperties(
				cmd,
				"dev-server",
				map[string]interface{}{
					"action": cmd.Name(),
				}))
		},
	}

	cmd.PersistentFlags().String(
		cliflags.DevStreamURIFlag,
		cliflags.DevStreamURIDefault,
		cliflags.DevStreamURIDescription,
	)
	_ = viper.BindPFlag(cliflags.DevStreamURIFlag, cmd.PersistentFlags().Lookup(cliflags.DevStreamURIFlag))

	cmd.PersistentFlags().String(
		cliflags.PortFlag,
		cliflags.PortDefault,
		cliflags.PortFlagDescription,
	)

	_ = viper.BindPFlag(cliflags.PortFlag, cmd.PersistentFlags().Lookup(cliflags.PortFlag))

	cmd.PersistentFlags().String(
		cliflags.HostFlag,
		cliflags.HostDefault,
		cliflags.HostFlagDescription,
	)

	_ = viper.BindPFlag(cliflags.HostFlag, cmd.PersistentFlags().Lookup(cliflags.HostFlag))

	cmd.PersistentFlags().Bool(
		cliflags.CorsEnabledFlag,
		false,
		cliflags.CorsEnabledFlagDescription,
	)
	_ = viper.BindPFlag(cliflags.CorsEnabledFlag, cmd.PersistentFlags().Lookup(cliflags.CorsEnabledFlag))

	cmd.PersistentFlags().String(
		cliflags.CorsOriginFlag,
		"*",
		cliflags.CorsOriginFlagDescription,
	)
	_ = viper.BindPFlag(cliflags.CorsOriginFlag, cmd.PersistentFlags().Lookup(cliflags.CorsOriginFlag))

	// Add subcommands here
	cmd.AddGroup(&cobra.Group{ID: "projects", Title: "Project commands:"})
	cmd.AddCommand(NewListProjectsCmd(client))
	cmd.AddCommand(NewGetProjectCmd(client))
	cmd.AddCommand(NewSyncProjectCmd(client))
	cmd.AddCommand(NewRemoveProjectCmd(client))
	cmd.AddCommand(NewAddProjectCmd(client))
	cmd.AddCommand(NewUpdateProjectCmd(client))
	cmd.AddCommand(NewImportProjectCmd())

	cmd.AddGroup(&cobra.Group{ID: "overrides", Title: "Override commands:"})
	cmd.AddCommand(NewAddOverrideCmd(client))
	cmd.AddCommand(NewRemoveOverrideCmd(client))
	cmd.AddCommand(NewDeleteOverridesCmd(client))
	cmd.AddGroup(&cobra.Group{ID: "server", Title: "Server commands:"})

	cmd.AddCommand(NewStartServerCmd(ldClient))
	cmd.AddCommand(NewUICmd())

	cmd.SetUsageTemplate(resourcecmd.SubcommandUsageTemplate())

	return cmd
}

func getDevServerUrl() string {
	host := viper.GetString(cliflags.HostFlag)
	if host == "0.0.0.0" {
		host = "localhost"
	}
	return fmt.Sprintf("http://%s:%s", host, viper.GetString(cliflags.PortFlag))
}