mirror of
https://github.com/hanzoai/runtime.git
synced 2026-10-30 18:23:13 +00:00
zeekay
0c7efccf40
- Add main Dockerfile for production builds - Add Dockerfile.runtime for secure code execution - Add Dockerfile.runtime.fixed with improved configuration - Include .dockerignore for build optimization - Add integration documentation and publishing guides - Update API and CLI components for better runtime support - Enhance sandbox and container management features
110 lines
2.0 KiB
Go
110 lines
2.0 KiB
Go
// Copyright 2025 Hanzo Industries Inc.
|
|
// SPDX-License-Identifier: AGPL-3.0
|
|
|
|
package common
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
"os"
|
|
|
|
"github.com/spf13/cobra"
|
|
"gopkg.in/yaml.v2"
|
|
)
|
|
|
|
const (
|
|
formatFlagDescription = `Output format. Must be one of (yaml, json)`
|
|
formatFlagName = "format"
|
|
formatFlagShortHand = "f"
|
|
)
|
|
|
|
var (
|
|
FormatFlag string
|
|
standardOut *os.File
|
|
)
|
|
|
|
type outputFormatter struct {
|
|
data interface{}
|
|
formatter Formatter
|
|
}
|
|
|
|
func NewFormatter(data interface{}) *outputFormatter {
|
|
var formatter Formatter
|
|
switch FormatFlag {
|
|
case "json":
|
|
formatter = JSONFormatter{}
|
|
case "yaml":
|
|
formatter = YAMLFormatter{}
|
|
case "":
|
|
formatter = nil
|
|
default:
|
|
formatter = JSONFormatter{} // Default to JSON
|
|
}
|
|
|
|
return &outputFormatter{
|
|
data: data,
|
|
formatter: formatter,
|
|
}
|
|
|
|
}
|
|
|
|
type Formatter interface {
|
|
Format(data interface{}) (string, error)
|
|
}
|
|
|
|
type JSONFormatter struct{}
|
|
|
|
func (f JSONFormatter) Format(data interface{}) (string, error) {
|
|
jsonData, err := json.MarshalIndent(data, "", " ") // Indent with two spaces
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
return string(jsonData), nil
|
|
}
|
|
|
|
type YAMLFormatter struct{}
|
|
|
|
func (f YAMLFormatter) Format(data interface{}) (string, error) {
|
|
yamlData, err := yaml.Marshal(data)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
return string(yamlData), nil
|
|
}
|
|
|
|
func (f *outputFormatter) Print() {
|
|
|
|
formattedOutput, err := f.formatter.Format(f.data)
|
|
if err != nil {
|
|
fmt.Printf("Error formatting output: %v\n", err)
|
|
os.Exit(1)
|
|
}
|
|
|
|
UnblockStdOut()
|
|
fmt.Println(formattedOutput)
|
|
BlockStdOut()
|
|
}
|
|
|
|
func BlockStdOut() {
|
|
if os.Stdout != nil {
|
|
standardOut = os.Stdout
|
|
os.Stdout = nil
|
|
}
|
|
}
|
|
|
|
func UnblockStdOut() {
|
|
if os.Stdout == nil {
|
|
os.Stdout = standardOut
|
|
standardOut = nil
|
|
}
|
|
}
|
|
|
|
func RegisterFormatFlag(cmd *cobra.Command) {
|
|
cmd.Flags().StringVarP(&FormatFlag, formatFlagName, formatFlagShortHand, FormatFlag, formatFlagDescription)
|
|
cmd.PreRun = func(cmd *cobra.Command, args []string) {
|
|
if FormatFlag != "" {
|
|
BlockStdOut()
|
|
}
|
|
}
|
|
}
|