Web_BAI_Manage_ApiServer/script/pocketbase.cart-order.js

import { createRequire } from 'module';
import PocketBase from 'pocketbase';

const require = createRequire(import.meta.url);

let runtimeConfig = {};
try {
    runtimeConfig = require('../pocket-base/bai_api_pb_hooks/bai_api_shared/config/runtime.js');
} catch (_error) {
    runtimeConfig = {};
}

const PB_URL = (process.env.PB_URL || 'https://bai-api.blv-oa.com/pb').replace(/\/+$/, '');
const AUTH_TOKEN = process.env.POCKETBASE_AUTH_TOKEN || runtimeConfig.POCKETBASE_AUTH_TOKEN || '';
const OWNER_AUTH_RULE = '@request.auth.id != ""';
const CART_OWNER_MATCH_RULE = 'cart_owner = @request.auth.openid';
const ORDER_OWNER_MATCH_RULE = 'order_owner = @request.auth.openid';

if (!AUTH_TOKEN) {
    console.error('❌ 缺少 POCKETBASE_AUTH_TOKEN，无法执行建表。');
    process.exit(1);
}

const pb = new PocketBase(PB_URL);

const collections = [
    {
        name: 'tbl_cart',
        type: 'base',
        listRule: `${OWNER_AUTH_RULE} && ${CART_OWNER_MATCH_RULE}`,
        viewRule: `${OWNER_AUTH_RULE} && ${CART_OWNER_MATCH_RULE}`,
        createRule: `${OWNER_AUTH_RULE} && @request.body.cart_owner = @request.auth.openid`,
        updateRule: `${OWNER_AUTH_RULE} && ${CART_OWNER_MATCH_RULE}`,
        deleteRule: `${OWNER_AUTH_RULE} && ${CART_OWNER_MATCH_RULE}`,
        fields: [
            { name: 'cart_id', type: 'text', required: true },
            { name: 'cart_number', type: 'text', required: true },
            { name: 'cart_create', type: 'autodate', onCreate: true, onUpdate: false },
            { name: 'cart_owner', type: 'text', required: true },
            { name: 'cart_product_id', type: 'text', required: true },
            { name: 'cart_product_quantity', type: 'number', required: true },
            { name: 'cart_status', type: 'text', required: true },
            { name: 'cart_at_price', type: 'number', required: true },
            { name: 'cart_remark', type: 'text' },
        ],
        indexes: [
            'CREATE UNIQUE INDEX idx_tbl_cart_cart_id ON tbl_cart (cart_id)',
            'CREATE INDEX idx_tbl_cart_cart_number ON tbl_cart (cart_number)',
            'CREATE INDEX idx_tbl_cart_cart_owner ON tbl_cart (cart_owner)',
            'CREATE INDEX idx_tbl_cart_cart_product_id ON tbl_cart (cart_product_id)',
            'CREATE INDEX idx_tbl_cart_cart_status ON tbl_cart (cart_status)',
            'CREATE INDEX idx_tbl_cart_cart_create ON tbl_cart (cart_create)',
            'CREATE INDEX idx_tbl_cart_owner_number ON tbl_cart (cart_owner, cart_number)',
            'CREATE INDEX idx_tbl_cart_owner_status ON tbl_cart (cart_owner, cart_status)',
        ],
    },
    {
        name: 'tbl_order',
        type: 'base',
        listRule: `${OWNER_AUTH_RULE} && ${ORDER_OWNER_MATCH_RULE}`,
        viewRule: `${OWNER_AUTH_RULE} && ${ORDER_OWNER_MATCH_RULE}`,
        createRule: `${OWNER_AUTH_RULE} && @request.body.order_owner = @request.auth.openid`,
        updateRule: `${OWNER_AUTH_RULE} && ${ORDER_OWNER_MATCH_RULE}`,
        deleteRule: `${OWNER_AUTH_RULE} && ${ORDER_OWNER_MATCH_RULE}`,
        fields: [
            { name: 'order_id', type: 'text', required: true },
            { name: 'order_number', type: 'text', required: true },
            { name: 'order_create', type: 'autodate', onCreate: true, onUpdate: false },
            { name: 'order_owner', type: 'text', required: true },
            { name: 'order_source', type: 'text', required: true },
            { name: 'order_status', type: 'text', required: true },
            { name: 'order_source_id', type: 'text', required: true },
            { name: 'order_snap', type: 'json', required: true },
            { name: 'order_amount', type: 'number', required: true },
            { name: 'order_remark', type: 'text' },
        ],
        indexes: [
            'CREATE UNIQUE INDEX idx_tbl_order_order_id ON tbl_order (order_id)',
            'CREATE UNIQUE INDEX idx_tbl_order_order_number ON tbl_order (order_number)',
            'CREATE INDEX idx_tbl_order_order_owner ON tbl_order (order_owner)',
            'CREATE INDEX idx_tbl_order_order_source ON tbl_order (order_source)',
            'CREATE INDEX idx_tbl_order_order_status ON tbl_order (order_status)',
            'CREATE INDEX idx_tbl_order_order_source_id ON tbl_order (order_source_id)',
            'CREATE INDEX idx_tbl_order_order_create ON tbl_order (order_create)',
            'CREATE INDEX idx_tbl_order_owner_status ON tbl_order (order_owner, order_status)',
        ],
    },
];

