docs: handle feature flags in doc generation

Signed-off-by: David Karlsson <35727626+dvdksn@users.noreply.github.com>

Author: dvdksn

Dockerfile

@@ -2,6 +2,7 @@
 
 ARG GO_VERSION=1.25.5
 ARG DOCS_FORMATS="md,yaml"
+ARG DOCS_FEATURES=""
 
 FROM --platform=${BUILDPLATFORM} golangci/golangci-lint:v2.8.0-alpine AS lint-base
 
@@ -20,11 +21,12 @@ FROM base AS docs-build
 COPY --from=docs-gen /out/docsgen /usr/bin
 ENV DOCKER_CLI_PLUGIN_ORIGINAL_CLI_COMMAND="mcp"
 ARG DOCS_FORMATS
+ARG DOCS_FEATURES
 RUN --mount=target=/context \
     --mount=target=.,type=tmpfs <<EOT
   set -e
   rsync -a /context/. .
-  docsgen --formats "$DOCS_FORMATS" --source "docs/generator/reference"
+  docsgen --formats "$DOCS_FORMATS" ${DOCS_FEATURES:+--features "$DOCS_FEATURES"} --source "docs/generator/reference"
   mkdir /out
   cp -r docs/generator/reference/* /out/
 EOT

docker-bake.hcl

@@ -13,6 +13,10 @@ variable "DOCS_FORMATS" {
   default = "md,yaml"
 }
 
+variable "DOCS_FEATURES" {
+  default = "profiles"
+}
+
 group default {
   targets = [
     "all",
@@ -97,7 +101,8 @@ target mcp-gateway-dind {
 target "validate-docs" {
   inherits = ["_common"]
   args = {
-    DOCS_FORMATS = DOCS_FORMATS
+    DOCS_FORMATS   = DOCS_FORMATS
+    DOCS_FEATURES  = DOCS_FEATURES
   }
   target = "docs-validate"
   output = ["type=cacheonly"]
@@ -106,7 +111,8 @@ target "validate-docs" {
 target "update-docs" {
   inherits = ["_common"]
   args = {
-    DOCS_FORMATS = DOCS_FORMATS
+    DOCS_FORMATS   = DOCS_FORMATS
+    DOCS_FEATURES  = DOCS_FEATURES
   }
   target = "docs-update"
   output = ["./docs/generator/reference"]

docs/generator/generate.go

@@ -19,8 +19,9 @@ import (
 const defaultSourcePath = "/reference/"
 
 type options struct {
-	source  string
-	formats []string
+	source   string
+	formats  []string
+	features []string
 }
 
 func gen(opts *options) error {
@@ -36,7 +37,7 @@ func gen(opts *options) error {
 		DisableAutoGenTag: true,
 	}
 
-	cmd.AddCommand(commands.Root(context.TODO(), "", dockerCLI, features.AllDisabled()))
+	cmd.AddCommand(commands.Root(context.TODO(), "", dockerCLI, features.WithEnabled(opts.features)))
 
 	c, err := clidocstool.New(clidocstool.Options{
 		Root:      cmd,
@@ -71,6 +72,7 @@ func run() error {
 	flags := pflag.NewFlagSet(os.Args[0], pflag.ContinueOnError)
 	flags.StringVar(&opts.source, "source", defaultSourcePath, "Docs source folder")
 	flags.StringSliceVar(&opts.formats, "formats", []string{}, "Format (md, yaml)")
+	flags.StringSliceVar(&opts.features, "features", []string{}, "Features to enable (e.g. profiles)")
 	if err := flags.Parse(os.Args[1:]); err != nil {
 		return err
 	}

pkg/features/features.go

@@ -46,6 +46,19 @@ func AllDisabled() Features {
 	}
 }
 
+// WithEnabled returns a Features with only the named features enabled.
+// Recognised names: "profiles".
+func WithEnabled(names []string) Features {
+	f := &featuresImpl{}
+	for _, name := range names {
+		switch name {
+		case "profiles":
+			f.profilesEnabled = true
+		}
+	}
+	return f
+}
+
 func (f *featuresImpl) InitError() error {
 	return f.initErr
 }