markpdfdown 0.1.8-beta.6 → 0.2.0

package/dist/server/controllers/FileController.js

@@ -1,64 +0,0 @@
-import path from 'path';
-import fs from 'fs';
-import multer from 'multer';
-import fileLogic from '../logic/File.js';
-// 文件上传处理
-const uploadFiles = (req, res, next) => {
-    // 获取taskId参数
-    const taskId = req.query.taskId;
-    if (!taskId) {
-        return res.status(400).json({ message: '缺少任务ID参数' });
-    }
-    // 动态配置存储位置
-    const storage = multer.diskStorage({
-        destination: (_req, _file, cb) => {
-            // 获取基础上传目录
-            const baseUploadDir = fileLogic.getUploadDir();
-            // 创建任务ID对应的目录
-            const uploadDir = path.join(baseUploadDir, taskId);
-            // 确保目录存在
-            if (!fs.existsSync(uploadDir)) {
-                fs.mkdirSync(uploadDir, { recursive: true });
-            }
-            cb(null, uploadDir);
-        },
-        filename: (_req, file, cb) => {
-            // 使用taskId作为文件名，保留原始扩展名
-            const extname = path.extname(file.originalname);
-            cb(null, taskId + extname);
-        }
-    });
-    // 创建multer实例
-    const upload = multer({ storage }).array('files');
-    // 使用multer中间件处理上传
-    upload(req, res, (err) => {
-        if (err) {
-            return next(err);
-        }
-        try {
-            const multerReq = req;
-            // 检查是否上传了文件
-            if (!multerReq.files || multerReq.files.length === 0) {
-                return res.status(400).json({ message: '没有上传文件' });
-            }
-            const uploadedFiles = multerReq.files.map(file => ({
-                originalName: file.originalname, // 原始文件名（包含中文）
-                savedName: path.basename(file.path), // 保存的文件名（taskId + 扩展名）
-                path: file.path,
-                size: file.size,
-                mimetype: file.mimetype,
-                taskId: taskId
-            }));
-            res.status(200).json({
-                message: '文件上传成功',
-                files: uploadedFiles
-            });
-        }
-        catch (error) {
-            next(error);
-        }
-    });
-};
-export default {
-    uploadFiles,
-};

package/dist/server/controllers/TaskController.js

@@ -1,57 +0,0 @@
-import taskDal from '../dal/TaskDal.js';
-import fileLogic from '../logic/File.js';
-// 批量创建任务
-const createTasks = async (req, res, next) => {
-    try {
-        const tasks = req.body;
-        // 批量创建任务
-        const createdTasks = await taskDal.createTasks(tasks);
-        res.status(201).json(createdTasks);
-    }
-    catch (error) {
-        next(error);
-    }
-};
-// 获取所有任务,支持分页
-const getAllTasks = async (req, res, next) => {
-    try {
-        const { page = 1, pageSize = 10 } = req.query;
-        const tasks = await taskDal.findAll(Number(page), Number(pageSize));
-        const total = await taskDal.getTotal();
-        res.status(200).json({ list: tasks, total });
-    }
-    catch (error) {
-        next(error);
-    }
-};
-// 更新任务
-const updateTask = async (req, res, next) => {
-    try {
-        const { id } = req.params;
-        const task = req.body;
-        const updatedTask = await taskDal.update(id, task);
-        res.status(200).json(updatedTask);
-    }
-    catch (error) {
-        next(error);
-    }
-};
-// 删除任务
-const deleteTask = async (req, res, next) => {
-    try {
-        const { id } = req.params;
-        const deletedTask = await taskDal.remove(id);
-        // 删除任务后，删除任务对应的文件
-        fileLogic.deleteTaskFiles(id);
-        res.status(200).json(deletedTask);
-    }
-    catch (error) {
-        next(error);
-    }
-};
-export default {
-    createTasks,
-    getAllTasks,
-    updateTask,
-    deleteTask,
-};

package/dist/server/controllers/completionController.js