function normalizeFieldPayload(field, existingField) {
    const payload = existingField
        ? Object.assign({}, existingField)
        : {
            name: field.name,
            type: field.type,
        };

    if (existingField && existingField.id) {
        payload.id = existingField.id;
    }

    payload.name = field.name;
    payload.type = field.type;

    if (typeof field.required !== 'undefined') {
        payload.required = field.required;
    }

    if (field.type === 'autodate') {
        payload.onCreate = typeof field.onCreate === 'boolean' ? field.onCreate : true;
        payload.onUpdate = typeof field.onUpdate === 'boolean' ? field.onUpdate : false;
    }

    return payload;
}

function buildCollectionPayload(collectionData, existingCollection) {
    if (!existingCollection) {
        return {
            name: collectionData.name,
            type: collectionData.type,
            listRule: Object.prototype.hasOwnProperty.call(collectionData, 'listRule') ? collectionData.listRule : null,
            viewRule: Object.prototype.hasOwnProperty.call(collectionData, 'viewRule') ? collectionData.viewRule : null,
            createRule: Object.prototype.hasOwnProperty.call(collectionData, 'createRule') ? collectionData.createRule : null,
            updateRule: Object.prototype.hasOwnProperty.call(collectionData, 'updateRule') ? collectionData.updateRule : null,
            deleteRule: Object.prototype.hasOwnProperty.call(collectionData, 'deleteRule') ? collectionData.deleteRule : null,
            fields: collectionData.fields.map((field) => normalizeFieldPayload(field, null)),
            indexes: collectionData.indexes,
        };
    }

    const targetFieldMap = new Map(collectionData.fields.map((field) => [field.name, field]));
    const fields = (existingCollection.fields || []).map((existingField) => {
        const targetField = targetFieldMap.get(existingField.name);
        if (!targetField) {
            return existingField;
        }

        targetFieldMap.delete(existingField.name);
        return normalizeFieldPayload(targetField, existingField);
    });

    for (const field of targetFieldMap.values()) {
        fields.push(normalizeFieldPayload(field, null));
    }

    return {
        name: collectionData.name,
        type: collectionData.type,
        listRule: Object.prototype.hasOwnProperty.call(collectionData, 'listRule') ? collectionData.listRule : existingCollection.listRule,
        viewRule: Object.prototype.hasOwnProperty.call(collectionData, 'viewRule') ? collectionData.viewRule : existingCollection.viewRule,
        createRule: Object.prototype.hasOwnProperty.call(collectionData, 'createRule') ? collectionData.createRule : existingCollection.createRule,
        updateRule: Object.prototype.hasOwnProperty.call(collectionData, 'updateRule') ? collectionData.updateRule : existingCollection.updateRule,
        deleteRule: Object.prototype.hasOwnProperty.call(collectionData, 'deleteRule') ? collectionData.deleteRule : existingCollection.deleteRule,
        fields: fields,
        indexes: collectionData.indexes,
    };
}

