feat: Add bucket schema management commands

cmd/influx/bucket.go

@@ -91,6 +91,12 @@ func newBucketCreateCmd() *cli.Command {
 				EnvVars:     []string{"INFLUX_ORG"},
 				Destination: &params.OrgName,
 			},
+			&cli.StringFlag{
+				Name:        "schema-type",
+				Usage:       "The schema type (implicit, explicit)",
+				DefaultText: "implicit",
+				Destination: &params.SchemaType,
+			},
 		),
 		Action: func(ctx *cli.Context) error {
 			clients := getBucketsClient(ctx)

cmd/influx/main.go

@@ -196,6 +196,7 @@ var app = cli.App{
 		newSetupCmd(),
 		newWriteCmd(),
 		newBucketCmd(),
+		newMeasurementSchemaCmd(),
 	},
 }
 

cmd/influx/measurement_schema.go

@@ -0,0 +1,74 @@
+package main
+
+import (
+	"github.com/influxdata/influx-cli/v2/internal/cmd/measurement_schema"
+	"github.com/influxdata/influx-cli/v2/pkg/cli/middleware"
+	"github.com/urfave/cli/v2"
+)
+
+func withMSClient() cli.BeforeFunc {
+	return middleware.WithBeforeFns(
+		withCli(),
+		withApi(),
+		func(ctx *cli.Context) error {
+			c := getCLI(ctx)
+			client := getAPI(ctx)
+			ctx.App.Metadata["measurement_schema"] = measurement_schema.Client{
+				OrgApi:           client.OrganizationsApi,
+				BucketApi:        client.BucketsApi,
+				BucketSchemasApi: client.BucketSchemasApi,
+				CLI:              c,
+			}
+			return nil
+		})
+}
+
+func getMSClient(ctx *cli.Context) measurement_schema.Client {
+	i, ok := ctx.App.Metadata["measurement_schema"].(measurement_schema.Client)
+	if !ok {
+		panic("missing measurement schema client")
+	}
+	return i
+}
+
+func newMeasurementSchemaCmd() *cli.Command {
+	return &cli.Command{
+		Name:  "bucket-schema",
+		Usage: "Bucket schema management commands",
+		Subcommands: []*cli.Command{
+			newMeasurementSchemaCreateCmd(),
+			newMeasurementSchemaListCmd(),
+		},
+	}
+}
+
+func newMeasurementSchemaCreateCmd() *cli.Command {
+	var params measurement_schema.CreateParams
+	return &cli.Command{
+		Name:   "create",
+		Usage:  "Create a measurement schema for a bucket",
+		Before: withMSClient(),
+		Flags: append(
+			commonFlags,
+			params.Flags()...),
+		Action: func(ctx *cli.Context) error {
+			return getMSClient(ctx).Create(ctx.Context, params)
+		},
+	}
+}
+
+func newMeasurementSchemaListCmd() *cli.Command {
+	var params measurement_schema.ListParams
+	return &cli.Command{
+		Name:   "list",
+		Usage:  "List schemas for a bucket",
+		Before: withMSClient(),
+		Flags: append(
+			commonFlags,
+			params.Flags()...,
+		),
+		Action: func(ctx *cli.Context) error {
+			return getMSClient(ctx).List(ctx.Context, params)
+		},
+	}
+}

etc/generate-openapi.sh

@@ -4,12 +4,14 @@ declare -r ETC_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" &> /dev/null && pwd)"
 declare -r ROOT_DIR="$(dirname ${ETC_DIR})"
 
 declare -r GENERATOR_DOCKER_IMG=openapitools/openapi-generator-cli:v5.1.0
-declare -r OPENAPI_COMMIT=e41d5e327a67e472a46cd6edfe673496e1b066dd
+declare -r OPENAPI_COMMIT=sgc/issues/9672
 
 # Download our target API spec.
 # NOTE: openapi-generator supports HTTP references to API docs, but using that feature
 # causes the host of the URL to be injected into the base paths of generated code.
-curl -o ${ROOT_DIR}/internal/api/cli.yml https://raw.githubusercontent.com/influxdata/openapi/${OPENAPI_COMMIT}/contracts/cli.yml
+# curl -o ${ROOT_DIR}/internal/api/cli.yml https://raw.githubusercontent.com/influxdata/openapi/${OPENAPI_COMMIT}/contracts/cli.yml
+
+cp -c ~/projects/go/influxdata/openapi/contracts/cli.yml ${ROOT_DIR}/internal/api/cli.yml
 
 # Run the generator - This produces many more files than we want to track in git.
 docker run --rm -it -u "$(id -u):$(id -g)" \