ios_rule_script/script/synology/downloadstation.js

const scriptName = "Download Station";
const synoUrlKey = "syno_https_url";
const synoAccountKey = "syno_account";
const synoPasswdKey = "syno_passwd";
const synoSidKey = "syno_sid";
let magicJS = MagicJS(scriptName, "INFO");

function SynoAuth(synoUrl, account, passwd) {
  return new Promise((resolve, reject) => {
    let options = {
      url: `${synoUrl}/webapi/auth.cgi?api=SYNO.API.Auth&version=2&method=login&account=${account}&passwd=${passwd}&session=DownloadStation&format=cookie`,
      headers: {
        "User-Agent": "Mozilla/5.0",
      },
    };
    magicJS.get(options, (err, resp, data) => {
      if (err) {
        magicJS.logError(`登录DownloadStation失败，请求异常：${err}`);
        reject("❌登录DownloadStation失败，请求异常，请查阅日志！");
      } else {
        try {
          magicJS.logDebug(`接口响应数据：${data}`);
          let obj = typeof data === "string" ? JSON.parse(data) : data;
          if (obj.success === true) {
            magicJS.logDebug(`登录获取的sid：${obj.data.sid}`);
            resolve(obj.data.sid);
          } else {
            magicJS.logWarning(`登录DownloadStation失败，接口响应：${data}`);
            reject(data);
          }
        } catch (err) {
          magicJS.logError(`登录DownloadStation失败，执行异常：${err}，接口响应：${data}`);
          reject(data);
        }
      }
    });
  });
}

function AddTask(synoUrl, sid, uri, destination) {
  return new Promise((resolve) => {
    let options = {
      url: `${synoUrl}/webapi/DownloadStation/task.cgi`,
      body: `_sid=${sid}&api=SYNO.DownloadStation.Task&version=1&method=create&uri=${uri}&destination=${destination}`,
    };
    magicJS.post(options, (err, resp, data) => {
      if (err) {
        magicJS.logError(`添加下载任务失败，请求异常：${err}`);
        resolve(false);
      } else {
        try {
          let obj = typeof data === "string" ? JSON.parse(data) : data;
          if (obj.success === true) {
            resolve(true);
          } else {
            magicJS.logWarning(`添加下载任务失败，接口响应：${data}`);
            resolve(false);
          }
        } catch (err) {
          magicJS.logError(`添加下载任务失败，执行异常：${err}，接口响应：${data}`);
          resolve(false);
        }
      }
    });
  });
}

/**
 * 获取DownloadStation下载位置
 * @param {*} synoUrl
 * @param {*} sid
 * @returns
 */
function GetLocation(synoUrl, sid) {
  return new Promise((resolve) => {
    let options = {
      url: `${synoUrl}/webapi/DownloadStation/entry.cgi`,
      body: `_sid=${sid}&api=SYNO.DownloadStation2.Settings.Location&version=1&method=get`,
    };
    magicJS.post(options, (err, resp, data) => {
      if (err) {
        magicJS.logError(`添加下载任务失败，请求异常：${err}`);
        resolve(false);
      } else {
        try {
          let obj = typeof data === "string" ? JSON.parse(data) : data;
          if (obj.success === true) {
            resolve(obj.data.default_destination);
          } else {
            magicJS.logWarning(`添加下载任务失败，接口响应：${data}`);
            resolve("");
          }
        } catch (err) {
          magicJS.logError(`添加下载任务失败，执行异常：${err}，接口响应：${data}`);
          resolve("");
        }
      }
    });
  });
}

/**
 * 创建下载目录
 * @param {*} synoUrl
 * @param {*} sid
 * @param {*} folderPath
 * @param {*} folderName
 * @returns
 */
function CreateFolder(synoUrl, sid, folderPath, folderName) {
  magicJS.logDebug(`创建下载目录，路径: ${folderPath}/${folderName}`);
  return new Promise((resolve, reject) => {
    let options = {
      url: encodeURI(`${synoUrl}/webapi/entry.cgi?_sid=${sid}&api=SYNO.FileStation.CreateFolder&force_parent=true&version=2&method=create&folder_path=["${folderPath}"]&name=["${folderName}"]`),
    };
    magicJS.get(options, (err, resp, data) => {
      if (err) {
        magicJS.logError(`创建文件目录失败，请求异常：${err}`);
        reject(false);
      } else {
        try {
          let obj = typeof data === "string" ? JSON.parse(data) : data;
          if (obj.success === true) {
            resolve(true);
          } else if (obj.error.code === "119") {
            magicJS.logWarning(`SID not found，接口响应：${data}`);
            magicJS.write(synoSidKey, "");
            reject(data);
          } else {
            magicJS.logWarning(`创建文件目录失败，接口响应：${data}`);
            reject(data);
          }
        } catch (err) {
          magicJS.logError(`创建文件目录失败，执行异常：${err}，接口响应：${data}`);
          reject(data);
        }
      }
    });
  });
}

