tools/fidl/fidlgen_python/codegen/codegen.go

// Copyright 2025 The Fuchsia Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

package codegen

import (
	"embed"
	"strings"
	"text/template"

	"go.fuchsia.dev/fuchsia/tools/fidl/lib/fidlgen"
)

//go:embed *.tmpl
var templates embed.FS

type Generator struct {
	fdomain bool
	*fidlgen.Generator
}

func EscapeQuotes(s string) string {
	// Replace backslashes first to avoid double-escaping
	s = strings.ReplaceAll(s, "\\", "\\\\")
	s = strings.ReplaceAll(s, "\"", "\\\"")
	s = strings.ReplaceAll(s, "'", "\\'")
	s = strings.ReplaceAll(s, "\x00", "\\x00")
	return s
}

func indent(s string, spaces int) string {
	if s == "" {
		return ""
	}
	prefix := strings.Repeat(" ", spaces)
	return prefix + strings.ReplaceAll(s, "\n", "\n"+prefix)
}

func IndentNonEmpty4(s string) string {
	return indent(s, 4)
}

func IndentNonEmpty8(s string) string {
	return indent(s, 8)
}

func IndentNonEmpty12(s string) string {
	return indent(s, 12)
}

func NewGenerator(blackPath string, pyprojectTomlPath string) *Generator {
	formatter := fidlgen.NewFormatter(blackPath, "--quiet", "--config", pyprojectTomlPath, "-")

	return &Generator{
		Generator: fidlgen.NewGenerator("PythonTemplates", templates, formatter, template.FuncMap{
			"escapeQuotes":     EscapeQuotes,
			"trimSpace":        strings.TrimSpace,
			"indentNonEmpty4":  IndentNonEmpty4,
			"indentNonEmpty8":  IndentNonEmpty8,
			"indentNonEmpty12": IndentNonEmpty12,
		})}
}

func (gen *Generator) GenerateFidl(ir fidlgen.Root, outputFilename string) error {
	tree := Compile(ir)
	return gen.GenerateFile(outputFilename, "GenerateSourceFile", tree)
}