// ==UserScript==
// @name         WeiyunHelper - 微云 Aria2 下载辅助脚本
// @namespace    https://github.com/loo2k/WeiyunHelper/
// @version      0.0.9
// @description  微云下载时文件支持导出到 aria2 下载，支持分享页面及个人云盘管理页
// @author       Luke
// @match        *://*.weiyun.com/*
// @grant        none
// @run-at       document-end
// @supportURL   https://github.com/loo2k/WeiyunHelper/issues
// @downloadURL https://update.greasyfork.icu/scripts/402669/WeiyunHelper%20-%20%E5%BE%AE%E4%BA%91%20Aria2%20%E4%B8%8B%E8%BD%BD%E8%BE%85%E5%8A%A9%E8%84%9A%E6%9C%AC.user.js
// @updateURL https://update.greasyfork.icu/scripts/402669/WeiyunHelper%20-%20%E5%BE%AE%E4%BA%91%20Aria2%20%E4%B8%8B%E8%BD%BD%E8%BE%85%E5%8A%A9%E8%84%9A%E6%9C%AC.meta.js
// ==/UserScript==

(function () {
  'use strict';

  var B64 = {
    alphabet: 'ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789+/=',
    lookup: null,
    ie: /MSIE /.test(navigator.userAgent),
    ieo: /MSIE [67]/.test(navigator.userAgent),
    encode: function (s) {
      /* jshint bitwise:false */
      var buffer = B64.toUtf8(s),
        position = -1,
        result,
        len = buffer.length,
        nan0, nan1, nan2, enc = [, , , ];
      
      if (B64.ie) {
        result = [];
        while (++position < len) {
          nan0 = buffer[position];
          nan1 = buffer[++position];
          enc[0] = nan0 >> 2;
          enc[1] = ((nan0 & 3) << 4) | (nan1 >> 4);
          if (isNaN(nan1))
            enc[2] = enc[3] = 64;
          else {
            nan2 = buffer[++position];
            enc[2] = ((nan1 & 15) << 2) | (nan2 >> 6);
            enc[3] = (isNaN(nan2)) ? 64 : nan2 & 63;
          }
          result.push(B64.alphabet.charAt(enc[0]), B64.alphabet.charAt(enc[1]), B64.alphabet.charAt(enc[2]), B64.alphabet.charAt(enc[3]));
        }
        return result.join('');
      } else {
        result = '';
        while (++position < len) {
          nan0 = buffer[position];
          nan1 = buffer[++position];
          enc[0] = nan0 >> 2;
          enc[1] = ((nan0 & 3) << 4) | (nan1 >> 4);
          if (isNaN(nan1))
            enc[2] = enc[3] = 64;
          else {
            nan2 = buffer[++position];
            enc[2] = ((nan1 & 15) << 2) | (nan2 >> 6);
            enc[3] = (isNaN(nan2)) ? 64 : nan2 & 63;
          }
          result += B64.alphabet[enc[0]] + B64.alphabet[enc[1]] + B64.alphabet[enc[2]] + B64.alphabet[enc[3]];
        }
        return result;
      }
    },
    decode: function (s) {
      /* jshint bitwise:false */
      s = s.replace(/\s/g, '');
      if (s.length % 4)
        throw new Error('InvalidLengthError: decode failed: The string to be decoded is not the correct length for a base64 encoded string.');
      if(/[^A-Za-z0-9+\/=\s]/g.test(s))
        throw new Error('InvalidCharacterError: decode failed: The string contains characters invalid in a base64 encoded string.');
  
      var buffer = B64.fromUtf8(s),
        position = 0,
        result,
        len = buffer.length;
  
      if (B64.ieo) {
        result = [];
        while (position < len) {
          if (buffer[position] < 128)
            result.push(String.fromCharCode(buffer[position++]));
          else if (buffer[position] > 191 && buffer[position] < 224)
            result.push(String.fromCharCode(((buffer[position++] & 31) << 6) | (buffer[position++] & 63)));
          else
            result.push(String.fromCharCode(((buffer[position++] & 15) << 12) | ((buffer[position++] & 63) << 6) | (buffer[position++] & 63)));
        }
        return result.join('');
      } else {
        result = '';
        while (position < len) {
          if (buffer[position] < 128)
            result += String.fromCharCode(buffer[position++]);
          else if (buffer[position] > 191 && buffer[position] < 224)
            result += String.fromCharCode(((buffer[position++] & 31) << 6) | (buffer[position++] & 63));
          else
            result += String.fromCharCode(((buffer[position++] & 15) << 12) | ((buffer[position++] & 63) << 6) | (buffer[position++] & 63));
        }
        return result;
      }
    },
    toUtf8: function (s) {
      /* jshint bitwise:false */
      var position = -1,
        len = s.length,
        chr, buffer = [];
      if (/^[\x00-\x7f]*$/.test(s)) while (++position < len)
        buffer.push(s.charCodeAt(position));
      else while (++position < len) {
        chr = s.charCodeAt(position);
        if (chr < 128)
          buffer.push(chr);
        else if (chr < 2048)
          buffer.push((chr >> 6) | 192, (chr & 63) | 128);
        else
          buffer.push((chr >> 12) | 224, ((chr >> 6) & 63) | 128, (chr & 63) | 128);
      }
      return buffer;
    },
    fromUtf8: function (s) {
      /* jshint bitwise:false */
      var position = -1,
        len, buffer = [],
        enc = [, , , ];
      if (!B64.lookup) {
        len = B64.alphabet.length;
        B64.lookup = {};
        while (++position < len)
          B64.lookup[B64.alphabet.charAt(position)] = position;
        position = -1;
      }
      len = s.length;
      while (++position < len) {
        enc[0] = B64.lookup[s.charAt(position)];
        enc[1] = B64.lookup[s.charAt(++position)];
        buffer.push((enc[0] << 2) | (enc[1] >> 4));
        enc[2] = B64.lookup[s.charAt(++position)];
        if (enc[2] === 64)
          break;
        buffer.push(((enc[1] & 15) << 4) | (enc[2] >> 2));
        enc[3] = B64.lookup[s.charAt(++position)];
        if (enc[3] === 64)
          break;
        buffer.push(((enc[2] & 3) << 6) | enc[3]);
      }
      return buffer;
    }
  };
  
  var B64url = {
    decode: function(input) {
      // Replace non-url compatible chars with base64 standard chars
      input = input
        .replace(/-/g, '+')
        .replace(/_/g, '/');
  
      // Pad out with standard base64 required padding characters
      var pad = input.length % 4;
      if(pad) {
        if(pad === 1) {
        throw new Error('InvalidLengthError: Input base64url string is the wrong length to determine padding');
        }
        input += new Array(5-pad).join('=');
      }
  
      return B64.decode(input);
    },
  
    encode: function(input) {
      var output = B64.encode(input);
      return output
        .replace(/\+/g, '-')
        .replace(/\//g, '_')
        .split('=', 1)[0];
    }
  };
  
  const base64 = {
    decode: B64.decode,
    encode: B64.encode,
    urldecode: B64url.decode,
    urlencode: B64url.encode,
  };

  Date.prototype.Format = function (fmt) {
    var o = {
      'M+': this.getMonth() + 1,
      'd+': this.getDate(),
      'h+': this.getHours(),
      'm+': this.getMinutes(),
      's+': this.getSeconds(),
      'q+': Math.floor((this.getMonth() + 3) / 3),
      S: this.getMilliseconds(),
    };
    if (/(y+)/.test(fmt)) fmt = fmt.replace(RegExp.$1, (this.getFullYear() + '').substr(4 - RegExp.$1.length));
    for (var k in o)
      if (new RegExp('(' + k + ')').test(fmt))
        fmt = fmt.replace(RegExp.$1, RegExp.$1.length == 1 ? o[k] : ('00' + o[k]).substr(('' + o[k]).length));
    return fmt;
  };

  /**
   * 将微云返回的下载地址解析到 Aria2 进行下载
   *
   * @param {void}
   */
  const handleResp2Aria2 = (ret) => {
    let downloadUrl = '';
    let cookieName = '';
    let cookieValue = '';
    let URI = {};
    let fileName = '';
    if (ret.file_list) {
      downloadUrl = ret.file_list[0].https_download_url;
      cookieName = ret.file_list[0].cookie_name;
      cookieValue = ret.file_list[0].cookie_value;
      URI = new URL(downloadUrl);
      fileName = decodeURI(URI.pathname.substr(URI.pathname.lastIndexOf('/') + 1));
    } else {
      downloadUrl = ret.https_download_url;
      cookieName = ret.cookie_name;
      cookieValue = ret.cookie_value;
      fileName = `微云合并下载文件_${new Date().Format('yyyy-MM-dd hh:mm:ss')}.zip`;
    }

    const ariaNgUrl = `http://ariang.mayswind.net/latest/#!/new/task?url=${base64.urlencode(downloadUrl)}&header=Cookie:${cookieName}=${cookieValue}&out=${encodeURI(fileName)}`;

    console.log('文件名称:', fileName);
    console.log('下载地址:', downloadUrl);
    console.log('请求参数:', `Cookie:${cookieName}=${cookieValue}`);
    console.log('AriaNg URL:', ariaNgUrl);

    // 使用 ariaNg 进行下载
    window.open(ariaNgUrl);
  }

  const injectChunkId = Math.random().toString(36).substring(7);

  // 微云文件分享页面注入脚本模块
  location.host === 'share.weiyun.com' && webpackJsonp([7892], {[injectChunkId]: function(modules, exports, require) {
    // 寻找 DownloadRequest 模块
    const [ DownloadRequest ] = Object.values(require.c)
      .filter((x) => x.exports && typeof x.exports.DownloadRequest === 'function' && typeof x.exports.DownloadType === 'object')
      .map((x) => x.exports.DownloadRequest);

    // 寻找 DownloadOperate 模块
    const [ DownloadOperate ] = Object.values(require.c)
      .filter((x) => x.exports && typeof x.exports.DownloadOperate === 'function')
      .map((x) => x.exports.DownloadOperate);

    // 获取 Vue 应用实例
    const $Vue = document.getElementById('app').__vue__;

    // 判断依赖模块是否存在
    if (!DownloadRequest || !DownloadOperate) {
      console.error('没有检测到适配模块，已退出 WeiyunHelper');
      console.error('你可以到 https://github.com/loo2k/WeiyunHelper/issues 向作者反馈问题')
      return false;
    }

    // 下载选中文件
    function downloadSelectedFiles() {
      const { shareFile } = $Vue.$store.state.sharefile;
      if (!$Vue.$store.getters["sharefile/isSelected"]) {
        if (shareFile.shareFile.childNodes.length === 1) {
          shareFile.shareFile.selectAllFiles();
        } else {
          return alert('你都还没有选择文件 :(');
        }
      }

      const downloadOptions = {
        fileOwner: shareFile.shareOwner,
        shareKey: shareFile.shareKey,
        sharePwd: shareFile.sharePwd,
        downloadType: 0
      }

      return new DownloadOperate(shareFile.shareFile, downloadOptions)
        .downloadWithType_(new DownloadRequest(), downloadOptions)
        .then(handleResp2Aria2).catch(e => { alert(e.msg) });
    }

    // 监听 body 的 DOM 变化并将下载入口植入
    const observeTarget = document.body;
    const observeConfig = { attributes: true, childList: true, subtree: true };
    const observeCallback = function (mutations, observer) {
      for (let mutation of mutations) {
        if (mutation.type === 'childList' && mutation.addedNodes.length > 0) {
          mutation.addedNodes.forEach((node) => {
            // 判断页面中增加的元素是否是针对文件的下拉菜单
            if (
              node.className &&
              node.className.indexOf('mod-bubble-context-menu') > -1 &&
              node.__vue__ &&
              node.__vue__.items.some(e => e.method === 'download')
            ) {
              const contextItems = node.querySelectorAll('.menu-item');
              const newContextItem = document.createElement('li')
              newContextItem.className = 'menu-item';
              newContextItem.innerHTML = '<span class="txt">使用 Aria 下载</span>';
              newContextItem.addEventListener('click', function() {
                downloadSelectedFiles();
                // 关闭右键菜单
                document.dispatchEvent(new Event('mousedown'));
              });
              contextItems[0].parentNode.insertBefore(newContextItem, contextItems[0].nextSibling);
            }
          })
        }
      }
    }
    const observeInstance = new MutationObserver(observeCallback);
    observeInstance.observe(observeTarget, observeConfig);

    // 直接注入工具条的下载入口
    const actionWrapCode = document.querySelector('.mod-action-wrap-code');
    const actionWrapAria = document.createElement('div');
    actionWrapAria.className = 'mod-action-wrap mod-action-wrap-menu mod-action-wrap-aria clearfix';

    const actionItem = document.createElement('div');
    actionItem.className = 'action-item';
    actionItem.innerHTML = '<div class="action-item-con"><i class="icon icon-download"></i><span class="act-txt">使用 Aria 下载</span></div>';
    actionItem.addEventListener('click', function () {
      downloadSelectedFiles();
    });
    actionWrapAria.appendChild(actionItem);
    actionWrapCode.parentNode.insertBefore(actionWrapAria, actionWrapCode);
  }}, [injectChunkId]);

  // 微云云盘管理页面注入脚本模块
  location.host === 'www.weiyun.com' && webpackJsonp([7891], {[injectChunkId]: function(modules, exports, require) {
    // 寻找云盘操作 API 模块
    const diskServices = Object.values(require.c)
      .filter((x) => x.exports && typeof x.exports.namespace === 'function' && typeof x.exports.namespace('PERSON').fetchUserInfo === 'function')
      .map((x) => x.exports.namespace);
    const diskService = diskServices && diskServices[0]('PERSON');

    if (diskServices.length === 0) {
      console.error('没有检测到适配模块，已退出 WeiyunHelper');
      console.error('你可以到 https://github.com/loo2k/WeiyunHelper/issues 向作者反馈问题')
      return false;
    }

    // 下载选中的文件
    function downloadSelectedFiles() {
      let request = null;
      const selected = document.querySelectorAll('.list-group-item.checked.act');
      const fileNodes = Array.from(selected).map(item => item.__vue__.fileNode);
      if (fileNodes.length === 1 && !fileNodes[0].isDir()) {
        request = diskService.fetchDownloadFileInfo({ fileNodes });
      } else {
        request = diskService.fetchPackDownloadDirFileInfo({ fileNodes });
      }

      request.then(handleResp2Aria2).catch(e => { alert(e.msg) });
    }

    // 监听 body 的 DOM 变化并将下载入口植入
    const observeTarget = document.body;
    const observeConfig = { attributes: true, childList: true, subtree: true };
    const observeCallback = function (mutations, observer) {
      for (let mutation of mutations) {
        if (mutation.type === 'childList' && mutation.addedNodes.length > 0) {
          mutation.addedNodes.forEach((node) => {
            // 判断页面中增加的元素是否是针对文件的下拉菜单
            if (
              node.className &&
              node.className.indexOf('mod-bubble-context-menu') > -1 &&
              node.__vue__ &&
              node.__vue__.items.some(e => e.method === 'download')
            ) {
              const contextItems = node.querySelectorAll('.menu-item');
              const newContextItem = document.createElement('li')
              newContextItem.className = 'menu-item';
              newContextItem.innerHTML = '<span class="txt">使用 Aria 下载</span>';
              newContextItem.addEventListener('click', function() {
                downloadSelectedFiles();
                // 关闭右键菜单
                document.dispatchEvent(new Event('mousedown'));
              });
              contextItems[0].parentNode.insertBefore(newContextItem, contextItems[0].nextSibling);
            }
          })
        }

        // 针对顶部下载菜单
        if (
          mutation.type === 'attributes' &&
          mutation.attributeName === 'style' &&
          mutation.target.className.indexOf('mod-action-wrap-menu') > -1 &&
          mutation.target.style.display !== 'none' &&
          mutation.target.querySelectorAll('#action-item-aria').length === 0
        ) {
          const actionItems = mutation.target.querySelectorAll('.action-item');
          const newActionItem = document.createElement('div');
          newActionItem.id = 'action-item-aria'
          newActionItem.className = 'action-item';
          newActionItem.innerHTML = '<div class="action-item-con"><i class="icon icon-download"></i><span class="act-txt">使用 Aria 下载</span></div>';
          newActionItem.addEventListener('click', function () {
            downloadSelectedFiles();
          });
          mutation.target.insertBefore(newActionItem, actionItems[0].nextSibling);
        }
      }
    }
    const observeInstance = new MutationObserver(observeCallback);
    observeInstance.observe(observeTarget, observeConfig);

    // 打开离线下载窗口并填写链接
    const openDownloadModal = (text = '') => {
      let wyCreateBtn = document.querySelectorAll('.mod-action-wrap-create');
      let $wyCreateBtn = wyCreateBtn[0] && wyCreateBtn[0].__vue__;
      $wyCreateBtn.offlineDownload();

      // 点击使用磁力链下载
      let modalBtNav = document.querySelectorAll('.modal-dialog-bt .modal-tab-nav .tab-nav-item');
      modalBtNav.forEach(nav => {
        if (nav.innerText.trim() === '链接下载') {
          nav.click();
        }
      });

      setTimeout(() => {
        // 填写 magent 或者 ed2k 链接
        let urlTextarea = document.querySelector('.modal-dialog-bt .tab-cont-item.online .input-block');
        if (text) {
          urlTextarea.value = text;
          urlTextarea.dispatchEvent(new Event('input'));
        }
      }, 0);
    }

    // 粘贴磁力链或者 ed2k 时自动启动下载
    document.addEventListener('paste', (event) => {
      // 针对非输入框的粘贴时间
      if (['TEXTAREA', 'INPUT'].includes(event.target.tagName)) {
        return;
      }

      // 剪切板数据对象
      let clipboardData = event.clipboardData || window.clipboardData;

      // 剪切板对象可以获取
      if (!clipboardData) { return; }

      let paste = clipboardData.getData('text');
      let isEd2k = /^ed2k:\/\//ig.test(paste);
      let isMagent = /^magnet:/ig.test(paste);
      if (isEd2k || isMagent) {
        openDownloadModal(paste);
      }
    });
  }}, [injectChunkId]);
})();