From 86f35341ffcf028a7fecda2db08d77a080a14e1d Mon Sep 17 00:00:00 2001
From: aykut-bozkurt <51649454+aykut-bozkurt@users.noreply.github.com>
Date: Fri, 27 Sep 2024 15:39:06 +0300
Subject: [PATCH] Add PG function declarations that are not generated by pgrx
 yet (#18)

---
 src/lib.rs                       |  1 +
 src/pgrx_missing_declerations.rs | 54 ++++++++++++++++++++++++++++++++
 2 files changed, 55 insertions(+)
 create mode 100644 src/pgrx_missing_declerations.rs

diff --git a/src/lib.rs b/src/lib.rs
index 797dc88..3118e7f 100644
--- a/src/lib.rs
+++ b/src/lib.rs
@@ -1,5 +1,6 @@
 use pgrx::prelude::*;
 
+mod pgrx_missing_declerations;
 mod pgrx_utils;
 
 pgrx::pg_module_magic!();
diff --git a/src/pgrx_missing_declerations.rs b/src/pgrx_missing_declerations.rs
new file mode 100644
index 0000000..8e6eeb6
--- /dev/null
+++ b/src/pgrx_missing_declerations.rs
@@ -0,0 +1,54 @@
+use pgrx::{
+    pg_sys::{self, Alias, Oid, ParseNamespaceItem, ParseState, Relation, TupleTableSlot},
+    PgBox,
+    prelude::*,
+};
+
+#[allow(improper_ctypes)]
+#[pg_guard]
+extern "C" {
+    pub(crate) fn get_extension_oid(name: *const i8, missing_ok: bool) -> Oid;
+
+    pub(crate) fn get_extension_schema(ext_oid: Oid) -> Oid;
+
+    pub(crate) fn addRangeTableEntryForRelation(
+        pstate: *mut ParseState,
+        rel: Relation,
+        lockmode: i32,
+        alias: *mut Alias,
+        inh: bool,
+        inFromCl: bool,
+    ) -> *mut ParseNamespaceItem;
+
+    pub(crate) fn addNSItemToQuery(
+        pstate: *mut ParseState,
+        nsitem: *mut ParseNamespaceItem,
+        add_to_join_list: bool,
+        add_to_rel_namespace: bool,
+        add_to_var_namespace: bool,
+    );
+
+    pub(crate) fn assign_expr_collations(
+        pstate: *mut ParseState,
+        expr: *mut pg_sys::Node,
+    ) -> *mut pg_sys::Node;
+}
+
+/*
+ * slot_getallattrs
+ *		This function forces all the entries of the slot's Datum/isnull
+ *		arrays to be valid.  The caller may then extract data directly
+ *		from those arrays instead of using slot_getattr.
+ */
+pub(crate) fn slot_getallattrs(slot: *mut TupleTableSlot) {
+    // copied from Postgres since this method was inlined in the original code
+    // (not found in pg_sys)
+    // handles select * from table
+    unsafe {
+        let slot = PgBox::from_pg(slot);
+        let tts_tupledesc = PgBox::from_pg(slot.tts_tupleDescriptor);
+        if (slot.tts_nvalid as i32) < tts_tupledesc.natts {
+            pg_sys::slot_getsomeattrs_int(slot.as_ptr(), tts_tupledesc.natts);
+        }
+    };
+}