// ==UserScript==
// @name 百度网盘共享文件库目录清单导出
// @namespace https://github.com/Avens666/BaidunNetDisk-script
// @version 0.3
// @description try to take over the world!
// @author Avens
// @match https://pan.baidu.com/mbox*
// @grant none
// @require https://cdnjs.cloudflare.com/ajax/libs/FileSaver.js/1.3.3/FileSaver.min.js
// @require https://code.jquery.com/jquery-3.1.1.min.js
// ==/UserScript==
(function() {
'use strict';
// Your code here...
var classMap = {
'title': 'sharelist-view-toggle',
};
var g_type=0;
function exportDirInfo(){
//遍历当前显示的目录及文件信息
//step1 取得gid 一个分享文件库的主ID 有两种type，一种使用form_uk 加gid查询 这是type2，一种使用 form_uk 加 to_uk查询,这是type1
var gid, _frm, _to;
var obj = document.getElementsByTagName("li");
var obj_length = obj.length;
var frm_arr = [];
var to_arr = [];
var selected_arr = [];
var dirinfo = []; //一级列表数组
for(var i = 0 ; i < obj_length ; i++) {
//获取每个子元素的 id 值，1-10 之间的保留
//alert(tag_obj[i].getAttribute('id'));
var nodetype = obj[i].getAttribute("node-type");
if( nodetype && nodetype == "sharelist-item") {
var is_on = obj[i].getAttribute("class");
console.log("ison: "+ is_on);
selected_arr.push(is_on);
}
else
{
continue;
}
_frm = obj[i].getAttributeNode("data-frm");
if( _frm ) {
frm_arr.push(_frm);
}
if( gid == undefined ) {
gid = obj[i].getAttributeNode("data-gid");
if(gid) g_type = 2; //取得gid
}
_to = obj[i].getAttributeNode("data-to");
if( _to ) {
to_arr.push(_to);
g_type = 1;
}
}
if( g_type === 0) {
alert("请在文件库根目录下执行!" );
return;
}
//step2 根据gid获得第一级目录列表
var gid_url ="";
if(g_type == 2)
{
// https://pan.baidu.com/mbox/group/listshare?gid=837398806018187810&limit=50&desc=1&type=2
gid_url = " https://pan.baidu.com/mbox/group/listshare?gid=" + gid.value + "&limit=50&desc=1&type=2";
$.ajax({
type:'GET',
url: gid_url,
data:{},
dataType: "json",
async: false,
success:function(res){
// var blob = new Blob(["Hello, world!"], {type: "text/plain;charset=utf-8"});
// saveAs(res, "hello.txt");
if(res.errno != 0)
{
console.warn("errno:" + res.errno + "url: " + gid_url);
return;
}
var msg_lst = res.records.msg_list;
var obj_length = msg_lst.length;
var info="" ;
var index = 0;
for(var i = 0 ; i < obj_length ; i++) {
var file_lst = msg_lst[i].file_list;
var _msg_id = msg_lst[i].msg_id;
var len = file_lst.length;
if( selected_arr[index] == "on" )
{//目录被选中才导出
for(var n = 0 ; n < len ; n++) {
// console.log( file_lst[n].server_filename);
var _name = file_lst[n].server_filename;
var _size = file_lst[n].size;
var _isDir = file_lst[n].isdir;
var _fs_id = file_lst[n].fs_id;
var dinfo = createFileInfo(_name, _isDir, _size, _fs_id, _msg_id, frm_arr[index].value, 0);
dirinfo.push(dinfo);
info = info + "\n" + dinfo.getInfo();
}
}
index++;
}
console.log( info );
},
error:function(err){
console.log(err);
}
});
//Step3 遍历主目录
queryDir(dirinfo, gid.value);// 传gid
}
else if(g_type == 1)
{
// https://pan.baidu.com/mbox/msg/sessioninfo?to_uk=2181389256&type=2
gid_url = "https://pan.baidu.com/mbox/msg/sessioninfo?to_uk=" + frm_arr[0].value + "&type=2";
// alert(gid_url);
$.ajax({
type:'GET',
url: gid_url,
data:{},
dataType: "json",
async: false,
success:function(res){
if(res.errno != 0)
{
console.log("errno:" + res.errno + "url: " + gid_url);
return;
}
var rlist = res.records.list;
var obj_length = rlist.length;
var info="" ;
var index = 0;
for(var i = 0 ; i < obj_length ; i++) {
var frm = rlist[i].from_uk;
var to_uk = rlist[i].to_uk;
var _msg_id = rlist[i].msg_id;
var filelist = rlist[i].filelist.list; //注意节点
var len = filelist.length;
//console.log( "22msg: " + _msg_id );
for(var n = 0 ; n < len ; n++) {
if( selected_arr[index] == "on" )
{//目录被选中才导出
var _name = filelist[n].server_filename;
var _size = filelist[n].size;
var _isDir = filelist[n].isdir;
var _fs_id = filelist[n].fs_id;
var dinfo = createFileInfo(_name, _isDir, _size, _fs_id, _msg_id, frm, to_uk); ;
dirinfo.push(dinfo);
info += dinfo.getInfo() + "\n";
}
index++;
}
}
// alert( info );
},
error:function(err){
console.error(err);
}
});
//Step3 遍历主目录
queryDir(dirinfo, 0, 1);
}
var output="";
for(var n = 0 ; n < dirinfo.length ; n++)
{
// console.log(dirinfo[n].getAllInfo("") );
output +=dirinfo[n].getAllInfo("", 0) +"\r\n";
}
output += "<end>";
var blob = new Blob([output], {type: "text/plain;charset=utf-8"});
saveAs(blob, "exportDirList.txt");
}
function queryDir(dir_info, gidv, bfirst){
var obj_length = dir_info.length;
for(var i = 0 ; i < obj_length ; i++) {
if(dir_info[i].isDir == 0)
{//文件跳过
continue;
}
var dir_url ="";
var fileinfo = [];
var b_has_dir=0;
var n_page=1;
var b_more = 1;
var is_failed_p = 0;
while(n_page <= 30 && b_more == 1 || is_failed_p > 0) //一次最多取100个，超出了要分页取 z最多取3000
{
if(is_failed_p > 0 )
{
console.log("is_failed_p:" +is_failed_p + " page: " + n_page + " dir_info[i].isDir: " + dir_info[i].isDir);
if(is_failed_p > 2 ) //多试几次
{
break;
}
}
if(g_type == 1)
{
//https://pan.baidu.com/mbox/msg/shareinfo?msg_id=2503597858817404855&page=1&from_uk=2181389256&to_uk=1817073614&type=1&fs_id=929228779091096&num=50
dir_url ="https://pan.baidu.com/mbox/msg/shareinfo?msg_id=" + dir_info[i].msg_id + "&page=" + n_page + "&from_uk=" + dir_info[i].from_uk + "&to_uk=" + dir_info[i].to_uk +"&type=1&fs_id="+ dir_info[i].fs_id +"&num=100";
}
else if(g_type == 2)
{
dir_url ="https://pan.baidu.com/mbox/msg/shareinfo?msg_id=" + dir_info[i].msg_id + "&page=" + n_page + "&from_uk=" + dir_info[i].from_uk + "&gid=" + gidv +"&type=2&fs_id="+ dir_info[i].fs_id +"&num=100";
}
console.log(dir_url);
$.ajax({
type:'GET',
url:dir_url,
data:{},
dataType: "json",
async: false,
success:function(res){
var info="" ;
if(res.errno != 0)
{
console.warn("errno:" + res.errno + "url: " + dir_url);
is_failed_p ++;
return;
}
is_failed_p =0; //成功重置
var file_lst = res.records;
var len = file_lst.length;
b_more = res.has_more;
for(var n = 0 ; n < len ; n++) {
var _name = file_lst[n].server_filename;
var _size = file_lst[n].size;
var _isDir = file_lst[n].isdir;
var _fs_id = file_lst[n].fs_id;
//console.log( "11msg: " + dir_info[i] );
//console.log( "33msg: " + dir_info[i].msg_id );
var dinfo = createFileInfo(_name, _isDir, _size, _fs_id, dir_info[i].msg_id, dir_info[i].from_uk, dir_info[i].to_uk );
dinfo.path = "";
var lastn = file_lst[n].path.lastIndexOf("/") ;
if(lastn > 0)
dinfo.path = file_lst[n].path.substring(0,lastn+1);
else
dinfo.path = file_lst[n].path;
fileinfo.push(dinfo);
info = info + "\n" + dinfo.getInfo();
if(_isDir == 1)
{
dir_info[i].num_subdir += 1;
b_has_dir =1;
}
else
{
dir_info[i].num_subfile += 1;
}
}
n_page++;
// alert( info );
// if(b_has_dir == 1) queryDir(fileinfo, gidv, 0); //有目录 递归调用
},
error:function(err){
console.error(err);
is_failed_p ++;
}
});
}
dir_info[i].sub_file_objs = fileinfo;
if(b_has_dir == 1) queryDir(fileinfo, gidv, 0); //有目录 递归调用
}
}
// 目录或文件对象
function createFileInfo(name,isDir,size, fs_id, msg_id, frm, to) {
var ofile = new Object;
ofile.name = name; //文件(夹)名称
ofile.isDir = isDir; //是否文件夹
ofile.size = size; //文件大小
ofile.fs_id= fs_id;
ofile.msg_id= msg_id;
ofile.from_uk=frm;
ofile.to_uk=to;
ofile.num_subfile = 0; //子文件数量
ofile.num_subdir = 0; //子目录数量
ofile.num_allfile = 0; //递归子目录数量
ofile.num_alldir = 0; //递归子文件数量
ofile.sub_file_objs=[]; //子文件对象
ofile.path="";
ofile.getSize = function( s) //返回尺寸
{
if(s < 1024)
{
return s+" Byte";
}
else if (s > 1024 && s < 1048576 ) //1024*1024
{
return Math.round(s/1024 * 100) / 100.00 +" KB";
}
else if (s> 1048576 && s < 1073741824) // 1024*1024*1024
{
return Math.round(s/(1048576) * 100) / 100.00+" MB";
}
else
{
return Math.round(s/(1073741824) * 100) / 100.00+" GB";
}
};
ofile.getInfo = function() {
if(this.isDir != 0)
{ //文件夹
return this.name + " [文件夹大小:" + this.getSize(this.size ) + " 子文件夹数: " + this.num_subdir + " 子文件数: " + this.num_subfile +"]" ;
}
else
{//文件
return this.name + " (" + this.getSize( this.size ) + ")" ;
}
};
ofile.getAllInfo = function( blank, level ) {
var info;
if(this.isDir != 0)
{ //文件夹
info = blank+"├─" + this.name + " [文件夹大小:" + this.getSize( this.getAllSize() ) + " 子文件夹数: " + this.num_subdir + " 子文件数: " + this.num_subfile +"]\r\n";
level++;
for(var n = 0 ; n < this.sub_file_objs.length ; n++) {
info += level+ this.sub_file_objs[n].getAllInfo(blank+"│ ", level);
}
}
else
{//文件
info = blank+ "│ "+this.name + " (" + this.getSize( this.size ) + ") -- " + this.path + "\r\n";
}
return info;
};
ofile.getAllSize= function(){
if(this.isDir == 1)
{ //文件夹
this.size=0;
for(var n = 0 ; n < this.sub_file_objs.length ; n++) {
this.size += this.sub_file_objs[n].getAllSize();
}
}
return this.size;
};
return ofile;
}
//添加导出按钮
function addButton() {
var $dropdownbutton = $('<a class="list-filter" href="javascript:void( 0 );" onclick="exportDirInfo()" node-type="btn_export" title="在文件库根目录，选择要导出的文件夹，点击按钮，清单生成后会提示保存文件。\n如果文件夹内部子文件夹很多，需要等待较长时间。8000个子文件夹大概1小时" style="display: inline;">导出目录信息</a>');
$('.' + classMap['title']).append($dropdownbutton);
$dropdownbutton.click(exportDirInfo);
}
addButton();
// alert("导出");
})();
