tmn
314 lines
12 KiB
JavaScript
314 lines
12 KiB
JavaScript
// 当popup页面的DOM内容完全加载后执行
|
||
document.addEventListener('DOMContentLoaded', function () {
|
||
// 获取页面中的按钮元素
|
||
const getStudentsBtn = document.getElementById('get-students-btn');
|
||
const getBlacklistBtn = document.getElementById('get-blacklist-btn');
|
||
const dataDisplay = document.getElementById('data-display');
|
||
|
||
// 全局变量存储企业信息
|
||
let enterpriseInfo = {
|
||
corpId: null,
|
||
corpName: null
|
||
};
|
||
|
||
/**
|
||
* 获取企业信息
|
||
* @param {number} tabId - 标签页ID
|
||
* @returns {Object} 企业信息 {corpId, corpName}
|
||
*/
|
||
async function getEnterpriseInfo(tabId) {
|
||
try {
|
||
const enterpriseResult = await chrome.scripting.executeScript({
|
||
target: {tabId: tabId},
|
||
func: () => {
|
||
return fetch('https://work.weixin.qq.com/wework_admin/profile/quotaupdate/log?lang=zh_CN&f=json', {
|
||
method: 'GET',
|
||
credentials: 'include'
|
||
}).then(r => r.json());
|
||
}
|
||
});
|
||
|
||
// 提取企业信息
|
||
const enterpriseData = enterpriseResult[0].result;
|
||
if (enterpriseData && enterpriseData.data && enterpriseData.data.basecertinfo) {
|
||
const info = {
|
||
corpId: enterpriseData.data.basecertinfo.appid,
|
||
corpName: enterpriseData.data.basecertinfo.subject_name
|
||
};
|
||
|
||
console.log('企业信息获取成功:', info);
|
||
return info;
|
||
} else {
|
||
throw new Error('无法获取企业信息：响应数据格式异常');
|
||
}
|
||
} catch (error) {
|
||
console.error('获取企业信息失败:', error);
|
||
throw new Error(`获取企业信息失败: ${error.message}`);
|
||
}
|
||
}
|
||
|
||
// 获取学员列表按钮的点击事件监听器
|
||
getStudentsBtn.addEventListener('click', async () => {
|
||
// 获取当前激活的标签页
|
||
const [tab] = await chrome.tabs.query({active: true, currentWindow: true});
|
||
|
||
try {
|
||
// 第一步：获取企业信息
|
||
dataDisplay.textContent = '正在获取企业信息...';
|
||
enterpriseInfo = await getEnterpriseInfo(tab.id);
|
||
|
||
// 第二步：获取所有学员数据
|
||
dataDisplay.textContent = '正在获取学员数据...';
|
||
const allStudentsData = await getAllStudentsData(tab.id);
|
||
|
||
// 第三步：发送数据到指定接口
|
||
dataDisplay.textContent = '正在发送数据...';
|
||
await sendStudentsData(allStudentsData, enterpriseInfo);
|
||
|
||
// 完成
|
||
dataDisplay.textContent = `✅ 学员数据处理完成！\n\n企业信息:\nAppID: ${enterpriseInfo.corpId}\n企业名称: ${enterpriseInfo.corpName}\n\n学员数量: ${allStudentsData.length}\n\n数据已成功发送到服务器。`;
|
||
|
||
} catch (error) {
|
||
console.error('处理失败:', error);
|
||
dataDisplay.textContent = `❌ 处理失败: ${error.message}`;
|
||
}
|
||
});
|
||
|
||
// 获取黑名单按钮的点击事件监听器
|
||
getBlacklistBtn.addEventListener('click', async () => {
|
||
// 获取当前激活的标签页
|
||
const [tab] = await chrome.tabs.query({active: true, currentWindow: true});
|
||
|
||
try {
|
||
// 第一步：获取企业信息
|
||
dataDisplay.textContent = '正在获取企业信息...';
|
||
enterpriseInfo = await getEnterpriseInfo(tab.id);
|
||
|
||
// 第二步：获取所有黑名单数据
|
||
dataDisplay.textContent = '正在获取黑名单数据...';
|
||
const allBlacklistData = await getAllBlacklistData(tab.id);
|
||
|
||
// 第三步：发送数据到指定接口
|
||
dataDisplay.textContent = '正在发送数据...';
|
||
await sendBlacklistData(allBlacklistData, enterpriseInfo);
|
||
|
||
// 完成
|
||
dataDisplay.textContent = `✅ 黑名单数据处理完成！\n\n企业信息:\nAppID: ${enterpriseInfo.corpId}\n企业名称: ${enterpriseInfo.corpName}\n\n黑名单数量: ${allBlacklistData.length}\n\n数据已成功发送到服务器。`;
|
||
|
||
} catch (error) {
|
||
console.error('处理失败:', error);
|
||
dataDisplay.textContent = `❌ 处理失败: ${error.message}`;
|
||
}
|
||
});
|
||
|
||
/**
|
||
* 获取所有学员数据（处理分页）
|
||
* @param {number} tabId - 标签页ID
|
||
* @returns {Array} 所有学员数据
|
||
*/
|
||
async function getAllStudentsData(tabId) {
|
||
let allStudents = [];
|
||
let lastPageMaxId = '';
|
||
let hasMore = true;
|
||
let pageCount = 0;
|
||
|
||
while (hasMore) {
|
||
pageCount++;
|
||
dataDisplay.textContent = `正在获取学员数据... 第${pageCount}页，已获取${allStudents.length}条`;
|
||
|
||
// 构建URL编码的请求参数（application/x-www-form-urlencoded）
|
||
const params = new URLSearchParams();
|
||
|
||
params.append('page', pageCount.toString());
|
||
params.append('time_since', '1744560000');
|
||
params.append('time_before', '1744991999');
|
||
params.append('last_page_max_id', lastPageMaxId);
|
||
params.append('perpage', '10000'); // 每页100条，可根据实际情况调整
|
||
|
||
const result = await chrome.scripting.executeScript({
|
||
target: {tabId: tabId},
|
||
func: (formDataStr) => {
|
||
return fetch('https://work.weixin.qq.com/wework_admin/customer/list?lang=zh_CN&f=json', {
|
||
method: 'POST',
|
||
headers: {
|
||
'Content-Type': 'application/x-www-form-urlencoded', // 关键：设置正确的Content-Type
|
||
},
|
||
body: formDataStr,
|
||
credentials: 'include'
|
||
}).then(r => r.json());
|
||
},
|
||
args: [params.toString()]
|
||
});
|
||
|
||
const data = result[0].result;
|
||
|
||
if (data && data.data) {
|
||
// 更新数据解析逻辑：从 data.data.detail_list.items 获取学员数据
|
||
if (data.data.detail_list && data.data.detail_list.items && data.data.detail_list.items.length > 0) {
|
||
allStudents = allStudents.concat(data.data.detail_list.items);
|
||
}
|
||
|
||
// 更新分页参数：使用 has_more 判断是否还有更多数据
|
||
hasMore = data.data.has_more || false;
|
||
lastPageMaxId = data.data.max_id || '';
|
||
|
||
console.log(`第${pageCount}页学员获取完成，本页${data.data.detail_list?.items?.length || 0}条，总计${allStudents.length}条，has_more: ${hasMore}`);
|
||
|
||
// 避免请求过快，添加小延迟
|
||
await new Promise(resolve => setTimeout(resolve, 100));
|
||
|
||
} else {
|
||
throw new Error('获取学员数据失败：响应数据格式异常');
|
||
}
|
||
}
|
||
|
||
console.log('学员数据获取完成，总计:', allStudents.length, '条');
|
||
return allStudents;
|
||
}
|
||
|
||
/**
|
||
* 获取所有黑名单数据（处理分页）
|
||
* @param {number} tabId - 标签页ID
|
||
* @returns {Array} 所有黑名单数据
|
||
*/
|
||
async function getAllBlacklistData(tabId) {
|
||
let allBlockedCustomers = [];
|
||
let lastPageMaxId = '';
|
||
let isEnd = false;
|
||
let pageCount = 0;
|
||
|
||
while (!isEnd) {
|
||
pageCount++;
|
||
dataDisplay.textContent = `正在获取黑名单数据... 第${pageCount}页，已获取${allBlockedCustomers.length}条`;
|
||
|
||
// 构建请求URL
|
||
const url = `https://work.weixin.qq.com/wework_admin/groupchat/blockedCustomers?lang=zh_CN&f=json&pageSize=15000&last_page_max_id=${lastPageMaxId}`;
|
||
|
||
const result = await chrome.scripting.executeScript({
|
||
target: {tabId: tabId},
|
||
func: (requestUrl) => {
|
||
return fetch(requestUrl, {
|
||
method: 'GET',
|
||
credentials: 'include'
|
||
}).then(r => r.json());
|
||
},
|
||
args: [url]
|
||
});
|
||
|
||
const data = result[0].result;
|
||
|
||
if (data && data.data) {
|
||
// 添加当前页的黑名单数据到总数组
|
||
if (data.data.blockedCustomers && data.data.blockedCustomers.length > 0) {
|
||
allBlockedCustomers = allBlockedCustomers.concat(data.data.blockedCustomers);
|
||
}
|
||
|
||
// 更新分页参数
|
||
isEnd = data.data.is_end;
|
||
lastPageMaxId = data.data.last_page_max_id || '';
|
||
|
||
console.log(`第${pageCount}页黑名单获取完成，本页${data.data.blockedCustomers?.length || 0}条，总计${allBlockedCustomers.length}条，is_end: ${isEnd}`);
|
||
|
||
// 避免请求过快，添加小延迟
|
||
await new Promise(resolve => setTimeout(resolve, 200));
|
||
|
||
} else {
|
||
throw new Error('获取黑名单数据失败：响应数据格式异常');
|
||
}
|
||
}
|
||
|
||
console.log('黑名单数据获取完成，总计:', allBlockedCustomers.length, '条');
|
||
return allBlockedCustomers;
|
||
}
|
||
|
||
/**
|
||
* 发送学员数据到指定接口
|
||
* @param {Array} studentsData - 学员数据
|
||
* @param {Object} enterpriseInfo - 企业信息
|
||
*/
|
||
async function sendStudentsData(studentsData, enterpriseInfo) {
|
||
// 构建要发送的数据
|
||
const payload = {
|
||
enterpriseInfo: enterpriseInfo,
|
||
studentsData: studentsData,
|
||
totalCount: studentsData.length,
|
||
fetchTime: new Date().toISOString()
|
||
};
|
||
|
||
console.log('准备发送学员数据:', {
|
||
enterprise: enterpriseInfo,
|
||
count: studentsData.length,
|
||
sample: studentsData.slice(0, 2) // 只显示前2条作为示例
|
||
});
|
||
|
||
// // TODO: 替换为您的实际接口地址
|
||
// const apiEndpoint = 'http://192.168.0.133:7788/wework/students/import';
|
||
//
|
||
// try {
|
||
// const response = await fetch(apiEndpoint, {
|
||
// method: 'POST',
|
||
// headers: {
|
||
// 'Content-Type': 'application/json',
|
||
// },
|
||
// body: JSON.stringify(payload)
|
||
// });
|
||
//
|
||
// if (!response.ok) {
|
||
// throw new Error(`服务器响应错误: ${response.status} ${response.statusText}`);
|
||
// }
|
||
//
|
||
// const result = await response.json();
|
||
// console.log('学员数据发送成功:', result);
|
||
//
|
||
// } catch (error) {
|
||
// console.error('发送学员数据失败:', error);
|
||
// throw new Error(`发送学员数据失败: ${error.message}`);
|
||
// }
|
||
}
|
||
|
||
/**
|
||
* 发送黑名单数据到指定接口
|
||
* @param {Array} blacklistData - 黑名单数据
|
||
* @param {Object} enterpriseInfo - 企业信息
|
||
*/
|
||
async function sendBlacklistData(blacklistData, enterpriseInfo) {
|
||
// 构建要发送的数据
|
||
const payload = {
|
||
enterpriseInfo: enterpriseInfo,
|
||
blacklistData: blacklistData,
|
||
totalCount: blacklistData.length,
|
||
fetchTime: new Date().toISOString()
|
||
};
|
||
|
||
console.log('准备发送黑名单数据:', {
|
||
enterprise: enterpriseInfo,
|
||
count: blacklistData.length,
|
||
sample: blacklistData.slice(0, 2) // 只显示前2条作为示例
|
||
});
|
||
|
||
// TODO: 替换为您的实际接口地址
|
||
const apiEndpoint = 'http://192.168.0.133:7788/wework/blacklist/import';
|
||
|
||
try {
|
||
const response = await fetch(apiEndpoint, {
|
||
method: 'POST',
|
||
headers: {
|
||
'Content-Type': 'application/json',
|
||
},
|
||
body: JSON.stringify(payload)
|
||
});
|
||
|
||
if (!response.ok) {
|
||
throw new Error(`服务器响应错误: ${response.status} ${response.statusText}`);
|
||
}
|
||
|
||
const result = await response.json();
|
||
console.log('黑名单数据发送成功:', result);
|
||
|
||
} catch (error) {
|
||
console.error('发送黑名单数据失败:', error);
|
||
throw new Error(`发送黑名单数据失败: ${error.message}`);
|
||
}
|
||
}
|
||
});
|