async function CreateTasks(mediaList, synoSid, synoUrl, synoAccount, synoPasswd, appName, userName) {
  let decodeMediaList = [];
  let synoDestination = "";
  for (let i = 0; i < mediaList.length; i++) {
    let decodeMedia = escape(mediaList[i]);
    decodeMediaList.push(decodeMedia);
  }
  // 获取sid
  if (!!!synoSid) {
    await SynoAuth(synoUrl, synoAccount, synoPasswd)
      .then((value) => {
        synoSid = value;
        magicJS.write(synoSidKey, synoSid);
      })
      .catch((err) => magicJS.notify(`登录失败，异常信息：${err}`));
  }
  // 获取DownloadStation默认目录
  await GetLocation(synoUrl, synoSid).then((value) => {
    synoDestination = value;
    magicJS.logDebug(`当前下载目录：${synoDestination}`);
  });
  // 根据用户名创建目录
  await magicJS
    .retry(CreateFolder, 1, 100)(synoUrl, synoSid, `/${synoDestination}/${appName}`, userName)
    .then((value) => {
      if (value === true) {
        magicJS.logDebug("创建下载目录成功");
      }
    })
    .catch((err) => {
      magicJS.logError(`在群晖上创建目录失败，异常信息：${err}`);
    });
  // 添加下载任务
  let uri = decodeMediaList.join(",");
  let downloadDir = `${synoDestination}/${appName}/${userName}`;
  let result = await AddTask(synoUrl, synoSid, uri, downloadDir);
  if (result === true) {
    magicJS.notify(`${scriptName}`, `添加成功 ${downloadDir}`, `${mediaList.join("\n")}`);
  } else if (mediaList) {
    magicJS.notify(scriptName, `添加失败 ${downloadDir}`, `${mediaList.join("\n")}`);
  } else {
    magicJS.notify(scriptName, `添加失败 ${downloadDir}`);
  }
  return { synoSid, synoDestination };
}

(async () => {
  // 群晖 Download Station 配置检查
  let synoUrl = magicJS.read(synoUrlKey);
  let synoAccount = magicJS.read(synoAccountKey);
  let synoPasswd = magicJS.read(synoPasswdKey);
  let synoSid = magicJS.read(synoSidKey);
  let mediaList = [];
  let appName = "";
  let userName = "";
  if (!synoUrl || !synoAccount || !synoPasswd) {
    magicJS.logWarning("请先在BoxJS中配置DownloadStation");
    magicJS.notify("请先在BoxJS中配置DownloadStation");
    return;
  }

  if (magicJS.isResponse) {
    // Twitter收藏下载
    if (/^https?:\/\/api\.twitter\.com\/[0-9.]*\/favorites\/create.json/.test(magicJS.request.url) === true) {
      try {
        appName = "twitter";
        // 获取媒体url
        let obj = JSON.parse(magicJS.response.body);
        if (obj.extended_entities && obj.extended_entities.media) {
          obj.extended_entities.media.forEach((element) => {
            // 使用推文作者名称作为子目录名
            userName = obj.user.screen_name.replace(/[^a-zA-Z0-9]+/, "");
            magicJS.logDebug(`当前推文的用户：${userName}`);
            if (element.type == "photo") {
              mediaList.push(element.media_url);
            } else if (element.type == "video") {
              let maxBitrate = 0;
              let videoUrl = "";
              element.video_info.variants.forEach((video) => {
                if (video.bitrate && video.bitrate > maxBitrate) {
                  maxBitrate = video.bitrate;
                  videoUrl = video.url;
                }
              });
              mediaList.push(videoUrl);
              magicJS.logDebug(videoUrl);
            }
          });
        }
      } catch (err) {
        magicJS.logError(`添加下载任务失败，异常信息：${err}`);
        magicJS.notify("添加下载任务失败，请查阅日志");
      }
    }

    // 添加下载任务至DownloadStation
    if (mediaList.length > 0) {
      await CreateTasks(mediaList, synoSid, synoUrl, synoAccount, synoPasswd, appName, userName);
    }
  } else {
    await SynoAuth(synoUrl, synoAccount, synoPasswd)
      .then((value) => {
        let msg = `登录成功，获取Sid：${value}`;
        magicJS.notify(msg);
        magicJS.logInfo(msg);
        synoSid = value;
        magicJS.write(synoSidKey, synoSid);
      })
      .catch((err) => {
        magicJS.notify(`登录失败，异常信息：${err}`);
      });
  }
  magicJS.done();
})();

