Method provisionerDaemonStart

enterprise/cli/provisionerdaemonstart.go:38–396

()

)

func (r *RootCmd) provisionerDaemonStart() *serpent.Command {
	var (
		cacheDir       string
		logHuman       string
		logJSON        string
		logStackdriver string
		logFilter      []string
		name           string
		rawTags        []string
		pollInterval   time.Duration
		pollJitter     time.Duration
		preSharedKey   string
		provisionerKey string
		verbose        bool
		experiments    []string

		prometheusEnable  bool
		prometheusAddress string
	)
	orgContext := agpl.NewOrganizationContext()
	cmd := &serpent.Command{
		Use:   "start",
		Short: "Run a provisioner daemon",
		Handler: func(inv *serpent.Invocation) error {
			ctx, cancel := context.WithCancel(inv.Context())
			defer cancel()
			client, err := r.InitClient(inv)
			if err != nil {
				return err
			}

			stopCtx, stopCancel := inv.SignalNotifyContext(ctx, agpl.StopSignalsNoInterrupt...)
			defer stopCancel()
			interruptCtx, interruptCancel := inv.SignalNotifyContext(ctx, agpl.InterruptSignals...)
			defer interruptCancel()

			orgID := uuid.Nil
			if preSharedKey == "" && provisionerKey == "" {
				// We can only select an organization if using user auth
				org, err := orgContext.Selected(inv, client)
				if err != nil {
					var cErr *codersdk.Error
					if !errors.As(err, &cErr) || cErr.StatusCode() != http.StatusUnauthorized {
						return xerrors.Errorf("current organization: %w", err)
					}

					return xerrors.New("must provide a pre-shared key or provisioner key when not authenticated as a user")
				}

				orgID = org.ID
			} else if orgContext.FlagSelect != "" {
				return xerrors.New("cannot provide --org value with --psk or --key flags")
			}

			if provisionerKey != "" {
				if preSharedKey != "" {
					return xerrors.New("cannot provide both provisioner key --key and pre-shared key --psk")
				}