// INFO //
var INFO =
<plugin name="pixiv.js" version="0.6"
        summary="Download image from pixiv"
        href="http://github.com/vimpr/vimperator-plugins/blob/master/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;

    let id;
    if(-1==contents.URL.search(/\&from_sid=/i)){
      id=contents.URL.substr(contents.URL.lastIndexOf('=')+1);
    }else{
      let st=contents.URL.search(/illust_id=/i)+'illust_id='.length;
      let end=contents.URL.lastIndexOf('&');
      id=contents.URL.substr(st,end-st);
    }

    let baseInfo;
    let scroll;
    let type=contents.getElementsByClassName('works_display').item(0)
              .firstChild.getAttribute('href');
    if(-1!=type.search(/big&illust_id=/i)){
      baseInfo="http://www.pixiv.net/member_illust.php?mode=big&illust_id=";
      scroll='';
    }else if(-1!=type.search(/manga&illust_id=/i)){
      baseInfo="http://www.pixiv.net/member_illust.php?mode=manga&illust_id=";
      scroll='&type=scroll';
    }else{
      liberator.echoerr("This page is not image page and not manga page.");
      return false;
    }
    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 saveDirectory=directoryPicker();
    if(saveDirectory.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{
          let doctype=document.implementation.createDocumentType(
                      'html',
                      '-//W3C//DTD HTML 4.01 Transitional//EN',
                      'http://www.w3.org/TR/html4/loose.dtd'
          );
          doc=document.implementation.createDocument(null,'html',doctype);
          range=doc.createRange();
          range.selectNodeContents(doc.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('/'));
      if (-1!=fileName.indexOf('?')){
        fileName=fileName.substr(0,fileName.indexOf('?'));
      }
      let tmpPath=saveDirectory+fileName;
      let instream=xhrImg.responseText;
      let aFile=Cc["@mozilla.org/file/local;1"]
        .createInstance(Ci.nsILocalFile);
      aFile.initWithPath(tmpPath);
      if(true===aFile.exists()){
        let value=window.prompt('すでに同じ名前のファイルがあります。デフォルトファイル名を変更してください。',fileName.substr(1));
        if(null===value){
          return false;
        }
        fileName='/'+value;
        tmpPath=saveDirectory+fileName;
        aFile.initWithPath(tmpPath);
      }
      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 saveImage=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,false);
      xhrImg.overrideMimeType('text/plain;charset=x-user-defined');
      xhrImg.setRequestHeader('Referer',contents.URL);
      xhrImg.setRequestHeader('Cookie',cookie);
      xhrImg.send(null);
    };

    let saveImageFile=function(){
      imgUrl=getImageUrl(xhrImgInfo.responseText);
      if(0<imgUrl.length){
        saveImage();
      }else{
        liberator.echoerr("You should login pixiv :<");
      }
    };

    let getImageUrls=function(pageContents){
      const BIG='_big';
      let url=[];
      let strScript;
      let fst,snd;
      let strFst='';
      let strSnd='';
      let tblElm;
      let i;
      let htmldoc=getDOMHtmlDocument(pageContents);
      if(htmldoc){
        let max=htmldoc.getElementsByClassName('image-container').length;
        for(i=0;i<max;i++){
          strScript=htmldoc.getElementsByClassName('image-container').item(i)
            .getElementsByTagName('script').item(0)
            .childNodes.item(0).nodeValue;
          fst=strScript.search(/unshift/i)+'unshift'.length+2;
          snd=strScript.lastIndexOf('_');
          strFst=strScript.substr(fst,snd-fst);

          fst=snd;
          snd=strScript.indexOf("'",fst);
          strSnd=strScript.substr(fst,snd-fst);
          
          url.push(strFst+BIG+strSnd);
        }
      }else{
        url.length=0;
      }
      return url;
    };

    let imgUrls;
    let saveMangaFiles=function(){
      imgUrls=getImageUrls(xhrImgInfo.responseText);
      if(0<imgUrls.length){
        let i;
        let max=imgUrls.length;
        for(i=0;i<max;i++){
          imgUrl=imgUrls[i];
          pnt=imgUrl.lastIndexOf('?');
          if(-1!=pnt){
            imgUrl=imgUrl.substr(0,pnt);
          }
          saveImage();
        }
      }else{
        liberator.echoerr("Not found image data on the manga page.");
      }
    };

    let trueImgInfo=function(){
      if(-1!=type.search(/big&illust_id=/i)){
        saveImageFile();
      }else if(-1!=type.search(/manga&illust_id=/i)){
        saveMangaFiles();
      }else{
        liberator.echoerr("This page is not image page and not manga page.");
        return false;
      }
    };

    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+scroll,true);
    xhrImgInfo.setRequestHeader('Referer',contents.URL);
    xhrImgInfo.setRequestHeader('Cookie',cookie);
    xhrImgInfo.send(null);

  }
);
 class="nx">license</span><span class="o">&gt;</span><span class="nx">GPL</span><span class="o">&lt;</span><span class="err">/license&gt;</span>
  <span class="o">&lt;</span><span class="nx">minVersion</span><span class="o">&gt;</span><span class="mf">1.2</span><span class="o">&lt;</span><span class="err">/minVersion&gt;</span>
  <span class="o">&lt;</span><span class="nx">maxVersion</span><span class="o">&gt;</span><span class="mf">2.1</span><span class="o">&lt;</span><span class="err">/maxVersion&gt;</span>
  <span class="o">&lt;</span><span class="nx">updateURL</span><span class="o">&gt;</span><span class="nx">https</span><span class="o">:</span><span class="c1">//github.com/vimpr/vimperator-plugins/raw/master/amazon_simple_uri.js&lt;/updateURL&gt;
