picosh 0.1.1 → 0.1.3

package/index.js

@@ -3,17 +3,43 @@
 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;
+function hasImage(clipboard) {
+  if (!clipboard.readImage().isEmpty()) return {type: 'bitmap'};
+  const html = clipboard.readHTML();
+  if (html) {
+    const match = html.match(/<img[^>]+src=["']([^"']+)["']/i);
+    if (match) return {type: 'url', src: match[1]};
+  }
+  return null;
+}
 
+function saveBitmap(clipboard) {
   fs.mkdirSync(SAVE_DIR, {recursive: true});
   const filepath = path.join(SAVE_DIR, `clip_${Date.now()}.png`);
-  fs.writeFileSync(filepath, img.toPNG());
-  return filepath;
+  fs.writeFileSync(filepath, clipboard.readImage().toPNG());
+  return Promise.resolve(filepath);
+}
+
+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 +50,21 @@ exports.decorateTerm = (Term, {React}) => {
 
         try {
           const {clipboard} = require('electron');
-          const filepath = saveClipboardImage(clipboard);
-          if (!filepath) return;
+          const found = hasImage(clipboard);
+          if (!found) return;
 
           e.preventDefault();
           e.stopPropagation();
 
-          if (this.props.onData) {
-            this.props.onData(filepath);
-          }
+          const save = found.type === 'bitmap'
+            ? saveBitmap(clipboard)
+            : downloadImage(found.src);
+
+          save.then((filepath) => {
+            if (filepath && this.props.onData) {
+              this.props.onData(filepath);
+            }
+          });
         } catch (_) {}
       };
 

package/package.json

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