// prettier-ignore
function MagicJS(scriptName="MagicJS",logLevel="INFO"){return new class{constructor(){if(this.version="2.2.3.3",this.scriptName=scriptName,this.logLevels={DEBUG:5,INFO:4,NOTIFY:3,WARNING:2,ERROR:1,CRITICAL:0,NONE:-1},this.isLoon="undefined"!=typeof $loon,this.isQuanX="undefined"!=typeof $task,this.isJSBox="undefined"!=typeof $drive,this.isNode="undefined"!=typeof module&&!this.isJSBox,this.isSurge="undefined"!=typeof $httpClient&&!this.isLoon,this.node={request:void 0,fs:void 0,data:{}},this.iOSUserAgent="Mozilla/5.0 (iPhone; CPU iPhone OS 13_3_1 like Mac OS X) AppleWebKit/605.1.15 (KHTML, like Gecko) Version/13.0.5 Mobile/15E148 Safari/604.1",this.pcUserAgent="Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/84.0.4147.125 Safari/537.36 Edg/84.0.522.59",this.logLevel=logLevel,this._barkUrl="",this.isNode){this.node.fs=require("fs"),this.node.request=require("request");try{this.node.fs.accessSync("./magic.json",this.node.fs.constants.R_OK|this.node.fs.constants.W_OK)}catch(err){this.node.fs.writeFileSync("./magic.json","{}",{encoding:"utf8"})}this.node.data=require("./magic.json")}else this.isJSBox&&($file.exists("drive://MagicJS")||$file.mkdir("drive://MagicJS"),$file.exists("drive://MagicJS/magic.json")||$file.write({data:$data({string:"{}"}),path:"drive://MagicJS/magic.json"}))}set barkUrl(url){this._barkUrl=url.replace(/\/+$/g,"")}set logLevel(level){this._logLevel="string"==typeof level?level.toUpperCase():"DEBUG"}get logLevel(){return this._logLevel}get isRequest(){return"undefined"!=typeof $request&&"undefined"==typeof $response}get isResponse(){return"undefined"!=typeof $response}get request(){return"undefined"!=typeof $request?$request:void 0}get response(){return"undefined"!=typeof $response?($response.hasOwnProperty("status")&&($response.statusCode=$response.status),$response.hasOwnProperty("statusCode")&&($response.status=$response.statusCode),$response):void 0}get platform(){return this.isSurge?"Surge":this.isQuanX?"Quantumult X":this.isLoon?"Loon":this.isJSBox?"JSBox":this.isNode?"Node.js":"Unknown"}read(key,session=""){let val="";this.isSurge||this.isLoon?val=$persistentStore.read(key):this.isQuanX?val=$prefs.valueForKey(key):this.isNode?val=this.node.data:this.isJSBox&&(val=$file.read("drive://MagicJS/magic.json").string);try{this.isNode&&(val=val[key]),this.isJSBox&&(val=JSON.parse(val)[key]),session&&("string"==typeof val&&(val=JSON.parse(val)),val=val&&"object"==typeof val?val[session]:null)}catch(err){this.logError(err),val=session?{}:null,this.del(key)}void 0===val&&(val=null);try{val&&"string"==typeof val&&(val=JSON.parse(val))}catch(err){}return this.logDebug(`READ DATA [${key}]${session?`[${session}]`:""}(${typeof val})\n${JSON.stringify(val)}`),val}write(key,val,session=""){let data=session?{}:"";if(session&&(this.isSurge||this.isLoon)?data=$persistentStore.read(key):session&&this.isQuanX?data=$prefs.valueForKey(key):this.isNode?data=this.node.data:this.isJSBox&&(data=JSON.parse($file.read("drive://MagicJS/magic.json").string)),session){try{"string"==typeof data&&(data=JSON.parse(data)),data="object"==typeof data&&data?data:{}}catch(err){this.logError(err),this.del(key),data={}}this.isJSBox||this.isNode?(data[key]&&"object"==typeof data[key]||(data[key]={}),data[key].hasOwnProperty(session)||(data[key][session]=null),void 0===val?delete data[key][session]:data[key][session]=val):void 0===val?delete data[session]:data[session]=val}else this.isNode||this.isJSBox?void 0===val?delete data[key]:data[key]=val:data=void 0===val?null:val;"object"==typeof data&&(data=JSON.stringify(data)),this.isSurge||this.isLoon?$persistentStore.write(data,key):this.isQuanX?$prefs.setValueForKey(data,key):this.isNode?this.node.fs.writeFileSync("./magic.json",data):this.isJSBox&&$file.write({data:$data({string:data}),path:"drive://MagicJS/magic.json"}),this.logDebug(`WRITE DATA [${key}]${session?`[${session}]`:""}(${typeof val})\n${JSON.stringify(val)}`)}del(key,session=""){this.logDebug(`DELETE KEY [${key}]${session?`[${session}]`:""}`),this.write(key,null,session)}notify(title=thi