From d2ee2199addf7fd1735738bbeb53ac6706ed28d4 Mon Sep 17 00:00:00 2001
From: AlejandroSuero <asuerome@gmail.com>
Date: Fri, 3 May 2024 21:42:13 +0200
Subject: [PATCH] feat(#5): add copying to clipboard

feat: added copy action
---
 lua/freeze/init.lua | 97 ++++++++++++++++++++++++++++++++++++++++++++-
 1 file changed, 96 insertions(+), 1 deletion(-)

diff --git a/lua/freeze/init.lua b/lua/freeze/init.lua
index 7f6f883..3703de5 100644
--- a/lua/freeze/init.lua
+++ b/lua/freeze/init.lua
@@ -8,6 +8,7 @@ local freeze = {
 		theme = "default",
 		config = "base",
 		open = false,
+		copy = false,
 	},
 	output = nil,
 }
@@ -27,6 +28,10 @@ local function onReadStdOut(err, data)
 		freeze.open(freeze.output)
 		freeze.output = nil
 	end
+	if freeze.opts.copy and freeze.output ~= nil then
+		freeze.copy(freeze.output)
+		freeze.output = nil
+	end
 end
 
 ---The callback for reading stderr.
@@ -48,7 +53,7 @@ end
 local function onExit(stdout, stderr)
 	return vim.schedule_wrap(function(code, _)
 		if code == 0 then
-			vim.notify("Successfully frozen 🍦", vim.log.levels.INFO, { title = "Freeze" })
+			vim.notify("Successfully frozen 🍦 " .. freeze.opts.output, vim.log.levels.INFO, { title = "Freeze" })
 		else
 			vim.notify(stdio.stdout, vim.log.levels.ERROR, { title = "Freeze" })
 		end
@@ -145,6 +150,96 @@ function freeze.open(filename)
 	end
 end
 
+---Copy command for Windows OS
+---@param filename string
+local function copy_windows(filename)
+	local cmd = {
+		"Add-Type",
+		"-AssemblyName",
+		"System.Windows.Forms;",
+		'[Windows.Forms.Clipboard]::SetImage($([System.Drawing.Image]::FromFile("'
+			.. loop.cwd()
+			.. "/"
+			.. filename
+			.. '")))',
+	}
+	local callback = {
+		on_sterr = vim.schedule_wrap(function(_, data, _)
+			local out = table.concat(data, "\n")
+			onReadStdErr(out)
+		end),
+		on_exit = vim.schedule_wrap(function()
+			vim.notify("frozen frame has been copied to the clipboard", vim.log.levels.INFO, { title = "Freeze" })
+		end),
+	}
+	local job = vim.fn.jobstart(cmd, callback)
+	vim.fn.jobstop(job)
+end
+
+---Copy command for Mac OS
+---@param filename string
+local function copy_macos(filename)
+	-- osascript -e 'set the clipboard to (read (POSIX file "/Users/aome/.dotfiles/freeze.png") as JPEG picture)'
+	local cmd = {
+		"osascript",
+		"-e",
+		"'set the clipboad to (read (POSIX file \"" .. loop.cwd() .. "/" .. filename .. "\") as JPEG picture)'",
+	}
+	local callback = {
+		on_sterr = vim.schedule_wrap(function(_, data, _)
+			local out = table.concat(data, "\n")
+			onReadStdErr(out)
+		end),
+		on_exit = vim.schedule_wrap(function()
+			vim.notify("frozen frame has been copied to the clipboard", vim.log.levels.INFO, { title = "Freeze" })
+		end),
+	}
+	local job = vim.fn.jobstart(cmd, callback)
+	vim.fn.jobstop(job)
+end
+
+---Copy command for Unix OS
+---@param filename string
+local function copy_unix(filename)
+	if vim.fn.exepath("xclip") == "" then
+		vim.notify("`xclip` is not installed", vim.log.level.ERROR, { title = "Freeze" })
+		return
+	end
+	local cmd = {
+		"xclip",
+		"-selection",
+		"clipboard",
+		"-t",
+		"image/png",
+		"-i",
+		loop.cwd() .. "/" .. filename,
+	}
+	local callback = {
+		on_sterr = vim.schedule_wrap(function(_, data, _)
+			local out = table.concat(data, "\n")
+			onReadStdErr(out)
+		end),
+		on_exit = vim.schedule_wrap(function()
+			vim.notify("frozen frame has been copied to the clipboard", vim.log.levels.INFO, { title = "Freeze" })
+		end),
+	}
+	local job = vim.fn.jobstart(cmd, callback)
+	vim.fn.jobstop(job)
+end
+
+---Copy the frozen frame to the clipboard
+---@param filename string
+function freeze.copy(filename)
+	local os = vim.loop.os_uname().sysname
+
+	if os == "Windows" or os == "Window_NT" then
+		copy_windows(filename)
+	elseif os == "Darwin" then
+		copy_macos(filename)
+	end
+	copy_unix(filename)
+end
+
 --- Setup function for enabling both user commands.
 --- Sets up :Freeze for freezing a selection and :FreezeLine
 --- to freeze a single line.