feat(#58): add copy to clipboard flag --copy

README.md

@@ -114,6 +114,7 @@ Screenshots can be customized with `--flags` or [Configuration](#configuration)
 - [`--font.file`](#font): File path to the font to use (embedded in the SVG).
 - [`--line-height`](#font): Line height relative to font size.
 - [`--show-line-numbers`](#line-numbers): Show line numbers.
+- [`--copy`](#copy): Copy the output image to the clipboard.
 - [`--lines`](#line-numbers): Lines to capture (start,end).
 
 ### Language
@@ -159,6 +160,14 @@ freeze main.go --output out.webp
 freeze main.go --output out.{svg,png,webp}
 ```
 
+### Copy
+
+Copy the output image to your clipboard, so you can paste it anywhere.
+
+```bash
+freeze main.go --copy
+```
+
 ### Font
 
 Specify the font family, font size, and font line height of the output image.

config.go

@@ -31,6 +31,7 @@ type Config struct {
 	Language    string `json:"language,omitempty" help:"Language of code file." short:"l" group:"Settings" placeholder:"go"`
 	Theme       string `json:"theme" help:"Theme to use for syntax highlighting." short:"t" group:"Settings" placeholder:"charm"`
 
+	Copy           bool          `json:"copy" help:"Copy the output image to the clipboard." group:"Settings"`
 	Output         string        `json:"output,omitempty" help:"Output location for {{.svg}}, {{.png}}, or {{.webp}}." short:"o" group:"Settings" default:"" placeholder:"freeze.svg"`
 	Execute        string        `json:"-" help:"Capture output of command execution." short:"x" group:"Settings" default:""`
 	ExecuteTimeout time.Duration `json:"-" help:"Execution timeout." group:"Settings" default:"10s" prefix:"execute." name:"timeout" hidden:""`

configurations/full.json

@@ -29,5 +29,6 @@
     "size": 14,
     "ligatures": true
   },
-  "line_height": 1.2
-}
+  "line_height": 1.2,
+  "copy": false
+}

freeze_test.go

@@ -10,6 +10,7 @@ import (
 	"testing"
 
 	"github.com/aymanbagabas/go-udiff"
+	"golang.design/x/clipboard"
 )
 
 const binary = "./test/freeze-test"
@@ -56,6 +57,31 @@ func TestFreezeOutput(t *testing.T) {
 	}
 }
 
+func TestFreezeCopy(t *testing.T) {
+	output := "artichoke-test.svg"
+	defer os.Remove(output)
+
+	cmd := exec.Command(binary, "test/input/artichoke.hs", "-o", output, "--copy")
+	err := cmd.Run()
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	_, err = os.Stat(output)
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	err = clipboard.Init()
+	if err != nil {
+		t.Fatal(err)
+	}
+	png := clipboard.Read(clipboard.FmtImage)
+	if png == nil {
+		t.Fatal("clipboard is empty")
+	}
+}
+
 func TestFreezeHelp(t *testing.T) {
 	out := bytes.Buffer{}
 	cmd := exec.Command(binary)
@@ -134,6 +160,11 @@ func TestFreezeConfigurations(t *testing.T) {
 			flags:  []string{"--language", "go", "--height", "800", "--width", "750", "--config", "full", "--window=false", "--show-line-numbers"},
 			output: "bubbletea",
 		},
+		{
+			input:  "test/input/bubbletea.model",
+			flags:  []string{"--language", "go", "--height", "800", "--width", "750", "--config", "full", "--window=false", "--show-line-numbers", "--copy"},
+			output: "bubbletea-copy",
+		},
 		// {
 		// 	flags:  []string{"--execute", "layout", "--height", "800", "--config", "full", "--margin", "50,10"},
 		// 	output: "composite-2",

go.mod

@@ -17,6 +17,7 @@ require (
 	github.com/kanrichan/resvg-go v0.0.2-0.20231001163256-63db194ca9f5
 	github.com/mattn/go-isatty v0.0.20
 	github.com/mattn/go-runewidth v0.0.15
+	golang.design/x/clipboard v0.7.0
 	golang.org/x/sys v0.18.0
 )
 
@@ -41,6 +42,9 @@ require (
 	github.com/rivo/uniseg v0.4.7 // indirect
 	github.com/tetratelabs/wazero v1.7.0 // indirect
 	golang.org/x/exp v0.0.0-20240325151524-a685a6edb6d8 // indirect
+	golang.org/x/exp/shiny v0.0.0-20240506185415-9bf2ced13842 // indirect
+	golang.org/x/image v0.14.0 // indirect
+	golang.org/x/mobile v0.0.0-20231127183840-76ac6878050a // indirect
 	golang.org/x/sync v0.6.0 // indirect
 	golang.org/x/term v0.18.0 // indirect
 	golang.org/x/text v0.14.0 // indirect

go.sum

@@ -82,8 +82,16 @@ github.com/stretchr/testify v1.9.0 h1:HtqpIVDClZ4nwg75+f6Lvsy/wHu+3BoSGCbBAcpTsT
 github.com/stretchr/testify v1.9.0/go.mod h1:r2ic/lqez/lEtzL7wO/rwa5dbSLXVDPFyf8C91i36aY=
 github.com/tetratelabs/wazero v1.7.0 h1:jg5qPydno59wqjpGrHph81lbtHzTrWzwwtD4cD88+hQ=
 github.com/tetratelabs/wazero v1.7.0/go.mod h1:ytl6Zuh20R/eROuyDaGPkp82O9C/DJfXAwJfQ3X6/7Y=
+golang.design/x/clipboard v0.7.0 h1:4Je8M/ys9AJumVnl8m+rZnIvstSnYj1fvzqYrU3TXvo=
+golang.design/x/clipboard v0.7.0/go.mod h1:PQIvqYO9GP29yINEfsEn5zSQKAz3UgXmZKzDA6dnq2E=
 golang.org/x/exp v0.0.0-20240325151524-a685a6edb6d8 h1:aAcj0Da7eBAtrTp03QXWvm88pSyOt+UgdZw2BFZ+lEw=
 golang.org/x/exp v0.0.0-20240325151524-a685a6edb6d8/go.mod h1:CQ1k9gNrJ50XIzaKCRR2hssIjF07kZFEiieALBM/ARQ=
+golang.org/x/exp/shiny v0.0.0-20240506185415-9bf2ced13842 h1:kEvPiBVeT1JJGw/3THfe1W1zvTAvU1V6pCFV0icZvQs=
+golang.org/x/exp/shiny v0.0.0-20240506185415-9bf2ced13842/go.mod h1:3F+MieQB7dRYLTmnncoFbb1crS5lfQoTfDgQy6K4N0o=
+golang.org/x/image v0.14.0 h1:tNgSxAFe3jC4uYqvZdTr84SZoM1KfwdC9SKIFrLjFn4=
+golang.org/x/image v0.14.0/go.mod h1:HUYqC05R2ZcZ3ejNQsIHQDQiwWM4JBqmm6MKANTp4LE=
+golang.org/x/mobile v0.0.0-20231127183840-76ac6878050a h1:sYbmY3FwUWCBTodZL1S3JUuOvaW6kM2o+clDzzDNBWg=
+golang.org/x/mobile v0.0.0-20231127183840-76ac6878050a/go.mod h1:Ede7gF0KGoHlj822RtphAHK1jLdrcuRBZg0sF1Q+SPc=
 golang.org/x/sync v0.6.0 h1:5BMeUDZ7vkXGfEr1x9B4bRcTH4lpkTkpdh0T/J+qjbQ=
 golang.org/x/sync v0.6.0/go.mod h1:Czt+wKu1gCyEFDUtn0jG5QVvpJ6rzVqr5aXyt9drQfk=
 golang.org/x/sys v0.0.0-20211025201205-69cdffdb9359/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=

interactive.go

@@ -67,6 +67,11 @@ func runForm(config *Config) (*Config, error) {
 				Prompt("").
 				Value(&config.Output),
 
+			huh.NewConfirm().Title("Clipboard").
+				// Description("Copy the output image to the clipboard.").
+				Inline(true).
+				Value(&config.Copy),
+
 			huh.NewSelect[string]().Title("Theme ").
 				// Description("Theme for syntax highlighting.").
 				Inline(true).

main.go

@@ -398,7 +398,7 @@ func main() {
 		}
 
 		// could not convert with libsvg, try resvg
-		svgConversionErr = resvgConvert(doc, imageWidth, imageHeight, config.Output)
+		svgConversionErr = resvgConvert(doc, imageWidth, imageHeight, config.Output, config.Copy)
 		if svgConversionErr != nil {
 			printErrorFatal("Unable to convert SVG to PNG", svgConversionErr)
 		}

png.go

@@ -9,6 +9,7 @@ import (
 	"github.com/beevik/etree"
 	"github.com/charmbracelet/freeze/font"
 	"github.com/kanrichan/resvg-go"
+	"golang.design/x/clipboard"
 )
 
 func libsvgConvert(doc *etree.Document, w, h float64, output string) error {
@@ -30,7 +31,7 @@ func libsvgConvert(doc *etree.Document, w, h float64, output string) error {
 	return err
 }
 
-func resvgConvert(doc *etree.Document, w, h float64, output string) error {
+func resvgConvert(doc *etree.Document, w, h float64, output string, copy bool) error {
 	svg, err := doc.WriteToBytes()
 	if err != nil {
 		return err
@@ -94,5 +95,12 @@ func resvgConvert(doc *etree.Document, w, h float64, output string) error {
 	if err != nil {
 		return err
 	}
+	if copy {
+		err = clipboard.Init()
+		if err != nil {
+			return err
+		}
+		clipboard.Write(clipboard.FmtImage, png)
+	}
 	return err
 }