Function parseDocsFlattenArgs

cli/plugins/builtin_docsflatten.go:257–341

parseDocsFlattenArgs supports flat JSON, the {"cmd","args"} envelope and --flag argv form.

(args []string)

// parseDocsFlattenArgs supports flat JSON, the {"cmd","args"} envelope and
// --flag argv form.
func parseDocsFlattenArgs(args []string) (docsFlattenArgs, error) {
	out := docsFlattenArgs{Branch: "main", Format: "text", MaxChars: 16000, StripFrontMatter: true, MaxPages: 50, MaxDepth: 2, SameHost: true, Kind: "docs"}
	payload := strings.TrimSpace(strings.Join(args, " "))
	var raw map[string]json.RawMessage
	switch {
	case strings.HasPrefix(payload, "{"):
		if err := json.Unmarshal([]byte(payload), &raw); err != nil {
			return out, fmt.Errorf("malformed JSON args: %w", err)
		}
		// Accept the {"cmd","args"} envelope the model tends to emit (mirroring
		// @context's shape) as well as flat JSON. Merge the nested args OVER any
		// sibling top-level keys instead of replacing the map wholesale: a partial
		// envelope like {"cmd":"flatten","repo":"…","args":{"format":"jsonl"}}
		// must keep its source key rather than silently drop it (which surfaced as
		// a misleading "source required"/"mutually exclusive" error and cost the
		// agent a retry turn). The "cmd" verb is dropped — it is not a flatten flag.
		if inner, ok := raw["args"]; ok {
			var innerMap map[string]json.RawMessage
			if err := json.Unmarshal(inner, &innerMap); err == nil {
				delete(raw, "args")
				delete(raw, "cmd")
				for k, v := range innerMap {
					raw[k] = v
				}
			}
		}
	default:
		var jsonErr error
		raw, jsonErr = docsFlattenArgvToMap(args)
		if jsonErr != nil {
			return out, jsonErr
		}
	}

	out.Root = jsonString(raw, "root", "dir", "path")
	out.Repo = jsonString(raw, "repo", "repoUrl")
	parseDocsFlattenCrawlArgs(raw, &out)
	if v := jsonString(raw, "branch"); v != "" {
		out.Branch = v
	}
	out.Subdir = jsonString(raw, "subdir")
	if v := strings.ToLower(jsonString(raw, "format")); v != "" {
		out.Format = v
	}
	if v, ok := raw["maxChars"]; ok && len(v) > 0 {
		out.MaxChars = jsonInt(raw, "maxChars")
	} else if _, ok := raw["max-chars"]; ok {
		out.MaxChars = jsonInt(raw, "max-chars")
	}
	out.Include = splitDocsFlattenCSV(jsonString(raw, "include"))
	out.Exclude = splitDocsFlattenCSV(jsonString(raw, "exclude"))
	if v, present := jsonBoolLookup(raw, "stripFrontMatter", "strip-front-matter"); present {
		out.StripFrontMatter = v
	}
	out.Output = jsonString(raw, "output", "out")
	if v := strings.ToLower(jsonString(raw, "kind")); v != "" {
		out.Kind = v
	}