dr/cat/alist_open.js

import { _ } from './lib/cat.js';
import { findBestLCS } from './lib/similarity.js';

const http = async function (url, options = {}) {
    if (options.method == 'POST' && options.data) {
        options.body = JSON.stringify(options.data);
        options.headers = Object.assign({ 'content-type': 'application/json' }, options.headers);
    }
    const res = await req(url, options);
    res.json = () => (res.content ? JSON.parse(res.content) : null);
    res.text = () => res.content;
    return res;
};
['get', 'post'].forEach((method) => {
    http[method] = function (url, options = {}) {
        return http(url, Object.assign(options, { method: method.toUpperCase() }));
    };
});

const __drives = {};
const __subtitle_cache = {};

async function get_drives_path(tid) {
    const index = tid.indexOf('/', 1);
    const name = tid.substring(1, index);
    const path = tid.substring(index);
    return { drives: await get_drives(name), path };
}

async function get_drives(name) {
    const { settings, api, server } = __drives[name];
    if (settings.v3 == null) {
        //获取 设置
        settings.v3 = false;
        const data = (await http.get(server + '/api/public/settings')).json().data;
        if (_.isArray(data)) {
            settings.title = data.find((x) => x.key == 'title')?.value;
            settings.v3 = false;
            settings.version = data.find((x) => x.key == 'version')?.value;
            settings.enableSearch = data.find((x) => x.key == 'enable search')?.value == 'true';
        } else {
            settings.title = data.title;
            settings.v3 = true;
            settings.version = data.version;
            settings.enableSearch = false; //v3 没有找到 搜索配置
        }
        //不同版本 接口不一样
        api.path = settings.v3 ? '/api/fs/list' : '/api/public/path';
        api.file = settings.v3 ? '/api/fs/get' : '/api/public/path';
        api.search = settings.v3 ? '/api/public/search' : '/api/public/search';
        api.other = settings.v3 ? '/api/fs/other' : null;
    }
    return __drives[name];
}

let siteKey = '';
let siteType = 0;

function init(cfg) {
    siteKey = cfg.skey;
    siteType = cfg.stype;
    cfg.ext.forEach(
        (item) =>
            (__drives[item.name] = {
                name: item.name,
                server: item.server.endsWith('/') ? item.server.substring(0, item.server.length - 1) : item.server,
                startPage: item.startPage || '/', //首页
                showAll: item.showAll === true, //默认只显示 视频和文件夹，如果想显示全部 showAll 设置true
                params: item.params || {},
                _path_param: item.params
                    ? _.sortBy(Object.keys(item.params), function (x) {
                          return -x.length;
                      })
                    : [],
                settings: {},
                api: {},
                getParams(path) {
                    const key = this._path_param.find((x) => path.startsWith(x));
                    return Object.assign({}, this.params[key], { path });
                },
                async getPath(path) {
                    const res = (await http.post(this.server + this.api.path, { data: this.getParams(path) })).json();
                    return this.settings.v3 ? res.data.content : res.data.files;
                },
                async getFile(path) {
                    const res = (await http.post(this.server + this.api.file, { data: this.getParams(path) })).json();
                    const data = this.settings.v3 ? res.data : res.data.files[0];
                    if (!this.settings.v3) data.raw_url = data.url; //v2 的url和v3不一样
                    return data;
                },
                async getOther(method, path) {
                    const data = this.getParams(path);
                    data.method = method;
                    const res = (await http.post(this.server + this.api.other, { data: data })).json();
                    return res;
                },
                isFolder(data) {
                    return data.type == 1;
                },
                isVideo(data) {
                    //判断是否是 视频文件
                    return this.settings.v3 ? data.type == 2 : data.type == 3;
                },
                isSubtitle(data) {
                    if (data.type == 1) return false;
                    const ext = ['.srt', '.ass', '.scc', '.stl', '.ttml'];
                    return ext.some((x) => data.name.endsWith(x));
                },
                getType(data) {
                    const isVideo = this.isVideo(data);
                    return this.isFolder(data) ? 0 : isVideo ? 10 : 1;
                },
                getPic(data) {
                    let pic = this.settings.v3 ? data.thumb : data.thumbnail;
                    return pic || (this.isFolder(data) ? 'http://img1.3png.com/281e284a670865a71d91515866552b5f172b.png' : '');
                },
                getSize(data) {
                    let sz = data.size || 0;
                    if (sz <= 0) return '';
                    let filesize = '';
                    if (sz > 1024 * 1024 * 1024 * 1024.0) {
                        sz /= 1024 * 1024 * 1024 * 1024.0;
                        filesize = 'TB';
                    } else if (sz > 1024 * 1024 * 1024.0) {
                        sz /= 1024 * 1024 * 1024.0;
                        filesize = 'GB';
                    } else if (sz > 1024 * 1024.0) {
                        sz /= 1024 * 1024.0;
                        filesize = 'MB';
                    } else {
                        sz /= 1024.0;
                        filesize = 'KB';
                    }
                    return sz.toFixed(2) + filesize;
                },
                getRemark(data) {
                    return '';
                },
            })
    );
}

