62 lines
1.5 KiB
Go
62 lines
1.5 KiB
Go
package d2cli
|
|
|
|
import (
|
|
"fmt"
|
|
"path/filepath"
|
|
"strings"
|
|
)
|
|
|
|
type exportExtension string
|
|
|
|
const GIF exportExtension = ".gif"
|
|
const PNG exportExtension = ".png"
|
|
const PPTX exportExtension = ".pptx"
|
|
const PDF exportExtension = ".pdf"
|
|
const SVG exportExtension = ".svg"
|
|
|
|
var SUPPORTED_EXTENSIONS = []exportExtension{SVG, PNG, PDF, PPTX, GIF}
|
|
|
|
var STDOUT_FORMAT_MAP = map[string]exportExtension{
|
|
"png": PNG,
|
|
"svg": SVG,
|
|
}
|
|
|
|
var SUPPORTED_STDOUT_FORMATS = []string{"png", "svg"}
|
|
|
|
func getOutputFormat(stdoutFormatFlag *string, outputPath string) (exportExtension, error) {
|
|
if stdoutFormatFlag != nil && *stdoutFormatFlag != "" {
|
|
format := strings.ToLower(*stdoutFormatFlag)
|
|
if ext, ok := STDOUT_FORMAT_MAP[format]; ok {
|
|
return ext, nil
|
|
}
|
|
return "", fmt.Errorf("%s is not a supported format. Supported formats are: %s", *stdoutFormatFlag, SUPPORTED_STDOUT_FORMATS)
|
|
}
|
|
return getExportExtension(outputPath), nil
|
|
}
|
|
|
|
func getExportExtension(outputPath string) exportExtension {
|
|
ext := filepath.Ext(outputPath)
|
|
for _, kext := range SUPPORTED_EXTENSIONS {
|
|
if kext == exportExtension(ext) {
|
|
return exportExtension(ext)
|
|
}
|
|
}
|
|
// default is svg
|
|
return exportExtension(SVG)
|
|
}
|
|
|
|
func (ex exportExtension) supportsAnimation() bool {
|
|
return ex == SVG || ex == GIF
|
|
}
|
|
|
|
func (ex exportExtension) requiresAnimationInterval() bool {
|
|
return ex == GIF
|
|
}
|
|
|
|
func (ex exportExtension) requiresPNGRenderer() bool {
|
|
return ex == PNG || ex == PDF || ex == PPTX || ex == GIF
|
|
}
|
|
|
|
func (ex exportExtension) supportsDarkTheme() bool {
|
|
return ex == SVG
|
|
}
|