src/search.ts

/*
 * Copyright (c) 2025 by frostime. All Rights Reserved.
 * @Author       : frostime
 * @Date         : 2025-01-01 16:53:40
 * @FilePath     : /src/search.ts
 * @LastEditTime : 2025-05-13 18:57:12
 * @Description  : 
 */
import { listDocsByPath, request, sql } from "./api";
import { getLute } from "./lute";
import { Block, BlockId } from "./types";


export async function getBlockByID(blockId: string): Promise<Block> {
    let sqlScript = `select * from blocks where id ='${blockId}'`;
    let data = await sql(sqlScript);
    return data?.[0];
}


export const getChildBlocks = async (id: BlockId): Promise<{ id: BlockId; markdown: string; type: string }[]> => {
    return request('/api/block/getChildBlocks', { id });
}

/**
 * 获取文档内容
 * @param docId 文档ID
 * @returns 文档Markdown内容
 */
export const getDocumentMarkdown = async (docId: string): Promise<string> => {
    try {
        // 获取文档内容
        const result = await getChildBlocks(docId);
        return result.map(b => b.markdown).join('\n\n');
    } catch (error) {
        console.error("获取文档内容失败", error);
        throw error;
    }
};

/**
 * Get the markdown content of a block
 * Special:
 * - If the block is a heading, it will return the markdown content of the heading and its children
 * - If the block is a document, it will return the markdown content of the document
 * @param id 
 * @returns 
 */
export const getMarkdown = async (id: BlockId): Promise<string> => {
    let block: Block = await getBlockByID(id);
    if (!block) return null;
    if (block.type === 'd' || block.type === 'h') {
        const childBlocks = await request('/api/block/getChildBlocks', { id });
        const bodyContent = childBlocks.map(b => b.markdown).join('\n\n');
        const markdown = block.type === 'd' ? bodyContent : `${block.markdown}\n\n${bodyContent}`;
        return markdown;
    } else {
        return block.markdown;
    }
}


export const searchIDs = async (ids: BlockId | BlockId[]): Promise<Block[]> => {
    let idList = Array.isArray(ids) ? ids.map((id) => `"${id}"`) : [`"${ids}"`];
    let sqlCode = `select * from blocks where id in (${idList.join(",")})`;
    let data = await sql(sqlCode);
    return data;
}

export const id2block = searchIDs;

export const searchChildDocs = async (documentId: BlockId): Promise<Block[]> => {
    let doc: Block = await getBlockByID(documentId);
    if (!doc) {
        return [];
    }
    let box = doc.box;
    let path = doc.path;

    let data = await listDocsByPath(box, path);
    let ids = data?.files.map((item) => item.id) || [];
    let childs = await id2block(ids);
    return childs;
}

export const searchBacklinks = async (id: BlockId, limit: number = 64): Promise<Block[]> => {
    return sql(`
    select * from blocks where id in (
        select block_id from refs where def_block_id = '${id}'
    ) order by updated desc ${limit ? `limit ${limit}` : ''};
    `);
}

export const searchAttr = async (name: string, val?: string, valMatch: '=' | 'like' = '=', limit?: number): Promise<Block[]> => {
    return sql(`
    SELECT B.*
    FROM blocks AS B
    WHERE B.id IN (
        SELECT A.block_id
        FROM attributes AS A
        WHERE A.name like '${name}'
        ${val ? `AND A.value ${valMatch} '${val}'` : ''}
        ${limit ? `limit ${limit}` : ''}
    );
    `);
}