@@ -1,64 +0,0 @@
-import modelLogic from '../logic/Model.js';
-// 转换图片为markdown
-const markImagedown = async (req, res, next) => {
-    // 验证请求
-    if (!req.body.providerId || !req.body.modelId || !req.body.url) {
-        return res.status(400).json({ message: 'providerId, modelId, url 为必填项' });
-    }
-    try {
-        const message = await modelLogic.transformImageMessage(req.body.url);
-        const completionOptions = {
-            messages: message,
-            model: req.body.modelId,
-            maxTokens: 204800,
-            temperature: 0.3,
-            stream: false
-        };
-        const completion = await modelLogic.completion(Number(req.body.providerId), completionOptions);
-        res.json(completion);
-    }
-    catch (error) {
-        next(error);
-    }
-};
-// 测试模型连接
-const testConnection = async (req, res, next) => {
-    // 验证请求
-    if (!req.body.providerId || !req.body.modelId) {
-        return res.status(400).json({ message: 'providerId, modelId 为必填项' });
-    }
-    try {
-        const { providerId, modelId } = req.body;
-        const imageBase64 = 'data:image/png;base64,iVBORw0KGgoAAAANSUhEUgAAACAAAAAgCAMAAABEpIrGAAAA9lBMVEUAAAD+/v79/f3////7+/v8/Pz29vbe3t7f399GRkb////+/v77+/tiYmL19fX////z8/Pv7+/y8vLl5eXj4+Pe3t7f39/j4+Pd3d3u7u65ubm9vb2wsLCJiYmBgYFcXFxAQED7+/vt7e3Dw8O2trb+/v78/Pz4+Pj29vbz8/Px8fHs7Ozj4+Pt7e3b29vm5ube3t7l5eXr6+vLy8vj4+Pv7+/09PTZ2dnS0tL////Pz8/09PTCwsLFxcW6urqKioqhoaHOzs7KysqQkJCQkJCpqamZmZmQkJC0tLRmZmaTk5N6enrt7e3v7+/s7OzT09PV1dX///8sTJOgAAAAUXRSTlMA/v3+/P3vpqUF+/V/Devy59XUzb27tqyihldMQzQsEgv6rDc0+vPl5OHLyMi5s7CpopuXlZCGgHhybmdlTURCQT4+OjUyMiwnIx0Xw8GwenjtgN1sAAABw0lEQVQ4y5WS13IbMQxFcUGukpVWvXfLknvvcXfiuKfh/38mC3I5I4/tB9+HvShnIFAkfUq9Vn27/3F7H0ZEwEn8fr/AaD+83DY5qj1fD9/2q2bJeRyLDjKF0ev+EOKDrQjYOGgKm6P5/t2yGaj/A3DmKpeR7VLQD0B0wEzAi6FYjLiYhUnENRiiLnhtIc3Hm/qlv1j3/QHbKQkTfeFHzX8z0NJApzrHFXkgl2YnjNk5jE2I8pLtv0oB0G32tNjJs9xkQEd6AaiU7M7upDx2/wvAHqjLUwBsJd07XrENrY8EBQc05D4AJXIqYXKsXsYftQvT9hvrDk65w1Mreqt69mB9saeHAfiaHbGNC80XTVnteILwEylgFLiXhr9oIyO3jV2JiYoVG4AnqTtAGKhqMC7b3R2bqwSgJx0P5G+E2cV7wHfKBWCVf2YAUWLB5zPwSZoH4ArhMpy3YKJfGjyyB6aWBx5Yxze1hc2xftcidFMANY6S8DTYPw13ZkCmbqpuE9S19tIFZ4CfNsDyHc3pCJE0DzaAaMsXoPvPa1SAvnWJY5fmUaU3Gl4/15a4efvy0GYu0PtK2A3CPn2o/na91aPP6D8AojIypyhEagAAAABJRU5ErkJggg==';
-        const message = [{
-                role: 'user',
-                content: [{
-                        type: 'text',
-                        text: 'What is in the picture?'
-                    },
-                    {
-                        type: 'image_url',
-                        image_url: {
-                            url: imageBase64
-                        }
-                    }
-                ]
-            }];
-        const completionOptions = {
-            messages: message,
-            model: modelId,
-            maxTokens: 1024,
-            temperature: 0.7,
-            stream: false
-        };
-        const test = await modelLogic.completion(Number(providerId), completionOptions);
-        res.json(test);
-    }
-    catch (error) {
-        next(error);
-    }
-};
-export default {
-    markImagedown,
-    testConnection
-};

