pg_pageindex--1.0.sql

-- complain if script is sourced in psql, rather than via CREATE EXTENSION
\echo Use "CREATE EXTENSION pg_pageindex" to load this file. \quit

-- Stable API (matches pkg/pageindex entrypoints)
CREATE FUNCTION pageindex.build_pdf(path text, options jsonb DEFAULT '{}'::jsonb)
RETURNS jsonb
LANGUAGE C STRICT
AS 'MODULE_PATHNAME', 'pg_pageindex_build_pdf';

CREATE FUNCTION pageindex.build_markdown(path text, options jsonb DEFAULT '{}'::jsonb)
RETURNS jsonb
LANGUAGE C STRICT
AS 'MODULE_PATHNAME', 'pg_pageindex_build_markdown';

CREATE FUNCTION pageindex.tree_search(doc jsonb, query text, options jsonb DEFAULT '{}'::jsonb)
RETURNS jsonb
LANGUAGE C STRICT
AS 'MODULE_PATHNAME', 'pg_pageindex_tree_search';

CREATE FUNCTION pageindex.tree_search_with_preference(
	doc jsonb,
	query text,
	preference text,
	options jsonb DEFAULT '{}'::jsonb
)
RETURNS jsonb
LANGUAGE C STRICT
AS 'MODULE_PATHNAME', 'pg_pageindex_tree_search_with_preference';

-- Go-style names (aliases)
CREATE FUNCTION pageindex.build_from_pdf(path text, options jsonb DEFAULT '{}'::jsonb)
RETURNS jsonb
LANGUAGE SQL
STRICT
AS $$SELECT pageindex.build_pdf($1, $2)$$;

CREATE FUNCTION pageindex.build_from_markdown(path text, options jsonb DEFAULT '{}'::jsonb)
RETURNS jsonb
LANGUAGE SQL
STRICT
AS $$SELECT pageindex.build_markdown($1, $2)$$;

-- Tree helpers (structure = JSON array of root nodes, same shape as Document.structure)
CREATE FUNCTION pageindex.structure_to_list(structure jsonb)
RETURNS jsonb
LANGUAGE C STRICT
AS 'MODULE_PATHNAME', 'pg_pageindex_structure_to_list';

CREATE FUNCTION pageindex.create_clean_structure_for_description(structure jsonb)
RETURNS jsonb
LANGUAGE C STRICT
AS 'MODULE_PATHNAME', 'pg_pageindex_create_clean_structure_for_description';

CREATE FUNCTION pageindex.remove_structure_text(structure jsonb)
RETURNS jsonb
LANGUAGE C STRICT
AS 'MODULE_PATHNAME', 'pg_pageindex_remove_structure_text';

-- items = JSON array of PostProcessingItem (see PageIndex pkg/pageindex)
CREATE FUNCTION pageindex.post_process_tree(items jsonb, end_physical_index integer)
RETURNS jsonb
LANGUAGE C STRICT
AS 'MODULE_PATHNAME', 'pg_pageindex_post_process_tree';

-- page_texts = JSON array of strings, one per 1-based page order (index 0 = page 1)
CREATE FUNCTION pageindex.add_node_text(structure jsonb, page_texts jsonb)
RETURNS jsonb
LANGUAGE C STRICT
AS 'MODULE_PATHNAME', 'pg_pageindex_add_node_text';

CREATE FUNCTION pageindex.add_node_text_with_labels(structure jsonb, page_texts jsonb)
RETURNS jsonb
LANGUAGE C STRICT
AS 'MODULE_PATHNAME', 'pg_pageindex_add_node_text_with_labels';

CREATE FUNCTION pageindex.generate_summaries_for_structure(structure jsonb, model text)
RETURNS jsonb
LANGUAGE C STRICT
AS 'MODULE_PATHNAME', 'pg_pageindex_generate_summaries_for_structure';

CREATE FUNCTION pageindex.generate_doc_description(structure jsonb, model text)
RETURNS text
LANGUAGE C STRICT
AS 'MODULE_PATHNAME', 'pg_pageindex_generate_doc_description';

CREATE FUNCTION pageindex.version()
RETURNS text
LANGUAGE C IMMUTABLE PARALLEL SAFE
AS 'MODULE_PATHNAME', 'pg_pageindex_version';