add left to right option

cmd/generate.go

@@ -16,6 +16,7 @@ var (
 	directories []string
 	files       []string
 	exclusion   string
+	leftToRight bool
 	recursive   bool
 	generateCmd = &cobra.Command{
 		Use:   "generate",
@@ -27,7 +28,7 @@ var (
 				packages = append(packages, astParser.ParseFile(file))
 			}
 
-			options := []astParser.ParserOptionFunc{}
+			var options []astParser.ParserOptionFunc
 			if recursive {
 				options = append(options, astParser.WithRecursive())
 			}
@@ -40,7 +41,8 @@ var (
 				packages = append(packages, astParser.ParseDirectory(directory, options...)...)
 			}
 
-			formattedPlantUML := formatter.FormatPlantUML(packages)
+			var formatterOptions = formatter.FormatterOptions{LeftToRight: leftToRight}
+			formattedPlantUML := formatter.FormatPlantUML(packages, formatterOptions)
 			err := os.WriteFile(outPath, []byte(formattedPlantUML), 0644)
 			if err != nil {
 				log.Fatal(err)
@@ -78,6 +80,13 @@ func init() {
 		false,
 		"traverse the given directories recursively",
 	)
+	generateCmd.Flags().BoolVarP(
+		&leftToRight,
+		"left-to-right",
+		"l",
+		false,
+		"display the UML diagram left to right (default: top to bottom)",
+	)
 	generateCmd.Flags().StringVarP(
 		&exclusion,
 		"exclude",

formatter/formatterOptions.go

@@ -0,0 +1,5 @@
+package formatter
+
+type FormatterOptions struct {
+	LeftToRight bool
+}

formatter/plantUml.go

@@ -29,6 +29,9 @@ const PlantUMLWrapper = `@startuml
 %s
 @enduml`
 
+const PlantUMLTopToBottomDirection = "top to bottom direction"
+const PlantUMLLeftToRightDirection = "left to right direction"
+
 func FormatField(field domain.Field) string {
 	visibilityCharacter := "+"
 	if field.IsPrivate() {
@@ -188,14 +191,20 @@ func FormatClasses(classes domain.Classes) string {
 	return strings.Join(formattedClasses, "\n")
 }
 
-func FormatPlantUML(packages domain.Packages) string {
+func FormatPlantUML(packages domain.Packages, formatterOptions FormatterOptions) string {
+	var direction = PlantUMLTopToBottomDirection
+
+	if formatterOptions.LeftToRight {
+		direction = PlantUMLLeftToRightDirection
+	}
+
 	formattedPackages := FormatPackages(packages)
 	formattedRelations := FormatRelations(packages.AllClasses())
 	formattedImplementationRelations := FormatImplementationRelations(
 		packages.AllClasses(),
 		packages.AllInterfaces(),
 	)
-	return FormatPlantUMLWrapper(formattedPackages, formattedRelations, formattedImplementationRelations)
+	return FormatPlantUMLWrapper(direction, formattedPackages, formattedRelations, formattedImplementationRelations)
 }
 
 func FormatRelation(class domain.Class, class2 domain.Class) string {

formatter/plantUml_test.go

@@ -175,10 +175,17 @@ func TestFormatClasses(t *testing.T) {
 	assert.Equal(t, expected, result)
 }
 
-func TestFormatPlantUML(t *testing.T) {
+func TestFormatPlantUML_default(t *testing.T) {
 	packages := domain.Packages{}
-	expected := "@startuml\n\n\n\n@enduml"
-	result := FormatPlantUML(packages)
+	expected := "@startuml\ntop to bottom direction\n\n\n\n@enduml"
+	result := FormatPlantUML(packages, FormatterOptions{LeftToRight: false})
+	assert.Equal(t, expected, result)
+}
+
+func TestFormatPlantUML_left_to_right(t *testing.T) {
+	packages := domain.Packages{}
+	expected := "@startuml\nleft to right direction\n\n\n\n@enduml"
+	result := FormatPlantUML(packages, FormatterOptions{LeftToRight: true})
 	assert.Equal(t, expected, result)
 }