node-weibo-simulation-api
Version:
241 lines (237 loc) • 8.57 kB
JavaScript
const request = require('request');
const {
urlToBase64
} = require('./pictureTool');
const fse = require('fs-extra');
const { NO_PICTURE } = require('../constans/code');
const format = require('./format');
const config = require('rc')('weibo', {
cookieErrorRetryTime: 3
});
exports.request = function (params) {
// console.log('request params', params);
return new Promise((resolve, reject) => {
// return resolve({
// statusCode:302,
// body:`{"code":"100000","msg":"","data":{"html":"<html></html>"}}`
// })
request(params, function (err, r, b) {
if (err) {
return reject(err);
} else {
if (r.statusCode >= 200 || r.statusCode < 400) {
return resolve(r);
} else {
return reject(r);
}
}
});
});
};
let retryTime = config.cookieErrorRetryTime;
exports.weiboRequest = function (params) {
const cookie = params.cookie;
// console.log('weiboRequest', params);
const host = 'https://weibo.com';
var headers = {
'Origin': host,
"Host": "weibo.com",
'User-Agent': 'Mozilla/5.0 (Macintosh; Intel Mac OS X 10_12_6) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/60.0.3112.113 Safari/537.36',
'Content-Type': 'application/x-www-form-urlencoded',
'Referer': host + '/friends'
};
let promise;
if (params.cookieStr) {
headers.Cookie = params.cookieStr;
promise = Promise.resolve();
} else {
promise = cookie.getWeiboCookie().then(cookieStr => {
headers.Cookie = cookieStr;
return Promise.resolve();
});
}
promise = promise.then(data => {
let requestParams = Object.assign({
url: host + params.url,
headers: headers,
method: params.method || 'POST',
form: params.form
}, params.extra);
return exports.request(requestParams).then(data => {
if (data.statusCode === 302) {
console.log('retryTime', retryTime);
//登录态问题，获取到登录态之后重试
//todo
if (retryTime > 0) {
retryTime--;
return cookie.getWeiboCookie({
force: true
}).then(() => {
return Promise.resolve();
}).then(() => {
return exports.weiboRequest(params);
});
} else {
return Promise.reject({
code: 9001,
message: "cookie 无效"
});
}
}
// console.log('status', data.statusCode);
// console.log('header', data.header);
// console.log('body', data.body);
return data;
});
});
return promise;
};
/**
* 微博批量上传图片
*/
exports.weiboMultiUploadPicture = function (params) {
// console.log('params',params);
let promises = [];
// console.log('pictureBase64Datas',params.pictureBase64Datas.length);
if (Array.isArray(params.pictureBase64Datas) && params.pictureBase64Datas.length > 0) {
promises = params.pictureBase64Datas.map(pic => {
return exports.weiboUploadPicture({
pictureBase64Data: pic,
cookie: params.cookie
});
});
} else if (Array.isArray(params.pictureUrls) && params.pictureUrls.length > 0) {
promises = params.pictureUrls.map(pic => {
return exports.weiboUploadPicture({
pictureUrl: pic,
cookie: params.cookie
});
});
} else if (Array.isArray(params.pictureFiles) && params.pictureFiles.length > 0) {
promises = params.pictureFiles.map(pic => {
return exports.weiboUploadPicture({
pictureFile: pic,
cookie: params.cookie
});
});
} else {
return Promise.reject({
code: NO_PICTURE,
msg: "图片不存在"
});
}
return Promise.all(promises);
};
/**
* 微博上传图片
*/
exports.weiboUploadPicture = async function (params) {
// console.log('params',params);
let base64Data = '';
//优先读取base64
if (params.pictureBase64Data) {
// console.log('here',);
base64Data = params.pictureBase64Data.split('base64,')[1] || params.pictureBase64Data.split('base64,')[0];
// console.log('base64Data',base64Data);
// return;
} else if (params.pictureUrl) {
try {
base64Data = await urlToBase64(params.pictureUrl);
} catch (error) {
console.log('error', error);
return Promise.reject(error);
}
} else if (params.pictureFile) {
try {
base64Data = await fse.readFile(params.pictureFile, 'base64');
} catch (error) {
console.log('error', error);
return Promise.reject(error);
}
// console.log('base64Data',base64Data);
} else {
return Promise.reject({
code: NO_PICTURE,
msg: "图片不存在"
});
}
const cookie = params.cookie;
const host = 'https://weibo.com';
var headers = {
'Origin': host,
'User-Agent': 'Mozilla/5.0 (Macintosh; Intel Mac OS X 10_12_6) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/60.0.3112.113 Safari/537.36',
'Content-Type': 'application/x-www-form-urlencoded',
'Referer': host + '/friends'
};
let promise = cookie.getWeiboCookie().then(cookieStr => {
headers.Cookie = cookieStr;
return Promise.resolve();
});
promise = promise.then(data => {
let requestParams = Object.assign({
url: 'https://picupload.weibo.com/interface/pic_upload.php',
qs: {
cb: `https://weibo.com/aj/static/upimgback.html?_wv=5&callback=STK_ijax_${new Date().getTime()}`,
mime: 'image/jpeg',
data: 'base64',
url: '0',
ori: '1', //原图
markpos: '1',
logo: '',
nick: '0',
marks: '0',
app: 'miniblog',
s: 'rdxt' },
headers: headers,
method: params.method || 'POST',
form: {
b64_data: base64Data
}
}, params.extra);
return exports.request(requestParams).then(data => {
// delete requestParams.form
// console.log('requestParams',requestParams);
// console.log('data.status',data.statusCode);
// console.log('data.body',data.body);
// console.log('data',data.headers);
// console.log('location',data.headers.location);
// return;
if (data.statusCode === 302) {
let response = format.parseUploadResponse(data.headers.location);
if (response.ret == 1 && response.pid) {
//上传成功
return {
pid: response.pid,
url: `http://ww3.sinaimg.cn/large/${response.pid}`
};
} else if (response.ret == -1) {
//cookie无效
console.log('retryTime', retryTime);
//登录态问题，获取到登录态之后重试
//todo
if (retryTime > 0) {
retryTime--;
return cookie.getWeiboCookie({
force: true
}).then(() => {
return Promise.resolve();
}).then(() => {
return exports.weiboUploadPicture(params);
});
} else {
return Promise.reject({
code: 9001,
message: "cookie 无效"
});
}
} else {
return Promise.reject({
code: response.ret,
message: "上传图片错误"
});
}
}
});
});
return promise;
};