duanshuwen/zn-ai
1
0
Fork 0
Code Issues Pull Requests 1 Actions Packages Projects Releases Wiki Activity

feat: 开关房功能开发

Browse Source
This commit is contained in:
duanshuwen
2026-03-09 07:39:14 +08:00
parent 03b85cffb0
commit 15a7d115cb
13 changed files with 119 additions and 123 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
forge.config.ts
View File

@@ -57,6 +57,13 @@ const config: ForgeConfig = {
}),
],
hooks: {
async packageAfterCopy(forgeConfig, buildPath, electronVersion, platform, arch) {
const src = path.join(__dirname, 'src/main/scripts');
const dest = path.join(buildPath, 'resources', 'scripts');
await fs.ensureDir(dest);
await fs.copy(src, dest);
},
async prePackage() {
const outDir = path.resolve(process.cwd(), 'out');
fs.rmSync(outDir, { recursive: true, force: true });

7
package-lock.json generated
View File

@@ -21,7 +21,6 @@
"crypto": "^1.0.1",
"crypto-js": "^4.2.0",
"dexie": "^4.2.1",
"dotenv": "^17.2.3",
"dotenv-cli": "^11.0.0",
"electron-log": "^5.4.3",
"electron-squirrel-startup": "^1.0.1",
@@ -4957,9 +4956,9 @@
}
},
"node_modules/dotenv": {
"version": "17.2.3",
"resolved": "https://registry.npmjs.org/dotenv/-/dotenv-17.2.3.tgz",
"integrity": "sha512-JVUnt+DUIzu87TABbhPmNfVdBDt18BLOWjMUFJMSi/Qqg7NTYtabbvSNJGOJ7afbRuv9D/lngizHtP7QyLQ+9w==",
"version": "17.3.1",
"resolved": "https://registry.npmjs.org/dotenv/-/dotenv-17.3.1.tgz",
"integrity": "sha512-IO8C/dzEb6O3F9/twg6ZLXz164a2fhTnEWb95H23Dm4OuN+92NmEAlTrupP9VW6Jm3sO26tQlqyvyi4CsnY9GA==",
"license": "BSD-2-Clause",
"engines": {
"node": ">=12"

1
package.json
View File

@@ -58,7 +58,6 @@
"crypto": "^1.0.1",
"crypto-js": "^4.2.0",
"dexie": "^4.2.1",
"dotenv": "^17.2.3",
"dotenv-cli": "^11.0.0",
"electron-log": "^5.4.3",
"electron-squirrel-startup": "^1.0.1",

15
src/main/process/runTaskOperationService.ts
View File

@@ -1,9 +1,10 @@
import { ipcMain } from 'electron';
import { ipcMain, app } from 'electron';
import { IPC_EVENTS } from '@common/constants';
import { launchLocalChrome } from '@main/utils/chrome/launchLocalChrome'
import { executeScriptService } from '@main/service/execute-script-service';
import path from 'path'
import log from 'electron-log';
export function runTaskOperationService() {
const executeScriptServiceInstance = new executeScriptService();
@@ -19,7 +20,17 @@ export function runTaskOperationService() {
*/
await launchLocalChrome(options)
const result = await executeScriptServiceInstance.executeScript(path.join(__dirname, '../../scripts/fg_trace.js'), options)
// 脚本路径处理
let scriptPath = ''
if (app.isPackaged) {
scriptPath = path.join(process.resourcesPath, 'scripts/fg_trace.js')
} else {
scriptPath = path.join(process.cwd(), 'src/main/scripts/fg_trace.js')
}
log.info(`Launching script with path: ${scriptPath}`);
const result = await executeScriptServiceInstance.executeScript(scriptPath, options)
return { success: true, result };
} catch (error: any) {

2
src/main/scripts/common/checkLoginStatus.js Normal file
View File

@@ -0,0 +1,2 @@
import log from 'electron-log';

37
src/main/scripts/fg_trace.js
View File

@@ -1,9 +1,36 @@
import { chromium } from 'playwright';
import { checkLoginStatus } from '@utils/checkLoginStatus';
import dotenv from 'dotenv';
import log from 'electron-log';
dotenv.config();
const checkLoginStatus = async (page) => {
try {
const currentUrl = await page.url();
log.info('current url==========>:', currentUrl);
const loginPagePatterns = ['/login', '/signin', '/auth'];
const isLoginPage = loginPagePatterns.some(pattern => currentUrl.includes(pattern));
if(!isLoginPage) {
return true;
}
// 检查页面内容中的关键字
const pageContent = await page.content();
const loginKeywords = ['退出', '房价房量管理'];
const hasLoginKeyword = loginKeywords.some(keyword =>
pageContent.includes(keyword)
);
if (hasLoginKeyword) {
return true;
}
return false;
} catch (error) {
return false;
}
}
(async () => {
const browser = await chromium.connectOverCDP('http://localhost:9222');
@@ -24,13 +51,13 @@ dotenv.config();
if(!isLogin) {
await page.getByRole('textbox', { name: '请输入账号' }).dblclick();
await page.getByRole('textbox', { name: '请输入账号' }).click();
await page.getByRole('textbox', { name: '请输入账号' }).fill(process.env.FZ_USERNAME, { delay: 80 + Math.random() * 120 });
await page.getByRole('textbox', { name: '请输入账号' }).fill('hoteltmwq123', { delay: 80 + Math.random() * 120 });
await page.waitForTimeout(1000 + Math.random() * 1000);
await page.getByRole('button', { name: '下一步' }).click();
const frame_1 = await page.frameLocator('#alibaba-login-box');
await page.locator('#alibaba-login-box').contentFrame().getByRole('textbox', { name: '请输入登录密码' }).dblclick();
await page.locator('#alibaba-login-box').contentFrame().getByRole('textbox', { name: '请输入登录密码' }).fill(process.env.FZ_PASSWORD, { delay: 80 + Math.random() * 120 });
await page.locator('#alibaba-login-box').contentFrame().getByRole('textbox', { name: '请输入登录密码' }).fill('Tmwq654321', { delay: 80 + Math.random() * 120 });
await page.waitForTimeout(1000 + Math.random() * 1000);
await page.locator('#alibaba-login-box').contentFrame().getByRole('button', { name: '登录' }).click();

31
src/main/utils/checkLoginStatus.ts
View File

@@ -1,31 +0,0 @@
import log from 'electron-log';
export const checkLoginStatus = async (page: any) => {
try {
const currentUrl = await page.url();
log.info('current url==========>:', currentUrl);
const loginPagePatterns = ['/login', '/signin', '/auth'];
const isLoginPage = loginPagePatterns.some(pattern => currentUrl.includes(pattern));
if(!isLoginPage) {
return true;
}
// 检查页面内容中的关键字
const pageContent = await page.content();
const loginKeywords = ['退出', '房价房量管理'];
const hasLoginKeyword = loginKeywords.some(keyword =>
pageContent.includes(keyword)
);
if (hasLoginKeyword) {
return true;
}
return false;
} catch (error) {
return false;
}
}

2
src/main/utils/chrome/launchLocalChrome.ts
View File

@@ -11,7 +11,7 @@ export async function launchLocalChrome (options: any) {
// 多账号隔离
// const profileDir = getProfileDir(accountId);
log.info(`Launching Chrome with user data dir: ${options}`);
// log.info(`Launching Chrome with user data dir: ${options}`);
// 检查端口是否被占用
const portInUse = await isPortInUse(9222);

44
src/renderer/api/typeMapping.ts
View File

@@ -4,15 +4,15 @@ import request from '@utils/request';
import * as API from './types';
/** 新增房型映射 新增房型映射新增房型映射 POST /hotelStaff/typeMapping/add */
export function hotelStaffTypeMappingAddUsingPost({
/** 查询房型映射列表 查询房型映射列表查询房型映射列表 POST /hotelStaff/typeMapping/list */
export function hotelStaffTypeMappingListUsingPost({
body,
options,
}: {
body: API.RoomTypeMapping;
options?: { [key: string]: unknown };
}) {
return request<API.RBoolean>('/hotelStaff/typeMapping/add', {
return request<API.RListRoomTypeMapping>('/hotelStaff/typeMapping/list', {
method: 'POST',
headers: {
'Content-Type': 'application/json',
@@ -22,25 +22,7 @@ export function hotelStaffTypeMappingAddUsingPost({
});
}
/** 获取房型映射详细信息 获取房型映射详细信息获取房型映射详细信息 POST /hotelStaff/typeMapping/getInfo */
export function hotelStaffTypeMappingGetInfoUsingPost({
body,
options,
}: {
body: API.RoomTypeMapping;
options?: { [key: string]: unknown };
}) {
return request<API.RRoomTypeMapping>('/hotelStaff/typeMapping/getInfo', {
method: 'POST',
headers: {
'Content-Type': 'application/json',
},
data: body,
...(options || {}),
});
}
/** 查询房型映射列表 查询房型映射列表查询房型映射列表 POST /hotelStaff/typeMapping/pageList */
/** 分页查询房型映射列表 查询房型映射列表查询房型映射列表 POST /hotelStaff/typeMapping/pageList */
export function hotelStaffTypeMappingPageListUsingPost({
body,
options,
@@ -58,24 +40,6 @@ export function hotelStaffTypeMappingPageListUsingPost({
});
}
/** 删除房型映射 删除房型映射删除房型映射 POST /hotelStaff/typeMapping/remove */
export function hotelStaffTypeMappingRemoveUsingPost({
body,
options,
}: {
body: API.RoomTypeMapping;
options?: { [key: string]: unknown };
}) {
return request<API.RBoolean>('/hotelStaff/typeMapping/remove', {
method: 'POST',
headers: {
'Content-Type': 'application/json',
},
data: body,
...(options || {}),
});
}
/** 修改房型映射 修改房型映射修改房型映射 POST /hotelStaff/typeMapping/update */
export function hotelStaffTypeMappingUpdateUsingPost({
body,

24
src/renderer/api/types.ts
View File

@@ -40,22 +40,14 @@ export type HotelStaffPcUserUpdatePasswordUsingPostResponses = {
200: RBoolean;
};
export type HotelStaffTypeMappingAddUsingPostResponses = {
200: RBoolean;
};
export type HotelStaffTypeMappingGetInfoUsingPostResponses = {
200: RRoomTypeMapping;
export type HotelStaffTypeMappingListUsingPostResponses = {
200: RListRoomTypeMapping;
};
export type HotelStaffTypeMappingPageListUsingPostResponses = {
200: RPageRoomTypeMapping;
};
export type HotelStaffTypeMappingRemoveUsingPostResponses = {
200: RBoolean;
};
export type HotelStaffTypeMappingUpdateUsingPostResponses = {
200: RBoolean;
};
@@ -148,6 +140,12 @@ export type RListPcConfigChannel = {
data?: PcConfigChannel[];
};
export type RListRoomTypeMapping = {
code?: number;
msg?: string;
data?: RoomTypeMapping[];
};
export type RoomTypeMapping = {
/** 创建者创建人 */
createBy?: string;
@@ -187,12 +185,6 @@ export type RPageRoomTypeMapping = {
data?: PageRoomTypeMapping;
};
export type RRoomTypeMapping = {
code?: number;
msg?: string;
data?: RoomTypeMapping;
};
export type UpdatePasswordForm = {
/** 旧密码 */
oldPassword?: string;

65
src/renderer/views/home/components/TaskOperationDialog.vue
View File

@@ -3,12 +3,11 @@
<el-form :model="form" :rules="rules" ref="formRef" label-position="top" class="pl-4 pr-4 pt-4">
<el-form-item label="选择房型" prop="roomType">
<el-select v-model="form.roomType" placeholder="请选择房型">
<el-option label="单人间" value="single"></el-option>
<el-option label="双人间" value="double"></el-option>
<el-option v-for="item in roomList" :label="item.pmsName" :value="item.id" />
</el-select>
</el-form-item>
<el-form-item label="选择日期" prop="date">
<el-date-picker v-model="ranger" type="daterange" value-format="YYYY-MM-DD" placeholder="请选择日期"
<el-form-item label="选择日期" prop="range">
<el-date-picker v-model="form.range" type="daterange" value-format="YYYY-MM-DD" placeholder="请选择日期"
style="width: 100%">
</el-date-picker>
</el-form-item>
@@ -24,30 +23,37 @@
<script setup lang="ts">
import { taskCenterItem } from '@constant/taskCenterList'
import { hotelStaffTypeMappingListUsingPost } from '@api/index'
const isVisible = ref(false)
const title = ref('')
const formRef = ref()
const form = ref({
roomType: '',
startTime: '',
endTime: '',
operation: '',
range: [],
})
const rules = ref({
roomType: [
{ required: true, message: '请选择房型', trigger: 'blur' },
],
ranger: [
{ required: true, message: '请选择日期范围', trigger: 'blur' },
range: [
{
required: true,
message: '请选择日期范围',
trigger: 'change',
type: 'array'
},
],
})
const ranger = ref([])
// 打开弹窗
const open = ({ type }: taskCenterItem) => {
title.value = type === 'open' ? '开启渠道房型' : '关闭渠道房型'
isVisible.value = true
form.value.operation = type
getRoomTypeList()
}
// 关闭弹窗
@@ -58,9 +64,9 @@ const close = () => {
// 重置form
const reset = () => {
form.value.roomType = ''
form.value.startTime = ''
form.value.endTime = ''
ranger.value = []
form.value.range = []
formRef.value.resetFields()
}
// 取消操作
@@ -71,14 +77,33 @@ const cancel = () => {
// 确认操作
const confirm = () => {
close()
form.value.startTime = ranger.value[0]
form.value.endTime = ranger.value[1]
console.log(form.value)
/**
* 坑：传给进程的参数不能是ref包裹的reactive对象
*/
window.api.executeScript({ ...form.value })
formRef.value.validate((valid: boolean) => {
if (!valid) {
return
}
close()
console.log(form.value)
const options = {
roomType: form.value.roomType,
startTime: form.value.range[0],
endTime: form.value.range[1],
operation: form.value.operation,
}
/**
* 坑：传给进程的参数不能是ref包裹的reactive对象
*/
window.api.executeScript(options)
reset()
})
}
// 获取房型列表
const roomList: any = ref([])
const getRoomTypeList = async () => {
const res = await hotelStaffTypeMappingListUsingPost({ body: {} })
roomList.value = res.data
}
defineExpose({

4
tsconfig.app.json
View File

@@ -46,6 +46,8 @@
"src/renderer/**/*.vue",
"**/*.d.ts",
"**/*.ts",
"src/renderer/permission.ts"
"**/*.js",
"src/renderer/permission.ts",
"src/main/scripts/*.js"
]
}

3
tsconfig.json
View File

@@ -10,8 +10,7 @@
"@assets/*": ["src/assets/*"],
"@common/*": ["src/common/*"],
"@service/*": ["src/main/service/*"],
"@locales/*": ["locales/*"],
"@utils/*": ["src/main/utils/*"]
"@locales/*": ["locales/*"]
}
}
}