Fix orientation when page size is set up (#363)

johnfercher · Nov 6, 2023 · 7493f4c · 7493f4c
1 parent 1249492
commit 7493f4c
Showing 1 changed file with 8 additions and 7 deletions.
diff --git a/pkg/config/builder.go b/pkg/config/builder.go
@@ -53,6 +53,7 @@ type builder struct {
 	pageNumberPlace   props.Place
 	protection        *entity.Protection
 	compression       bool
+	pageSize          *pagesize.Type
 	orientation       orientation.Type
 	metadata          *entity.Metadata
 	backgroundImage   *entity.Image
@@ -83,12 +84,7 @@ func (b *builder) WithPageSize(size pagesize.Type) Builder {
 		return b
 	}
 
-	width, height := pagesize.GetDimensions(size)
-	b.dimensions = &entity.Dimensions{
-		Width:  width,
-		Height: height,
-	}
-
+	b.pageSize = &size
 	return b
 }
 
@@ -311,7 +307,12 @@ func (b *builder) getDimensions() *entity.Dimensions {
 		return b.dimensions
 	}
 
-	width, height := pagesize.GetDimensions(pagesize.A4)
+	pageSize := pagesize.A4
+	if b.pageSize != nil {
+		pageSize = *b.pageSize
+	}
+
+	width, height := pagesize.GetDimensions(pageSize)
 	dimensions := &entity.Dimensions{
 		Width:  width,
 		Height: height,