269 lines
8.6 KiB
JavaScript
269 lines
8.6 KiB
JavaScript
// pythonBridge.js (Refactored to Electron IPC)
|
||
import { ref, onMounted } from 'vue';
|
||
import { isElectron, safeElectronCall } from '@/utils/electronBridge';
|
||
|
||
// Check if we are in Electron environment
|
||
const inElectron = isElectron();
|
||
|
||
export function usePythonBridge() {
|
||
// ========== tk爬虫的接口 ==========
|
||
|
||
|
||
// fetchDataConfig (maps to updateStartConfig)
|
||
const fetchDataConfig = async (data) => {
|
||
if (!inElectron) return null;
|
||
try {
|
||
return await window.electronAPI.tk.updateStartConfig(data);
|
||
} catch (e) {
|
||
console.error('fetchDataConfig error:', e);
|
||
return null;
|
||
}
|
||
};
|
||
|
||
// fetchDataCount
|
||
const fetchDataCount = async () => {
|
||
if (!inElectron) return JSON.stringify({ totalCount: 0 });
|
||
try {
|
||
return await window.electronAPI.tk.getDataCount();
|
||
} catch (e) {
|
||
console.error('fetchDataCount error:', e);
|
||
return JSON.stringify({ totalCount: 0 });
|
||
}
|
||
};
|
||
|
||
// loginTikTok
|
||
const loginTikTok = async () => {
|
||
if (!inElectron) return;
|
||
await window.electronAPI.tk.loginTikTok();
|
||
};
|
||
// loginTikTok
|
||
const loginBigTikTok = async () => {
|
||
if (!inElectron) return;
|
||
await window.electronAPI.tk.loginBigTikTok();
|
||
};
|
||
// loginBackStage
|
||
const loginBackStage = async (data) => {
|
||
if (!inElectron) return;
|
||
if (data.index == 0) {
|
||
await window.electronAPI.tk.loginBackStage(JSON.stringify(data));
|
||
} else if (data.index == 1) {
|
||
await window.electronAPI.tk.loginBackStageCopy(JSON.stringify(data));
|
||
}
|
||
};
|
||
|
||
// givePyAnchorId -> visitAnchor
|
||
const givePyAnchorId = async (id) => {
|
||
if (!inElectron) return;
|
||
await window.electronAPI.tk.visitAnchor(id);
|
||
};
|
||
|
||
// 查询后台登录状态（合并接口，通过 account 参数区分）
|
||
// account: 公会账号，不传则返回所有账号状态
|
||
const backStageloginStatus = async (account) => {
|
||
if (!inElectron) return null;
|
||
try {
|
||
const res = await window.electronAPI.tk.checkBackStageLoginStatus(account);
|
||
return typeof res === 'string' ? JSON.parse(res) : res;
|
||
} catch (e) {
|
||
console.error('backStageloginStatus error:', e);
|
||
return null;
|
||
}
|
||
};
|
||
|
||
// 兼容旧接口：查询副账号登录状态（内部调用合并后的接口）
|
||
const backStageloginStatusCopy = async (account) => {
|
||
return await backStageloginStatus(account);
|
||
};
|
||
|
||
// exportToExcel
|
||
const exportToExcel = async (data) => {
|
||
if (!inElectron) return;
|
||
await window.electronAPI.tk.exportData(JSON.stringify(data));
|
||
};
|
||
|
||
const stopScript = async () => {
|
||
if (!inElectron) return;
|
||
await window.electronAPI.tk.stopCrawl();
|
||
};
|
||
|
||
// getVersion
|
||
const getVersion = async () => {
|
||
if (!inElectron) return 'Web Mode';
|
||
return await window.electronAPI.tk.getVersion();
|
||
};
|
||
|
||
// getTkLoginStatus
|
||
const getTkLoginStatus = async () => {
|
||
if (!inElectron) return "false";
|
||
return await window.electronAPI.tk.checkTkLoginStatus();
|
||
};
|
||
|
||
// ========== 粉丝助手的接口 ==========
|
||
|
||
const controlTask = async (data) => {
|
||
if (!inElectron) return;
|
||
// data is JSON string in original, ensure consistent
|
||
await window.electronAPI.tk.controlTask(data);
|
||
};
|
||
|
||
const controlCheckTask = async (isRunning, model) => {
|
||
if (!inElectron) return { success: false, error: 'Not in Electron' };
|
||
return await window.electronAPI.tk.controlCheckTask({
|
||
isRunning: Boolean(isRunning),
|
||
model: Boolean(model)
|
||
});
|
||
};
|
||
|
||
const getBrotherInfo = async () => {
|
||
if (!inElectron) return { total: 0, valid: 0 };
|
||
const res = await window.electronAPI.tk.getBrotherInfo();
|
||
return JSON.parse(res);
|
||
};
|
||
|
||
const Specifystreaming = async (data) => {
|
||
if (!inElectron) return;
|
||
await window.electronAPI.tk.findBigBrother(data);
|
||
};
|
||
|
||
const storageSetInfos = async (data) => {
|
||
// data is { key, data } object
|
||
if (!inElectron) return;
|
||
await window.electronAPI.tk.storageSet(JSON.stringify(data));
|
||
};
|
||
|
||
const readSetInfos = async (key) => {
|
||
if (!inElectron) return "";
|
||
const res = await window.electronAPI.tk.storageRead(JSON.stringify(key));
|
||
return JSON.parse(res || '""'); // Handle potential empty string response
|
||
};
|
||
|
||
// Maps to visitAnchor
|
||
const openAnchorIdRooms = async (id) => {
|
||
if (!inElectron) return;
|
||
await window.electronAPI.tk.openRoom(id);
|
||
};
|
||
|
||
// Clipboard helper - 优先使用 Python RPC，fallback 到浏览器 API
|
||
const setClipboards = async (text) => {
|
||
if (inElectron) {
|
||
try {
|
||
const result = await window.electronAPI.tk.setClipboard(text);
|
||
if (result.success) return true;
|
||
} catch (e) {
|
||
console.warn('Electron clipboard failed, fallback to browser:', e);
|
||
}
|
||
}
|
||
// Fallback to browser API
|
||
try {
|
||
await navigator.clipboard.writeText(text);
|
||
return true;
|
||
} catch (e) {
|
||
console.error('Clipboard failed', e);
|
||
return false;
|
||
}
|
||
}
|
||
|
||
// ========== 新增接口 ==========
|
||
|
||
// 启动大哥监控（TikTok 登录）
|
||
const startBrotherMonitor = async () => {
|
||
if (!inElectron) return { success: false, error: 'Not in Electron' };
|
||
try {
|
||
return await window.electronAPI.tk.startBrotherMonitor();
|
||
} catch (e) {
|
||
console.error('startBrotherMonitor error:', e);
|
||
return { success: false, error: String(e) };
|
||
}
|
||
};
|
||
|
||
// 获取大哥模块 TikTok 登录状态
|
||
const getBrotherLoginStatus = async () => {
|
||
if (!inElectron) return { isLoggedIn: false };
|
||
try {
|
||
const res = await window.electronAPI.tk.getBrotherLoginStatus();
|
||
return JSON.parse(res);
|
||
} catch (e) {
|
||
console.error('getBrotherLoginStatus error:', e);
|
||
return { isLoggedIn: false };
|
||
}
|
||
};
|
||
|
||
// 打开大哥个人主页
|
||
const visitGifter = async (data) => {
|
||
if (!inElectron) return { success: false };
|
||
try {
|
||
// data 可以是 { id: 'xxx' } 或 { uniqueId: 'xxx' }
|
||
return await window.electronAPI.tk.visitGifter(JSON.stringify(data));
|
||
} catch (e) {
|
||
console.error('visitGifter error:', e);
|
||
return { success: false, error: String(e) };
|
||
}
|
||
};
|
||
|
||
// 关闭所有浏览器
|
||
const closeAllBrowsers = async () => {
|
||
if (!inElectron) return { success: false, error: 'Not in Electron' };
|
||
try {
|
||
return await window.electronAPI.tk.closeAllBrowsers();
|
||
} catch (e) {
|
||
console.error('closeAllBrowsers error:', e);
|
||
return { success: false, error: String(e) };
|
||
}
|
||
};
|
||
|
||
// 加密存储账号信息
|
||
const storageAccountInfo = async (data) => {
|
||
if (!inElectron) return { success: false, error: 'Not in Electron' };
|
||
try {
|
||
// data: { key: string, data: object }
|
||
return await window.electronAPI.tk.storageAccount(JSON.stringify(data));
|
||
} catch (e) {
|
||
console.error('storageAccountInfo error:', e);
|
||
return { success: false, error: String(e) };
|
||
}
|
||
};
|
||
|
||
// 解密读取账号信息
|
||
const readAccountInfo = async (key) => {
|
||
if (!inElectron) return { status: 'error', message: 'Not in Electron', data: null };
|
||
try {
|
||
const res = await window.electronAPI.tk.readAccount(JSON.stringify({ key }));
|
||
return JSON.parse(res);
|
||
} catch (e) {
|
||
console.error('readAccountInfo error:', e);
|
||
return { status: 'error', message: String(e), data: null };
|
||
}
|
||
};
|
||
|
||
return {
|
||
fetchDataConfig,
|
||
fetchDataCount,
|
||
loginBackStage,
|
||
loginTikTok,
|
||
loginBigTikTok,
|
||
givePyAnchorId,
|
||
backStageloginStatus,
|
||
backStageloginStatusCopy,
|
||
exportToExcel,
|
||
stopScript,
|
||
getVersion,
|
||
getTkLoginStatus,
|
||
// New Fan Workbench exports
|
||
controlTask,
|
||
controlCheckTask,
|
||
getBrotherInfo,
|
||
Specifystreaming,
|
||
storageSetInfos,
|
||
readSetInfos,
|
||
openAnchorIdRooms,
|
||
setClipboards,
|
||
// 新增接口
|
||
startBrotherMonitor,
|
||
getBrotherLoginStatus,
|
||
visitGifter,
|
||
closeAllBrowsers,
|
||
storageAccountInfo,
|
||
readAccountInfo
|
||
};
|
||
}
|