package/dist/server/controllers/modelController.js

@@ -1,74 +0,0 @@
-import modelDal from '../dal/ModelDal.js';
-import providerDal from '../dal/ProviderDal.js';
-// 获取所有启用的模型
-const getAllModels = async (_req, res, next) => {
-    try {
-        const providers = await providerDal.findAll();
-        // 遍历providers，获取每个provider的模型
-        const models = await Promise.all(providers.map(async (provider) => {
-            const models = await modelDal.findByProviderId(provider.id);
-            return {
-                provider: provider.id,
-                providerName: provider.name,
-                models: models
-            };
-        }));
-        res.json(models);
-    }
-    catch (error) {
-        next(error);
-    }
-};
-// 获取指定服务商的模型列表
-const getModelsByProviderId = async (req, res, next) => {
-    try {
-        const { provider } = req.params;
-        const models = await modelDal.findByProviderId(Number(provider));
-        res.json(models);
-    }
-    catch (error) {
-        next(error);
-    }
-};
-// 创建模型
-const createModel = async (req, res, next) => {
-    try {
-        const { id, provider, name } = req.body;
-        // 验证请求
-        if (!provider || !name || !id) {
-            return res.status(400).json({ message: '模型ID、服务商ID、名称和为必填项' });
-        }
-        // 创建模型
-        const newModel = await modelDal.create({
-            id,
-            provider: Number(provider),
-            name,
-        });
-        res.status(201).json(newModel);
-    }
-    catch (error) {
-        next(error);
-    }
-};
-// 删除模型
-const deleteModel = async (req, res, next) => {
-    try {
-        const { id, provider } = req.params;
-        // 验证请求
-        if (!id || !provider) {
-            return res.status(400).json({ message: '模型ID和服务商ID为必填项' });
-        }
-        // 删除模型
-        await modelDal.remove(id, Number(provider));
-        res.status(204).json({ message: '模型删除成功' });
-    }
-    catch (error) {
-        next(error);
-    }
-};
-export default {
-    getAllModels,
-    getModelsByProviderId,
-    createModel,
-    deleteModel
-};

package/dist/server/controllers/providerController.js

@@ -1,120 +0,0 @@
-import providerDal from '../dal/ProviderDal.js';
-// 获取所有服务商
-const getAllProviders = async (_req, res, next) => {
-    try {
-        const providers = await providerDal.findAll();
-        res.json(providers);
-    }
-    catch (error) {
-        next(error);
-    }
-};
-// 根据ID获取服务商
-const getProviderById = async (req, res, next) => {
-    try {
-        const { id } = req.params;
-        const provider = await providerDal.findById(Number(id));
-        if (!provider) {
-            return res.status(404).json({ message: '服务商不存在' });
-        }
-        res.json(provider);
-    }
-    catch (error) {
-        next(error);
-    }
-};
-// 创建服务商
-const createProvider = async (req, res, next) => {
-    try {
-        const { name, type } = req.body;
-        // 验证请求
-        if (!name || !type) {
-            return res.status(400).json({ message: '名称和协议类型为必填项' });
-        }
-        // 创建服务商
-        const newProvider = await providerDal.create({
-            name,
-            type,
-            api_key: '',
-            base_url: '',
-            suffix: '',
-            status: 0 // 默认启用
-        });
-        res.status(201).json(newProvider);
-    }
-    catch (error) {
-        next(error);
-    }
-};
-// 更新服务商
-const updateProvider = async (req, res, next) => {
-    try {
-        const { id } = req.params;
-        const { api_key, base_url, suffix } = req.body;
-        // 验证服务商存在
-        const existingProvider = await providerDal.findById(Number(id));
-        if (!existingProvider) {
-            return res.status(404).json({ message: '服务商不存在' });
-        }
-        // 准备更新数据
-        const updateData = {};
-        if (api_key !== undefined)
-            updateData.api_key = api_key;
-        if (base_url !== undefined)
-            updateData.base_url = base_url;
-        if (suffix !== undefined)
-            updateData.suffix = suffix;
-        // 更新服务商
-        const updatedProvider = await providerDal.update(Number(id), updateData);
-        res.json(updatedProvider);
-    }
-    catch (error) {
-        next(error);
-    }
-};
-// 删除服务商
-const deleteProvider = async (req, res, next) => {
-    try {
-        const { id } = req.params;
-        // 验证服务商存在
-        const existingProvider = await providerDal.findById(Number(id));
-        if (!existingProvider) {
-            return res.status(404).json({ message: '服务商不存在' });
-        }
-        // 删除服务商
-        await providerDal.remove(Number(id));
-        res.status(204).send();
-    }
-    catch (error) {
-        next(error);
-    }
-};
-// 更新服务商状态
-const updateProviderStatus = async (req, res, next) => {
-    try {
-        const { id } = req.params;
-        const { status } = req.body;
-        if (status === undefined) {
-            return res.status(400).json({ message: '状态值不合法' });
-        }
-        // 验证服务商存在
-        const existingProvider = await providerDal.findById(Number(id));
-        if (!existingProvider) {
-            return res.status(404).json({ message: '服务商不存在' });
-        }
-        // 更新服务商状态
-        const updatedProvider = await providerDal.updateStatus(Number(id), Number(status));
-        res.json(updatedProvider);
-    }
-    catch (error) {
-        next(error);
-    }
-};
-export default {
-    getAllProviders,
-    getProviderById,
-    createProvider,
-    updateProvider,
-    deleteProvider,
-    updateProviderStatus
-};

