picosh 0.1.1 → 0.1.2

package/index.js

@@ -3,17 +3,46 @@
 const path = require('path');
 const fs = require('fs');
 const os = require('os');
+const https = require('https');
+const http = require('http');
 
 const SAVE_DIR = path.join(os.tmpdir(), 'picosh');
 
 function saveClipboardImage(clipboard) {
   const img = clipboard.readImage();
-  if (img.isEmpty()) return null;
+  if (!img.isEmpty()) {
+    fs.mkdirSync(SAVE_DIR, {recursive: true});
+    const filepath = path.join(SAVE_DIR, `clip_${Date.now()}.png`);
+    fs.writeFileSync(filepath, img.toPNG());
+    return Promise.resolve(filepath);
+  }
 
-  fs.mkdirSync(SAVE_DIR, {recursive: true});
-  const filepath = path.join(SAVE_DIR, `clip_${Date.now()}.png`);
-  fs.writeFileSync(filepath, img.toPNG());
-  return filepath;
+  const html = clipboard.readHTML();
+  if (html) {
+    const match = html.match(/<img[^>]+src=["']([^"']+)["']/i);
+    if (match) {
+      return downloadImage(match[1]);
+    }
+  }
+
+  return Promise.resolve(null);
+}
+
+function downloadImage(url) {
+  return new Promise((resolve) => {
+    fs.mkdirSync(SAVE_DIR, {recursive: true});
+    const filepath = path.join(SAVE_DIR, `clip_${Date.now()}.png`);
+    const file = fs.createWriteStream(filepath);
+    const client = url.startsWith('https') ? https : http;
+
+    client.get(url, (res) => {
+      res.pipe(file);
+      file.on('finish', () => file.close(() => resolve(filepath)));
+    }).on('error', () => {
+      fs.unlink(filepath, () => {});
+      resolve(null);
+    });
+  });
 }
 
 exports.decorateTerm = (Term, {React}) => {
@@ -24,15 +53,14 @@ exports.decorateTerm = (Term, {React}) => {
 
         try {
           const {clipboard} = require('electron');
-          const filepath = saveClipboardImage(clipboard);
-          if (!filepath) return;
-
-          e.preventDefault();
-          e.stopPropagation();
-
-          if (this.props.onData) {
-            this.props.onData(filepath);
-          }
+          saveClipboardImage(clipboard).then((filepath) => {
+            if (!filepath) return;
+            e.preventDefault();
+            e.stopPropagation();
+            if (this.props.onData) {
+              this.props.onData(filepath);
+            }
+          });
         } catch (_) {}
       };
 

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "picosh",
-  "version": "0.1.1",
+  "version": "0.1.2",
   "description": "Hyper plugin: paste clipboard images as file paths",
   "main": "index.js",
   "license": "MIT",