feat(#89): added input file as window title

README.md

@@ -103,6 +103,8 @@ Screenshots can be customized with `--flags` or [Configuration](#configuration)
 - [`-t`](#theme), [`--theme`](#theme): Theme to use for syntax highlighting.
 - [`-w`](#window), [`--window`](#window): Display window controls.
 - [`-H`](#height), [`--height`](#height): Height of terminal window.
+- [`--title.text`](#window-title): Display input file or custom text when window controls are displayed.
+- [`--title.position`](#window-title): Position of the title text.
 - [`--border.width`](#border-width): Border width thickness.
 - [`--border.color`](#border-width): Border color.
 - [`--shadow.blur`](#shadow): Shadow Gaussian Blur.
@@ -216,6 +218,26 @@ freeze artichoke.hs --window
   <img alt="output of freeze command, Haskell code block with window controls applied" src="./test/golden/svg/window.svg" width="600" />
 </a>
 
+#### Window Title
+
+Display the input file as the title of the window if `--title.text` is not passed in the arguments (`--title.text=auto` as default).
+
+Display a custom title if `--title.text` is passed in the arguments (`--title.text="custom title"`).
+
+Don't display the title if `--title.text` is passed in the arguments as empty string (`--title.text=""`).
+
+To position the title text, use `--title.position` with `left`, `center` or `right` (`--title.position=center` as default).
+
+```bash
+freeze artichoke.hs --window --title.text "My artichoke code"
+```
+
+<img alt="output of freeze command, Haskell code block with window title applied" src="./test/golden/svg/title.svg" width="600" />
+
+> [!WARNING]
+>
+> The **title** can only be supported when using **window** mode (`--window=true`).
+
 ### Background
 
 Set the background color of the terminal window.

config.go

@@ -22,6 +22,7 @@ type Config struct {
 	Margin     []float64 `json:"margin" help:"Apply margin to the window." short:"m" placeholder:"0" group:"Window"`
 	Padding    []float64 `json:"padding" help:"Apply padding to the code." short:"p" placeholder:"0" group:"Window"`
 	Window     bool      `json:"window" help:"Display window controls." group:"Window"`
+	Title      Title     `json:"title" embed:"" prefix:"title." group:"Window"`
 	Width      float64   `json:"width" help:"Width of terminal window." short:"W" group:"Window"`
 	Height     float64   `json:"height" help:"Height of terminal window." short:"H" group:"Window"`
 
@@ -49,6 +50,11 @@ type Config struct {
 	ShowLineNumbers bool    `json:"show_line_numbers" help:"" group:"Line" placeholder:"false"`
 }
 
+type Title struct {
+	Text     string `json:"title" help:"Display window title. {{--title.text=auto}} as default for input filename." default:"auto"`
+	Position string `json:"position" help:"Position of window title, one of {{left}}, {{center}}, or {{right}}. {{--title.position=center}} as default." default:"center"`
+}
+
 // Shadow is the configuration options for a drop shadow.
 type Shadow struct {
 	Blur float64 `json:"blur" help:"Shadow Gaussian Blur." placeholder:"0"`

configurations/full.json

@@ -1,5 +1,9 @@
 {
   "window": true,
+	"title": {
+		"text": "auto",
+		"position": "center"
+	},
   "theme": "charm",
   "border": {
     "radius": 8,
@@ -30,4 +34,4 @@
     "ligatures": true
   },
   "line_height": 1.2
-}
+}

freeze_test.go

@@ -10,6 +10,7 @@ import (
 	"testing"
 
 	"github.com/aymanbagabas/go-udiff"
+	"github.com/beevik/etree"
 )
 
 const binary = "./test/freeze-test"
@@ -104,6 +105,62 @@ func TestFreezeErrorFileMissing(t *testing.T) {
 	}
 }
 
+func TestFreezeWindowTitleFilename(t *testing.T) {
+	output := "artichoke-default-title.svg"
+	t.Cleanup(func() { os.Remove(output) })
+	testTitle := "artichoke.hs"
+	cmd := exec.Command(binary, "test/input/artichoke.hs", "--output", output, "--window")
+	err := cmd.Run()
+
+	if err != nil {
+		t.Fatal("unexpected error", err)
+	}
+
+	doc := etree.NewDocument()
+	err = doc.ReadFromFile(output)
+	if err != nil {
+		t.Fatal("unexpected error", err)
+	}
+	childs := doc.ChildElements()
+	if len(childs) == 0 {
+		t.Fatal("no child elements")
+	}
+	lastChild := childs[len(childs)-1]
+	got := lastChild.FindElement("text").Text()
+
+	if got != testTitle {
+		t.Fatalf("expected %s to be %s", got, testTitle)
+	}
+}
+
+func TestFreezeCustomWindowTitle(t *testing.T) {
+	output := "artichoke-custom-title.svg"
+	t.Cleanup(func() { os.Remove(output) })
+	testTitle := "custom-test title"
+	cmd := exec.Command(binary, "test/input/artichoke.hs", "--output", output, "--title.text", testTitle, "--window")
+	err := cmd.Run()
+
+	if err != nil {
+		t.Fatal("unexpected error", err)
+	}
+
+	doc := etree.NewDocument()
+	err = doc.ReadFromFile(output)
+	if err != nil {
+		t.Fatal("unexpected error", err)
+	}
+	childs := doc.ChildElements()
+	if len(childs) == 0 {
+		t.Fatal("no child elements")
+	}
+	lastChild := childs[len(childs)-1]
+	got := lastChild.FindElement("text").Text()
+
+	if got != testTitle {
+		t.Fatalf("expected %s to be %s", got, testTitle)
+	}
+}
+
 func TestFreezeConfigurations(t *testing.T) {
 	tests := []struct {
 		input  string
@@ -258,6 +315,11 @@ func TestFreezeConfigurations(t *testing.T) {
 			flags:  []string{"--wrap", "80", "--width", "600"},
 			output: "wrap",
 		},
+		{
+			input:  "test/input/artichoke.hs",
+			flags:  []string{"--border.radius", "8", "--window", "--title.text", "My artichoke code"},
+			output: "title",
+		},
 	}
 
 	err := os.RemoveAll("test/output/svg")

interactive.go

@@ -103,6 +103,17 @@ func runForm(config *Config) (*Config, error) {
 				Inline(true).
 				Value(&config.Window),
 
+			huh.NewInput().Title("Title text").
+				Placeholder("auto").
+				Inline(true).
+				Prompt("").
+				Value(&config.Title.Text),
+
+			huh.NewSelect[string]().Title("Title position").
+				Inline(true).
+				Options(huh.NewOptions("left", "center", "right")...).
+				Value(&config.Title.Position),
+
 			huh.NewNote().Title("Font"),
 
 			huh.NewInput().Title("Font Family ").

main.go

@@ -287,10 +287,28 @@ func main() {
 	}
 
 	if config.Window {
-		windowControls := svg.NewWindowControls(5.5*float64(scale), 19.0*scale, 12.0*scale)
+		x := 19.0 * scale
+		y := 12.0 * scale
+		windowControls := svg.NewWindowControls(5.5*float64(scale), x, y)
 		svg.Move(windowControls, float64(config.Margin[left]), float64(config.Margin[top]))
 		image.AddChild(windowControls)
 		config.Padding[top] += (15 * scale)
+		if config.Title.Text != "" {
+			windowChilds := windowControls.ChildElements()
+			if len(windowChilds) == 0 {
+				printErrorFatal("Unable to add title", errors.New("no window controls found"))
+			}
+			controlsWidth := float64(len(windowChilds)) * float64(x)
+			titlePos := getPositions(config, x, y, imageWidth, controlsWidth)
+			title, err := NewWindowTitle(config, titlePos, scale, s)
+			if err != nil {
+				printErrorFatal("Unable to add title", err)
+			}
+			image.AddChild(title)
+		}
+	} else if config.Title.Text != "auto" || config.Title.Position != "center" {
+		err := errors.New("Title is not supported when not using a window controls")
+		printErrorFatal("Unable to add title", err)
 	}
 
 	if config.Border.Radius > 0 {
@@ -455,3 +473,67 @@ var outputHeader = lipgloss.NewStyle().Foreground(lipgloss.Color("#F1F1F1")).Bac
 func printFilenameOutput(filename string) {
 	fmt.Println(lipgloss.JoinHorizontal(lipgloss.Center, outputHeader.String(), filename))
 }
+
+type Positions struct {
+	Left   float64
+	Center float64
+	Right  float64
+	Top    float64
+}
+
+func getPositions(config Config, x, y, imgW, controlsWidth float64) Positions {
+	return Positions{
+		Left:   config.Margin[left] + x + controlsWidth,
+		Center: imgW / 2,
+		Right:  imgW - (config.Margin[right] + x),
+		Top:    config.Margin[top] + y,
+	}
+}
+
+func (title Title) Validate(value string) error {
+	if title.Text == "" || title.Text == "-" {
+		return errors.New("Invalid title provided")
+	}
+	if title.Position != "left" && title.Position != "center" && title.Position != "right" {
+		return errors.New("Invalid title position. Must be one of \"left\", \"center\", or \"right\"")
+	}
+	return nil
+}
+
+// NewWindowTitle returns a title element with the given text.
+func NewWindowTitle(config Config, positions Positions, scale float64, s *chroma.Style) (*etree.Element, error) {
+	titleText := config.Title.Text
+	err := config.Title.Validate(titleText)
+	if err != nil {
+		return nil, err
+	}
+	if titleText == "auto" {
+		titleText = filepath.Base(config.Input)
+	}
+	x := 0.0
+	y := positions.Top
+	var anchor string
+	switch config.Title.Position {
+	case "left":
+		x = positions.Left
+		anchor = "start"
+		break
+	case "center":
+		x = positions.Center
+		anchor = "middle"
+		break
+	case "right":
+		x = positions.Right
+		anchor = "end"
+		break
+	}
+	input := etree.NewElement("text")
+	input.CreateAttr("font-size", fmt.Sprintf("%.2fpx", config.Font.Size*float64(scale)))
+	input.CreateAttr("fill", s.Get(chroma.Text).Colour.String())
+	input.CreateAttr("font-family", config.Font.Family)
+	input.CreateAttr("text-anchor", anchor)
+	input.CreateAttr("alignment-baseline", "middle")
+	input.SetText(titleText)
+	svg.Move(input, float64(x), float64(y))
+	return input, err
+}

test/configurations/full.json

@@ -1,5 +1,9 @@
 {
   "window": true,
+	"title": {
+		"text": "auto",
+		"position": "center"
+	},
   "theme": "charm",
   "border": {
     "radius": 8,
@@ -24,4 +28,4 @@
     "size": 14
   },
   "line_height": 1.2
-}
+}