package/dist/server/dal/TaskDal.js

@@ -1,67 +0,0 @@
-import { prisma } from "../db/index.js";
-import { v4 as uuidv4 } from 'uuid';
-// 查找所有任务，支持分页
-const findAll = async (page, pageSize) => {
-    return await prisma.task.findMany({
-        skip: (page - 1) * pageSize,
-        take: pageSize,
-        orderBy: {
-            createdAt: 'desc',
-        },
-    });
-};
-// 查询任务总数
-const getTotal = async () => {
-    return await prisma.task.count();
-};
-// 创建任务
-const create = async (task) => {
-    return await prisma.task.create({
-        data: {
-            id: uuidv4(),
-            filename: task?.filename || '',
-            type: task?.type || '',
-            page_range: task?.page_range || '',
-            pages: task?.pages || 0,
-            provider: task?.provider || 0,
-            model: task?.model || '',
-            model_name: task?.model_name || '',
-            progress: 0,
-            status: 0,
-        }
-    });
-};
-// 批量创建任务
-const createTasks = async (tasks) => {
-    const createdTasks = [];
-    for (const task of tasks) {
-        const createdTask = await create(task);
-        createdTasks.push(createdTask);
-    }
-    return createdTasks;
-};
-// 更新任务
-const update = async (id, task) => {
-    return await prisma.task.update({
-        where: { id },
-        data: task,
-    });
-};
-// 删除任务
-const remove = async (id) => {
-    await prisma.task.delete({
-        where: { id },
-    });
-    // 删除任务对应detail
-    return await prisma.taskDetail.deleteMany({
-        where: { task: id },
-    });
-};
-export default {
-    findAll,
-    create,
-    createTasks,
-    getTotal,
-    update,
-    remove,
-};

package/dist/server/dal/modelDal.js

@@ -1,44 +0,0 @@
-import { prisma } from "../db/index.js";
-// 查找所有模型
-const findAll = async () => {
-    return await prisma.model.findMany({
-        orderBy: [{ createdAt: "desc" }],
-    });
-};
-// 根据服务商ID查找模型
-const findByProviderId = async (provider) => {
-    return await prisma.model.findMany({
-        where: { provider },
-        orderBy: [{ createdAt: "desc" }],
-    });
-};
-// 创建模型
-const create = async (modelData) => {
-    return await prisma.model.create({
-        data: modelData,
-    });
-};
-// 删除模型 根据id和provider删除
-const remove = async (id, provider) => {
-    return await prisma.model.delete({
-        where: {
-            id_provider: {
-                id: id,
-                provider: provider
-            }
-        }
-    });
-};
-// 批量删除指定服务商的模型
-const removeByProviderId = async (provider) => {
-    return await prisma.model.deleteMany({
-        where: { provider },
-    });
-};
-export default {
-    findAll,
-    findByProviderId,
-    create,
-    remove,
-    removeByProviderId,
-};