async function dir(dir, pg) {
    for (const k in __subtitle_cache) {
        delete __subtitle_cache[k];
    }
    pg = pg || 1;
    if (pg == 0) pg == 1;
    if (dir === '/' || dir === '') {
        const result = _.map(__drives, function (d) {
            return { name: d.name, path: '/' + d.name + d.startPage, type: 0, thumb: '' };
        });
        return JSON.stringify({
            parent: '',
            page: pg,
            pagecount: pg,
            list: result,
        });
    }

    let { drives, path } = await get_drives_path(dir);
    const id = dir.endsWith('/') ? dir : dir + '/';
    const list = await drives.getPath(path);
    let subtList = [];
    let videos = [];
    let allList = [];
    list.forEach((item) => {
        if (drives.isSubtitle(item)) subtList.push(item.name);
        const isVideo = drives.isVideo(item);
        if (!drives.showAll && !drives.isFolder(item) && !isVideo) return;
        const file = {
            name: item.name.replaceAll('$', '_').replaceAll('#', '_'),
            path: id + item.name + (drives.isFolder(item) ? '/' : ''),
            thumb: drives.getPic(item),
            type: drives.getType(item),
            size: drives.getSize(item),
            remark: drives.getRemark(item),
        };
        if (drives.isVideo(item)) videos.push(file);
        allList.push(file);
    });
    if (subtList.length > 0) {
        videos.forEach((item) => {
            var sbust = findBestLCS(item.name, subtList);
            if (sbust.bestMatch) __subtitle_cache[item.path] = [id + sbust.bestMatch.target];
        });
    }
    return JSON.stringify({
        parent: id,
        page: pg,
        pagecount: pg,
        list: allList,
    });
}

async function file(file) {
    let { drives, path } = await get_drives_path(file);
    const item = await drives.getFile(path);
    const subs = [];
    if (__subtitle_cache[file]) {
        for (const sub of __subtitle_cache[file]) {
            try {
                let subP = await get_drives_path(sub);
                const subItem = await drives.getFile(subP.path);
                subs.push(subItem.raw_url);
            } catch (error) {}
        }
    }
    if (item.provider === 'AliyundriveShare2Open' && drives.api.other) {
        const urls = ['原画', item.raw_url];
        try {
            const res = await drives.getOther('video_preview', path);
            for (const live of res.data.video_preview_play_info.live_transcoding_task_list) {
                if (live.status === 'finished') {
                    urls.push(live.template_id);
                    urls.push(live.url);
                }
            }
        } catch (error) {}
        const result = {
            name: item.name,
            url: urls,
            size: drives.getSize(item),
            remark: drives.getRemark(item),
            header: {},
            extra: {
                subt: subs,
            },
        };
        return JSON.stringify(result);
    } else if (item.provider === '123Pan') {
        let url = item.raw_url;
        try {
            url = (await http.get(url)).json().data.redirect_url;
        } catch (error) {}
        const result = {
            name: item.name,
            url: url,
            size: drives.getSize(item),
            remark: drives.getRemark(item),
            header: {},
            extra: {
                subt: subs,
            },
        };
        return JSON.stringify(result);
    } else {
        const result = {
            name: item.name,
            url: item.raw_url,
            size: drives.getSize(item),
            remark: drives.getRemark(item),
            header: {},
            extra: {
                subt: subs,
            },
        };
        return JSON.stringify(result);
    }
}

function search(wd) {
    return JSON.stringify({
        list: [],
    });
}

export function __jsEvalReturn() {
    return {
        init: init,
        dir: dir,
        file: file,
        search: search,
    };
}