function normalizeFieldList(fields) {
    return (fields || []).map((field) => ({
        name: field.name,
        type: field.type,
        required: !!field.required,
    }));
}

async function createOrUpdateCollection(collectionData) {
    console.log(`🔄 正在处理表: ${collectionData.name} ...`);

    try {
        const list = await pb.collections.getFullList({
            sort: '-created',
        });
        const existing = list.find((item) => item.name === collectionData.name);

        if (existing) {
            await pb.collections.update(existing.id, buildCollectionPayload(collectionData, existing));
            console.log(`♻️ ${collectionData.name} 已存在，已按最新结构更新。`);
            return;
        }

        await pb.collections.create(buildCollectionPayload(collectionData, null));
        console.log(`✅ ${collectionData.name} 创建完成。`);
    } catch (error) {
        console.error(`❌ 处理集合 ${collectionData.name} 失败:`, {
            status: error.status,
            message: error.message,
            response: error.response,
        });
        throw error;
    }
}

async function getCollectionByName(collectionName) {
    const list = await pb.collections.getFullList({
        sort: '-created',
    });
    return list.find((item) => item.name === collectionName) || null;
}

async function verifyCollections(targetCollections) {
    console.log('\n🔍 开始校验购物车与订单表结构及索引...');

    for (const target of targetCollections) {
        const remote = await getCollectionByName(target.name);
        if (!remote) {
            throw new Error(`${target.name} 不存在`);
        }

        const remoteFields = normalizeFieldList(remote.fields);
        const targetFields = normalizeFieldList(target.fields);
        const remoteFieldMap = new Map(remoteFields.map((field) => [field.name, field.type]));
        const remoteRequiredMap = new Map(remoteFields.map((field) => [field.name, field.required]));
        const missingFields = [];
        const mismatchedTypes = [];
        const mismatchedRequired = [];

        for (const field of targetFields) {
            if (!remoteFieldMap.has(field.name)) {
                missingFields.push(field.name);
                continue;
            }

            if (remoteFieldMap.get(field.name) !== field.type) {
                mismatchedTypes.push(`${field.name}:${remoteFieldMap.get(field.name)}!=${field.type}`);
            }

            if (remoteRequiredMap.get(field.name) !== !!field.required) {
                mismatchedRequired.push(`${field.name}:${remoteRequiredMap.get(field.name)}!=${!!field.required}`);
            }
        }

        const remoteIndexes = new Set(remote.indexes || []);
        const missingIndexes = target.indexes.filter((indexSql) => !remoteIndexes.has(indexSql));

        if (remote.type !== target.type) {
            throw new Error(`${target.name} 类型不匹配，期望 ${target.type}，实际 ${remote.type}`);
        }

        if (!missingFields.length && !mismatchedTypes.length && !mismatchedRequired.length && !missingIndexes.length) {
            console.log(`✅ ${target.name} 校验通过。`);
            continue;
        }

        console.log(`❌ ${target.name} 校验失败：`);
        if (missingFields.length) {
            console.log(`  - 缺失字段: ${missingFields.join(', ')}`);
        }
        if (mismatchedTypes.length) {
            console.log(`  - 字段类型不匹配: ${mismatchedTypes.join(', ')}`);
        }
        if (mismatchedRequired.length) {
            console.log(`  - 字段必填属性不匹配: ${mismatchedRequired.join(', ')}`);
        }
        if (missingIndexes.length) {
            console.log(`  - 缺失索引: ${missingIndexes.join(' | ')}`);
        }

        throw new Error(`${target.name} 结构与预期不一致`);
    }
}

async function init() {
    try {
        console.log(`🔄 正在连接 PocketBase: ${PB_URL}`);
        pb.authStore.save(AUTH_TOKEN, null);
        console.log('✅ 已使用 POCKETBASE_AUTH_TOKEN 载入认证状态。');

        for (const collectionData of collections) {
            await createOrUpdateCollection(collectionData);
        }

        await verifyCollections(collections);
        console.log('\n🎉 购物车与订单表结构初始化并校验完成！');
    } catch (error) {
        console.error('❌ 初始化失败:', error.response?.data || error.message);
        process.exitCode = 1;
    }
}

init();