Added bindings for table functions

Author: Giorgi

DuckDB.NET.Bindings/NativeMethods/NativeMethods.TableFunction.cs

@@ -0,0 +1,60 @@
+using System.Runtime.InteropServices;
+using System;
+
+namespace DuckDB.NET.Native;
+
+public partial class NativeMethods
+{
+    public static class TableFunction
+    {
+        [DllImport(DuckDbLibrary, CallingConvention = CallingConvention.Cdecl, EntryPoint = "duckdb_create_tableFunction")]
+        public static extern IntPtr DuckDBCreateTableFunction();
+
+        [DllImport(DuckDbLibrary, CallingConvention = CallingConvention.Cdecl, EntryPoint = "duckdb_destroy_tableFunction")]
+        public static extern void DuckDBDestroyTableFunction(out IntPtr tableFunction);
+
+        [DllImport(DuckDbLibrary, CallingConvention = CallingConvention.Cdecl, EntryPoint = "duckdb_table_function_set_name")]
+        public static extern void DuckDBTableFunctionSetName(IntPtr tableFunction, SafeUnmanagedMemoryHandle name);
+
+        [DllImport(DuckDbLibrary, CallingConvention = CallingConvention.Cdecl, EntryPoint = "duckdb_table_function_add_parameter")]
+        public static extern void DuckDBTableFunctionAddParameter(IntPtr tableFunction, DuckDBLogicalType type);
+
+        [DllImport(DuckDbLibrary, CallingConvention = CallingConvention.Cdecl, EntryPoint = "duckdb_table_function_set_extra_info")]
+        public static extern unsafe void DuckDBTableFunctionSetExtraInfo(IntPtr tableFunction, IntPtr extraInfo, delegate* unmanaged[Cdecl]<IntPtr, void> destroy);
+
+        [DllImport(DuckDbLibrary, CallingConvention = CallingConvention.Cdecl, EntryPoint = "duckdb_table_function_set_bind")]
+        public static extern unsafe void DuckDBTableFunctionSetBind(IntPtr tableFunction, delegate* unmanaged[Cdecl]<IntPtr, void> bind);
+
+        //[DllImport(DuckDbLibrary, CallingConvention = CallingConvention.Cdecl, EntryPoint = "duckdb_table_function_set_init")]
+        //public static extern unsafe void DuckDBTableFunctionSetInit(IntPtr tableFunction, duckdb_tableFunction_init_t init);
+
+        [DllImport(DuckDbLibrary, CallingConvention = CallingConvention.Cdecl, EntryPoint = "duckdb_table_function_set_function")]
+        public static extern unsafe void DuckDBTableFunctionSetFunction(IntPtr tableFunction, delegate* unmanaged[Cdecl]<IntPtr, IntPtr, IntPtr, void> callback);
+
+        [DllImport(DuckDbLibrary, CallingConvention = CallingConvention.Cdecl, EntryPoint = "duckdb_register_table_function")]
+        public static extern DuckDBState DuckDBRegisterTableFunction(DuckDBNativeConnection con, IntPtr tableFunction);
+
+        #region TableFunctionBind
+
+        [DllImport(DuckDbLibrary, CallingConvention = CallingConvention.Cdecl, EntryPoint = "duckdb_bind_add_result_column")]
+        public static extern unsafe void DuckDBBindAddResultColumn(IntPtr info, SafeUnmanagedMemoryHandle name, DuckDBLogicalType type);
+
+        [DllImport(DuckDbLibrary, CallingConvention = CallingConvention.Cdecl, EntryPoint = "duckdb_bind_get_parameter_count")]
+        public static extern unsafe ulong DuckDBBindGetParameterCount(IntPtr info);
+
+        [DllImport(DuckDbLibrary, CallingConvention = CallingConvention.Cdecl, EntryPoint = "duckdb_bind_get_parameter")]
+        public static extern unsafe DuckDBValue DuckDBBindGetParameter(IntPtr info, ulong index);
+
+        [DllImport(DuckDbLibrary, CallingConvention = CallingConvention.Cdecl, EntryPoint = "duckdb_bind_set_bind_data")]
+        public static extern unsafe void DuckDBBindSetBindData(IntPtr info, IntPtr bindData, delegate* unmanaged[Cdecl]<IntPtr, void> destroy);
+
+        #endregion
+
+        #region TableFunction
+
+        [DllImport(DuckDbLibrary, CallingConvention = CallingConvention.Cdecl, EntryPoint = "duckdb_function_get_bind_data")]
+        public static extern unsafe void* duckdb_function_get_bind_data(IntPtr info);
+
+        #endregion
+    }
+}

DuckDB.NET.Bindings/NativeMethods/NativeMethods.Value.cs

@@ -70,6 +70,9 @@ public static class Value
 
         [DllImport(DuckDbLibrary, CallingConvention = CallingConvention.Cdecl, EntryPoint = "duckdb_create_blob")]
         public static extern DuckDBValue DuckDBCreateBlob([In] byte[] value, long length);
+
+        [DllImport(DuckDbLibrary, CallingConvention = CallingConvention.Cdecl, EntryPoint = "duckdb_get_int32")]
+        public static extern int DuckDBGetInt32(DuckDBValue value);
 
         [DllImport(DuckDbLibrary, CallingConvention = CallingConvention.Cdecl, EntryPoint = "duckdb_create_list_value")]
         public static extern DuckDBValue DuckDBCreateListValue(DuckDBLogicalType logicalType, IntPtr[] values, long count);

DuckDB.NET.Data/DuckDBConnection.TableFunction.cs

@@ -0,0 +1,38 @@
+using System;
+using System.Runtime.CompilerServices;
+using System.Runtime.InteropServices;
+using DuckDB.NET.Data.Extensions;
+using DuckDB.NET.Data.Internal;
+using DuckDB.NET.Native;
+
+namespace DuckDB.NET.Data;
+
+partial class DuckDBConnection
+{
+    public unsafe void RegisterTableFunction<T, TResult>(string name)
+    {
+        var function = NativeMethods.TableFunction.DuckDBCreateTableFunction();
+        NativeMethods.TableFunction.DuckDBTableFunctionSetName(function, name.ToUnmanagedString());
+
+        using (var logicalType = DuckDBTypeMap.GetLogicalType<T>())
+        {
+            NativeMethods.TableFunction.DuckDBTableFunctionAddParameter(function, logicalType);
+        }
+
+        NativeMethods.TableFunction.DuckDBTableFunctionSetBind(function, &Bind);
+    }
+
+    [UnmanagedCallersOnly(CallConvs = [typeof(CallConvCdecl)])]
+    public static unsafe void Bind(IntPtr info)
+    {
+        var parameters = new object[NativeMethods.TableFunction.DuckDBBindGetParameterCount(info)];
+
+        for (var i = 0; i < parameters.Length; i++)
+        {
+            using var value = NativeMethods.TableFunction.DuckDBBindGetParameter(info, (ulong)i);
+            parameters[i] = NativeMethods.Value.DuckDBGetInt32(value);
+        }
+
+        NativeMethods.TableFunction.DuckDBBindSetBindData(info, parameters.ToHandle(), &DestroyExtraInfo);
+    }
+}