package/dist/server/dal/providerDal.js

@@ -1,83 +0,0 @@
-import { prisma } from "../db/index.js";
-// 查找所有启用的提供商
-const findAll = async () => {
-    return await prisma.provider.findMany({
-        where: {
-            status: 0
-        },
-        select: {
-            id: true,
-            name: true,
-            type: true,
-            api_key: true,
-            base_url: true,
-            suffix: true,
-            status: true,
-            createdAt: true,
-            updatedAt: true,
-        },
-        orderBy: [{ createdAt: "desc" }],
-    });
-};
-// 根据ID查找提供商
-const findById = async (id) => {
-    return await prisma.provider.findUnique({
-        where: { id },
-        select: {
-            id: true,
-            name: true,
-            type: true,
-            api_key: true,
-            base_url: true,
-            suffix: true,
-            status: true,
-            createdAt: true,
-            updatedAt: true,
-        },
-    });
-};
-// 创建提供商
-const create = async (data) => {
-    return await prisma.provider.create({
-        data: {
-            name: data?.name || '',
-            type: data?.type || '',
-            api_key: data?.api_key || '',
-            base_url: data?.base_url || '',
-            suffix: data?.suffix || '',
-            status: data?.status || 0,
-        }
-    });
-};
-// 更新提供商
-const update = async (id, data) => {
-    return await prisma.provider.update({
-        where: { id },
-        data: data,
-    });
-};
-// 删除提供商
-const remove = async (id) => {
-    // 同时删除关联的model
-    await prisma.model.deleteMany({
-        where: { provider: id },
-    });
-    return await prisma.provider.delete({
-        where: { id },
-    });
-};
-// 更新提供商状态
-const updateStatus = async (id, status) => {
-    return await prisma.provider.update({
-        where: { id },
-        data: { status },
-    });
-};
-export default {
-    findAll,
-    findById,
-    create,
-    update,
-    remove,
-    updateStatus,
-};

package/dist/server/db/index.js

@@ -1,57 +0,0 @@
-import pkg from '@prisma/client';
-const { PrismaClient } = pkg;
-import { runMigrations } from './Migration.js';
-import path from 'path';
-import { app } from 'electron';
-import fs from 'fs';
-import isDev from 'electron-is-dev';
-// 设置和获取数据库URL
-function getDatabaseUrl() {
-    // 否则，为打包应用生成一个默认路径
-    if (!isDev) {
-        const userDataPath = app.getPath('userData');
-        const dbDir = path.join(userDataPath, 'db');
-        if (!fs.existsSync(dbDir)) {
-            fs.mkdirSync(dbDir, { recursive: true });
-        }
-        console.log('Using userData database path:', `file:${path.join(dbDir, 'app.db')}`);
-        return `file:${path.join(dbDir, 'app.db')}`;
-    }
-    // 开发环境回退路径
-    console.log('Using default development database path:', `file:${path.join(process.cwd(), 'src', 'server', 'db', 'dev.db')}`);
-    return `file:${path.join(process.cwd(), 'src', 'server', 'db', 'dev.db')}`;
-}
-// 获取数据库URL
-const dbUrl = getDatabaseUrl();
-// 创建Prisma实例，使用环境变量中的数据库URL
-const prisma = new PrismaClient({
-    datasources: {
-        db: {
-            url: dbUrl
-        }
-    }
-});
-// 初始化数据库，包括运行迁移
-const initDatabase = async () => {
-    try {
-        console.log(`Initializing database(url:${dbUrl})...`);
-        // 验证数据库连接
-        await prisma.$queryRaw `SELECT 1`;
-        console.log('Database connection established successfully.');
-        // 运行迁移，传递现有的prisma实例
-        await runMigrations(prisma);
-        return true;
-    }
-    catch (error) {
-        console.error('Database initialization error:', error);
-        return false;
-    }
-};
-/**
- * 关闭数据库连接
- */
-const disconnect = async () => {
-    await prisma.$disconnect();
-    console.log('Database connection has been closed');
-};
-export { prisma, disconnect, initDatabase };