Snider
adaa4131f9
- Fix remaining 187 pkg/ files referencing core/cli → core/go - Move SDK library code from internal/cmd/sdk/ → pkg/sdk/ (new package) - Create pkg/rag/helpers.go with convenience functions from internal/cmd/rag/ - Fix pkg/mcp/tools_rag.go to use pkg/rag instead of internal/cmd/rag - Fix pkg/build/buildcmd/cmd_sdk.go and pkg/release/sdk.go to use pkg/sdk - Remove all non-library content: main.go, internal/, cmd/, docker/, scripts/, tasks/, tools/, .core/, .forgejo/, .woodpecker/, Taskfile.yml - Run go mod tidy to trim unused dependencies core/go is now a pure Go package suite (library only). Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com> Co-authored-by: Claude <developers@lethean.io> Reviewed-on: #3
82 lines
2.1 KiB
Go
82 lines
2.1 KiB
Go
package generators
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"os"
|
|
"os/exec"
|
|
"path/filepath"
|
|
|
|
coreio "forge.lthn.ai/core/go/pkg/io"
|
|
)
|
|
|
|
// PythonGenerator generates Python SDKs from OpenAPI specs.
|
|
type PythonGenerator struct{}
|
|
|
|
// NewPythonGenerator creates a new Python generator.
|
|
func NewPythonGenerator() *PythonGenerator {
|
|
return &PythonGenerator{}
|
|
}
|
|
|
|
// Language returns the generator's target language identifier.
|
|
func (g *PythonGenerator) Language() string {
|
|
return "python"
|
|
}
|
|
|
|
// Available checks if generator dependencies are installed.
|
|
func (g *PythonGenerator) Available() bool {
|
|
_, err := exec.LookPath("openapi-python-client")
|
|
return err == nil
|
|
}
|
|
|
|
// Install returns instructions for installing the generator.
|
|
func (g *PythonGenerator) Install() string {
|
|
return "pip install openapi-python-client"
|
|
}
|
|
|
|
// Generate creates SDK from OpenAPI spec.
|
|
func (g *PythonGenerator) Generate(ctx context.Context, opts Options) error {
|
|
if err := coreio.Local.EnsureDir(opts.OutputDir); err != nil {
|
|
return fmt.Errorf("python.Generate: failed to create output dir: %w", err)
|
|
}
|
|
|
|
if g.Available() {
|
|
return g.generateNative(ctx, opts)
|
|
}
|
|
return g.generateDocker(ctx, opts)
|
|
}
|
|
|
|
func (g *PythonGenerator) generateNative(ctx context.Context, opts Options) error {
|
|
parentDir := filepath.Dir(opts.OutputDir)
|
|
|
|
cmd := exec.CommandContext(ctx, "openapi-python-client", "generate",
|
|
"--path", opts.SpecPath,
|
|
"--output-path", opts.OutputDir,
|
|
)
|
|
cmd.Dir = parentDir
|
|
cmd.Stdout = os.Stdout
|
|
cmd.Stderr = os.Stderr
|
|
return cmd.Run()
|
|
}
|
|
|
|
func (g *PythonGenerator) generateDocker(ctx context.Context, opts Options) error {
|
|
specDir := filepath.Dir(opts.SpecPath)
|
|
specName := filepath.Base(opts.SpecPath)
|
|
|
|
args := []string{"run", "--rm"}
|
|
args = append(args, dockerUserArgs()...)
|
|
args = append(args,
|
|
"-v", specDir+":/spec",
|
|
"-v", opts.OutputDir+":/out",
|
|
"openapitools/openapi-generator-cli", "generate",
|
|
"-i", "/spec/"+specName,
|
|
"-g", "python",
|
|
"-o", "/out",
|
|
"--additional-properties=packageName="+opts.PackageName,
|
|
)
|
|
|
|
cmd := exec.CommandContext(ctx, "docker", args...)
|
|
cmd.Stdout = os.Stdout
|
|
cmd.Stderr = os.Stderr
|
|
return cmd.Run()
|
|
}
|