From 526dba6c12617d18ce7bbea549c91c48139e62cf Mon Sep 17 00:00:00 2001
From: yutent <yutent.io@gmail.com>
Date: Wed, 9 Aug 2023 19:50:10 +0800
Subject: [PATCH] Create shotme

---
 shotme | 123 +++++++++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 123 insertions(+)
 create mode 100644 shotme

diff --git a/shotme b/shotme
new file mode 100644
index 0000000..001956c
--- /dev/null
+++ b/shotme
@@ -0,0 +1,123 @@
+#!/usr/bin/env python3
+
+
+import gi, sys, os
+import dbus
+import dbus.service, dbus.mainloop.glib
+
+gi.require_version('Gtk', '3.0')
+
+from gi.repository import Gtk, Gdk, GLib, GdkPixbuf
+from PIL import Image, ImageFilter, ImageChops
+
+app_id = 'fun.wkit.shotme'
+
+
+def pixbuf_to_pil(pixbuf):
+  data = pixbuf.get_pixels()
+  w = pixbuf.get_width()
+  h = pixbuf.get_height()
+  stride = pixbuf.get_rowstride()
+  mode = "RGBA" if pixbuf.get_has_alpha() else "RGB"
+  img = Image.frombytes(mode, (w, h), data, "raw", mode, stride)
+  return img
+
+
+def pil_to_pixbuf(img):
+  data = []
+  for p in img.getdata():
+    data.extend(p)
+  data = bytes(data)
+  alpha = img.mode == 'RGBA'
+  w, h = img.size
+  rowstride = w * 4
+  return GdkPixbuf.Pixbuf.new_from_bytes(GLib.Bytes.new(data), GdkPixbuf.Colorspace.RGB, alpha, 8, w, h, rowstride)
+
+
+def add_shadow(img):
+  # 加载图片并确认该图片为RGBA模式，保证透明度
+  img = pixbuf_to_pil(img).convert('RGBA')
+  w,h = img.size
+
+  w += 64
+  h += 64
+
+  shadow = Image.new('RGBA', img.size, (80,80,80,100))
+
+  new_img = Image.new('RGBA', (w,h), (0,0,0,0))
+  # new_img = Image.new('RGBA', (w,h), (255,255,255,255))
+  new_img.paste(shadow, (32, 32))
+
+  new_img = new_img.filter(ImageFilter.GaussianBlur(radius = 12))
+  new_img.paste(img, (32, 32))
+
+  return pil_to_pixbuf(new_img)
+
+
+class Application(Gtk.Application):
+  def __init__(self):
+    Gtk.Application.__init__(self, application_id = app_id)
+
+    display = Gdk.Display.get_default()
+    self.clipboard = Gtk.Clipboard.get(Gdk.SELECTION_CLIPBOARD)
+
+  
+  def do_activate(self):
+    self.shot_the_window()
+ 
+
+  def copy(self, image):
+    
+    # 写图片进剪切板
+    self.clipboard.set_image(image)
+    self.clipboard.store()
+
+
+  def shot_the_window(self):
+
+    screen = Gdk.Screen.get_default()
+    win = screen.get_active_window()
+
+    width = win.get_width()
+    height = win.get_height()
+
+    # 获取窗口图像
+    image = Gdk.pixbuf_get_from_window(win, 0, 0, width, height)
+    image = add_shadow(image)
+
+    self.copy(image)
+
+    # print('截图成功...')
+
+
+class ApplicationService(dbus.service.Object):
+  def __init__(self, app):
+    self.app = app
+    bus_name = dbus.service.BusName(app_id, bus = dbus.SessionBus())
+    dbus.service.Object.__init__(self, bus_name, '/')
+
+
+  @dbus.service.method(app_id)
+  def call_app(self):
+    self.app.shot_the_window()
+
+
+if __name__ == "__main__":
+  dbus.mainloop.glib.DBusGMainLoop(set_as_default = True)
+  bus = dbus.SessionBus()
+
+  try:
+    obj = bus.get_object(app_id, '/')
+    obj.call_app()
+    sys.exit(0)
+  except dbus.DBusException:
+    pass
+
+
+  app = Application()
+  app.run(sys.argv)
+
+  ApplicationService(app)
+
+  Gtk.main()
+