add pixiv.js

1 files changed, 164 insertions, 0 deletions

diff --git a/pixiv.js b/pixiv.js
new file mode 100644
index 0000000..12ec7d5
--- /dev/null
+++ b/pixiv.js
@@ -0,0 +1,164 @@
+// INFO //
+var INFO = 
+<plugin name="pixiv.js" version="0.1"
+        summary="pixiv.js"
+        xmlns="http://vimperator.org/namespaces/liberator">
+  <author email="mitsugu.oyama@gmail.com">Mitsugu Oyama</author>
+  <license href="http://opensource.org/licenses/mit-license.php">MIT</license>
+  <project name="Vimperator" minVersion="2.3"/>
+  <p>
+    You can save image from pixiv by this plugin.
+  </p>
+  <item>
+    <tags>'pixiv'</tags>
+    <spec>:pixiv</spec>
+    <description>
+      <p>You can save image from <link topic="http://www.pixiv.net/">pixiv</link> by this plugin.</p>
+      <p>You must login pixiv.</p>
+    </description>
+  </item>
+</plugin>;
+
+commands.addUserCommand(
+  ['pixiv'],
+  'Save Image File from pixiv',
+  function(){
+    let contents=gBrowser.selectedBrowser.contentDocument;
+    if(contents.domain!="www.pixiv.net"){
+      liberator.echoerr('This page is not pixiv.');
+      return false;
+    }
+    if(contents.URL.search(/medium&illust_id=/i)==-1){
+      liberator.echoerr("This page is not pixiv's image page.");
+      return false;
+    }
+
+    let Cc=Components.classes;
+    let Ci=Components.interfaces;
+    const baseInfo="http://www.pixiv.net/member_illust.php?mode=big&illust_id=";
+    let id=contents.URL.substr(contents.URL.lastIndexOf('=')+1);
+    let cookie=contents.cookie;
+
+    let directoryPicker=function() {
+      let path;
+      let fp=Cc["@mozilla.org/filepicker;1"].createInstance(Ci.nsIFilePicker);
+      fp.init(window,'Select Directory',Ci.nsIFilePicker.modeGetFolder);
+      let result=fp.show();
+      switch(result){
+        case Ci.nsIFilePicker.returnOK:
+          path=fp.file.path;
+          break;
+        default:
+        case Ci.nsIFilePicker.returnCancel:
+          return '';
+      }
+      return path;
+    };
+    let savePath=directoryPicker();
+    if(savePath.length<1) return;
+
+    let getDOMHtmlDocument=function(str){
+      let doc;
+      let range;
+      try{
+        if(document.implementation.createHTMLDocument){
+          doc=document.implementation.createHTMLDocument('');
+          range=doc.createRange();
+          range.selectNodeContents(doc.documentElement);
+          range.deleteContents();
+          doc.documentElement.appendChild(range.createContextualFragment(str));
+        }else{
+          doc=document.implementation.createDocument(null,'html',null);
+          range=document.createRange();
+          range.selectNodeContents(document.documentElement);
+          let content=doc.adoptNode(range.createContextualFragment(str));
+          doc.documentElement.appendChild(content);
+        }
+      }catch(e){
+        doc=null;
+      }
+      return doc;
+    };
+
+    let getImageUrl=function(pageContents){
+      let url;
+      let htmldoc=getDOMHtmlDocument(pageContents);
+      if(htmldoc){
+        if(0<htmldoc.getElementsByTagName('img').length)
+          url=htmldoc.getElementsByTagName('img').item(0).getAttribute('src');
+        else
+          url='';
+      }else{
+        let s=pageContents.indexOf('src="')+5;
+        let e=pageContents.indexOf('"',s);
+        url=pageContents.substr(s,e-s);
+      }
+      return url;
+    };
+
+    let imgUrl;
+
+    let truePixivImg=function(){
+      let fileName=imgUrl.substr(imgUrl.lastIndexOf('/'));
+      savePath=savePath+fileName;
+      let instream=xhrImg.responseText;
+      let aFile=Cc["@mozilla.org/file/local;1"]
+        .createInstance(Ci.nsILocalFile);
+      aFile.initWithPath(savePath);
+      let outstream=Cc["@mozilla.org/network/safe-file-output-stream;1"]
+        .createInstance(Ci.nsIFileOutputStream);
+      outstream.init(aFile,0x02|0x08|0x20,0664,0);
+      outstream.write(instream,instream.length);
+      if (outstream instanceof Ci.nsISafeOutputStream) {
+        outstream.finish();
+      }else{
+        outstream.close();
+      }
+    };
+
+    let falsePixivImg=function(){
+      liberator.echoerr("Image file accept error.");
+      return false;
+    };
+
+    let saveImag=function(){
+      xhrImg=Cc["@mozilla.org/xmlextras/xmlhttprequest;1"]
+        .createInstance();
+      xhrImg.QueryInterface(Ci.nsIDOMEventTarget);
+      xhrImg.addEventListener("load",truePixivImg,false);
+      xhrImg.addEventListener("error",falsePixivImg,false);
+      xhrImg.QueryInterface(Ci.nsIXMLHttpRequest);
+      xhrImg.open("GET",imgUrl,true);
+      xhrImg.overrideMimeType('text/plain;charset=x-user-defined');
+      xhrImg.setRequestHeader('Referer',contents.URL);
+      xhrImg.setRequestHeader('Cookie',cookie);
+      xhrImg.send(null);
+    };
+
+    let trueImgInfo=function(){
+      imgUrl=getImageUrl(xhrImgInfo.responseText);
+      if(0<imgUrl.length){
+        saveImag();
+      }else{
+        liberator.echoerr("You should login TINAMI :<");
+      }
+    };
+
+    let falseImgInfo=function(){
+      liberator.echo("Image Infomation page accept error.");
+      return false;
+    };
+
+    let xhrImgInfo;
+    xhrImgInfo=Cc["@mozilla.org/xmlextras/xmlhttprequest;1"].createInstance();
+    xhrImgInfo.QueryInterface(Ci.nsIDOMEventTarget);
+    xhrImgInfo.addEventListener("load",trueImgInfo,false);
+    xhrImgInfo.addEventListener("error",falseImgInfo,false);
+    xhrImgInfo.QueryInterface(Ci.nsIXMLHttpRequest);
+    xhrImgInfo.open("GET",baseInfo+id,true);
+    xhrImgInfo.setRequestHeader('Referer',contents.URL);
+    xhrImgInfo.setRequestHeader('Cookie',cookie);
+    xhrImgInfo.send(null);
+
+  }
+);