</span><span class="c1"></span>  <span class="o">&lt;</span><span class="nx">detail</span><span class="o">&gt;&lt;!</span><span class="p">[</span><span class="nx">CDATA</span><span class="p">[</span>

<span class="o">==</span> <span class="nx">Option</span> <span class="o">==</span>
<span class="o">&gt;||</span>
  <span class="kd">let</span> <span class="nx">g</span><span class="o">:</span><span class="nx">amazon_asamashi</span> <span class="o">=</span> <span class="s2">&#34;hogehoge-22&#34;</span>
<span class="o">||&lt;</span>
<span class="nx">と設定することにより</span><span class="err">、</span><span class="nx">Amazon</span> <span class="nx">アソシエイトID</span><span class="p">(</span><span class="nx">上の例ではhogehoge</span><span class="o">-</span><span class="mi">22</span><span class="p">)</span><span class="nx">をURLに追加します</span><span class="err">。</span>

  <span class="p">]]</span><span class="o">&gt;&lt;</span><span class="err">/detail&gt;</span>
<span class="o">&lt;</span><span class="err">/VimperatorPlugin&gt;;</span>
<span class="c1">//}}}
</span><span class="c1">//
</span><span class="c1"></span><span class="p">(</span><span class="kd">function</span><span class="p">()</span>
<span class="p">{</span>
  <span class="c1">// URLを正規表現でチェックしてISBNっぽい10桁,13桁の数字と取り出す
</span><span class="c1"></span>  <span class="kd">function</span> <span class="nx">getIsbn</span><span class="p">(</span><span class="nx">uri</span><span class="p">)</span>
  <span class="p">{</span>
    <span class="kd">var</span> <span class="nx">regex</span> <span class="o">=</span> <span class="k">new</span> <span class="nb">RegExp</span><span class="p">(</span><span class="s2">&#34;([0-9]{9,13}[0-9Xx])&#34;</span><span class="p">);</span>
    <span class="kd">var</span> <span class="nx">match</span> <span class="o">=</span> <span class="nx">uri</span><span class="p">.</span><span class="nx">match</span><span class="p">(</span><span class="nx">regex</span><span class="p">);</span>
    <span class="kd">var</span> <span class="nx">isbn</span> <span class="o">=</span> <span class="nx">match</span><span class="p">[</span><span class="mi">0</span><span class="p">];</span>
    <span class="k">if</span> <span class="p">(</span><span class="nx">isbn</span><span class="p">.</span><span class="nx">length</span> <span class="o">==</span> <span class="mi">13</span><span class="p">)</span>
    <span class="p">{</span>
      <span class="nx">isbn</span> <span class="o">=</span> <span class="nx">getIsbn10</span><span class="p">(</span><span class="nx">isbn</span><span class="p">);</span>
    <span class="p">}</span>
    <span class="k">return</span> <span class="nx">isbn</span><span class="p">;</span>
  <span class="p">}</span>

  <span class="c1">// SSBのAPIを使ってISBNから書籍情報を取得
</span><span class="c1"></span>  <span class="kd">function</span> <span class="nx">getBookInfo</span><span class="p">(</span><span class="nx">isbn</span><span class="p">)</span>
  <span class="p">{</span>
    <span class="kd">var</span> <span class="nx">uri</span> <span class="o">=</span> <span class="s2">&#34;http://stack.nayutaya.jp/api/book/&#34;</span><span class="p">;</span>
    <span class="k">if</span> <span class="p">(</span><span class="nx">isbn</span><span class="p">.</span><span class="nx">length</span> <span class="o">==</span> <span class="mi">10</span><span class="p">)</span>
    <span class="p">{</span>
      <span class="nx">uri</span> <span class="o">+=</span> <span class="s2">&#34;isbn10/&#34;</span> <span class="o">+</span> <span class="nx">isbn</span> <span class="o">+</span> <span class="s2">&#34;.json&#34;</span><span class="p">;</span>
    <span class="p">}</span>
    <span class="k">else</span> <span class="k">if</span> <span class="p">(</span><span class="nx">isbn</span><span class="p">.</span><span class="nx">length</span> <span class="o">==</span> <span class="mi">13</span><span class="p">)</span>
    <span class="p">{</span>
      <span class="nx">uri</span> <span class="o">+=</span> <span class="s2">&#34;isbn13/&#34;</span> <span class="o">+</span> <span class="nx">isbn</span> <span class="o">+</span> <span class="s2">&#34;.json&#34;</span><span class="p">;</span>
    <span class="p">}</span>
    <span class="kd">var</span> <span class="nx">xhr</span> <span class="o">=</span> <span class="k">new</span> <span class="nx">XMLHttpRequest</span><span class="p">();</span>
    <span class="nx">xhr</span><span class="p">.</span><span class="nx">open</span><span class="p">(</span><span class="s1">&#39;GET&#39;</span><span class="p">,</span> <span class="nx">uri</span><span class="p">,</span> <span class="kc">false</span><span class="p">);</span>
    <span class="nx">xhr</span><span class="p">.</span><span class="nx">send</span><span class="p">(</span><span class="kc">null</span><span class="p">);</span>
    <span class="k">if</span> <span class="p">(</span><span class="nx">xhr</span><span class="p">.</span><span class="nx">status</span> <span class="o">!=</span> <span class="mi">200</span><span class="p">)</span> <span class="p">{</span>
        <span class="nx">liberator</span><span class="p">.</span><span class="nx">echoerr</span><span class="p">(</span><span class="s1">&#39;false&#39;</span><span class="p">);</span>
        <span class="k">return</span><span class="p">;</span>
    <span class="p">}</span>
    <span class="k">return</span> <span class="nb">window</span><span class="p">.</span><span class="nb">eval</span><span class="p">(</span><span class="s1">&#39;(&#39;</span> <span class="o">+</span> <span class="nx">xhr</span><span class="p">.</span><span class="nx">responseText</span> <span class="o">+</span> <span class="s1">&#39;)&#39;</span><span class="p">);</span>
  <span class="p">}</span>

  <span class="c1">//SSBのAPIから書籍情報を取り、そこからISBN10を取得
</span><span class="c1"></span>  <span class="kd">function</span> <span class="nx">getIsbn10</span><span class="p">(</span><span class="nx">isbn13</span><span class="p">)</span>
  <span class="p">{</span>
    <span class="kd">var</span> <span class="nx">info</span> <span class="o">=</span> <span class="nx">getBookInfo</span><span class="p">(</span><span class="nx">isbn13</span><span class="p">);</span>
    <span class="k">return</span> <span class="nx">info</span><span class="p">.</span><span class="nx">response</span><span class="p">.</span><span class="nx">book</span><span class="p">.</span><span class="nx">isbn10</span><span class="p">;</span>
  <span class="p">}</span>

  <span class="nx">commands</span><span class="p">.</span><span class="nx">addUserCommand</span><span class="p">(</span>
    <span class="p">[</span><span class="s1">&#39;amazoncopy&#39;</span><span class="p">,</span><span class="s1">&#39;asc&#39;</span><span class="p">],</span>
    <span class="s1">&#39;Copy Amazon Short URI&#39;</span><span class="p">,</span>
    <span class="kd">function</span><span class="p">(</span><span class="nx">args</span><span class="p">)</span>
    <span class="p">{</span>
      <span class="kd">var</span> <span class="nx">asin</span> <span class="o">=</span> <span class="nb">window</span><span class="p">.</span><span class="nx">content</span><span class="p">.</span><span class="nb">document</span><span class="p">.</span><span class="nx">getElementById</span><span class="p">(</span><span class="s1">&#39;ASIN&#39;</span><span class="p">);</span>
      <span class="c1">// ASINが取得できなかった場合 / Amazon以外の書籍情報ページから取得する場合
</span><span class="c1"></span>      <span class="k">if</span> <span class="p">(</span><span class="nx">asin</span> <span class="o">==</span> <span class="kc">null</span><span class="p">)</span>
      <span class="p">{</span>
        <span class="nx">asin</span> <span class="o">=</span> <span class="nx">getIsbn</span><span class="p">(</span><span class="nx">buffer</span><span class="p">.</span><span class="nx">URL</span><span class="p">);</span>
      <span class="p">}</span>
      <span class="k">else</span>
      <span class="p">{</span>
        <span class="nx">asin</span> <span class="o">=</span> <span class="nx">asin</span><span class="p">.</span><span class="nx">value</span>
      <span class="p">}</span>

      <span class="kd">var</span> <span class="nx">uri</span> <span class="o">=</span> <span class="s2">&#34;http://www.amazon.co.jp/dp/&#34;</span> <span class="o">+</span> <span class="nx">asin</span> <span class="o">+</span> <span class="s2">&#34;/&#34;</span><span class="p">;</span>
      <span class="kd">var</span> <span class="nx">asamashi</span> <span class="o">=</span> <span class="k">typeof</span> <span class="nx">liberator</span><span class="p">.</span><span class="nx">globalVariables</span><span class="p">.</span><span class="nx">amazon_asamashi</span> <span class="o">==</span> <span class="s2">&#34;undefined&#34;</span> <span class="o">?</span> <span class="s1">&#39;&#39;</span> <span class="o">:</span> <span class="nx">liberator</span><span class="p">.</span><span class="nx">globalVariables</span><span class="p">.</span><span class="nx">amazon_asamashi</span><span class="p">;</span>
      <span class="k">if</span> <span class="p">(</span><span class="nx">args</span> <span class="o">==</span> <span class="s2">&#34;+a&#34;</span><span class="p">)</span>
      <span class="p">{</span>
        <span class="nx">uri</span> <span class="o">+=</span> <span class="nx">asamashi</span><span class="p">;</span>
      <span class="p">}</span>
      <span class="nx">util</span><span class="p">.</span><span class="nx">copyToClipboard</span><span class="p">(</span><span class="nx">uri</span><span class="p">);</span>
      <span class="nx">liberator</span><span class="p">.</span><span class="nx">echo</span><span class="p">(</span><span class="s2">&#34;[ASC] Copy to clipboard.&#34;</span><span class="p">);</span>
    <span class="p">}</span>
  <span class="p">);</span>
<span class="p">})();</span>
</pre>