refactor: Cleanup

Author: furesoft

src/Moss.NET.Sdk/Core/InternalFunctions.cs

@@ -3,8 +3,16 @@
 
 namespace Moss.NET.Sdk.Core;
 
-public class InternalFunctions
+public static class InternalFunctions
 {
     [DllImport(Functions.DLL, EntryPoint = "moss_em_export_statistical_data")]
     public static extern void ExportStatisticalData();
+
+    [DllImport(Functions.DLL, EntryPoint = "moss_api_document_export")]
+    private static extern void ExportDocument(ulong documentUuidPtr);
+
+    public static void ExportDocument(string documentUuid)
+    {
+        ExportDocument(documentUuid.GetPointer());
+    }
 }

src/Moss.NET.Sdk/Core/RunOnce.cs

@@ -0,0 +1,29 @@
+using System.Runtime.CompilerServices;
+
+namespace Moss.NET.Sdk.Core;
+
+using System;
+using System.Collections.Generic;
+
+public static class RunOnce
+{
+    private static readonly Dictionary<string, bool> actionsRun = new Dictionary<string, bool>();
+    private static readonly object _lockObj = new();
+
+    public static void Execute(string actionKey, Action action)
+    {
+        lock (_lockObj)
+        {
+            if (actionsRun.ContainsKey(actionKey) && actionsRun[actionKey])
+                return;
+
+            action();
+            actionsRun[actionKey] = true;
+        }
+    }
+
+    public static unsafe void Execute(Action action)
+    {
+        Execute(new IntPtr(Unsafe.AsPointer(ref action)).ToString(), action);
+    }
+}

src/Moss.NET.Sdk/Storage.cs

@@ -1,6 +1,7 @@
 using System.Runtime.InteropServices;
 using Extism;
 using Moss.NET.Sdk.API;
+using Moss.NET.Sdk.Core;
 using Moss.NET.Sdk.FFI;
 using Moss.NET.Sdk.FFI.Dto;
 
@@ -10,7 +11,7 @@ namespace Moss.NET.Sdk;
 public static class Storage
 {
     [DllImport(Functions.DLL, EntryPoint = "moss_api_document_metadata_get_all")]
-    private static extern ulong GetDocumentMetadata(ulong uuidPtr);
+    private static extern ulong GetApiDocumentMetadata(ulong uuidPtr);
 
     [DllImport(Functions.DLL, EntryPoint = "moss_api_document_duplicate")]
     private static extern ulong DuplicateDocument(ulong uuidPtr); // -> string
@@ -30,15 +31,21 @@ public static class Storage
     [DllImport(Functions.DLL, EntryPoint = "moss_api_document_unload_files")]
     private static extern void UnloadFiles(ulong uuidPtr);
 
+    [DllImport(Functions.DLL, EntryPoint = "moss_api_document_ensure_download")]
+    private static extern void EnsureDownload(ulong uuidPtr);
+
+    [DllImport(Functions.DLL, EntryPoint = "moss_api_document_load_files_from_cache")]
+    private static extern void LoadFilesFromCache(ulong uuidPtr);
+
     /// <summary>
     /// Retrieves the metadata of a document given its UUID.
     /// </summary>
     /// <param name="uuid">The UUID of the document.</param>
     /// <returns>The metadata of the document.</returns>
-    public static Metadata GetDocumentMetadata(string uuid)
+    public static Metadata GetApiDocumentMetadata(string uuid)
     {
         var uuidPtr = Pdk.Allocate(uuid).Offset;
-        var resultPtr = GetDocumentMetadata(uuidPtr);
+        var resultPtr = GetApiDocumentMetadata(uuidPtr);
 
         return Utils.Deserialize(resultPtr, JsonContext.Default.Metadata);
     }
@@ -187,4 +194,22 @@ public static void UnloadFiles(string uuid)
         var ptr = Pdk.Allocate(uuid).Offset;
         UnloadFiles(ptr);
     }
+
+    /// <summary>
+    /// Ensures that the document is downloaded.
+    /// </summary>
+    /// <param name="uuid"></param>
+    public static void EnsureDownload(string uuid)
+    {
+        EnsureDownload(uuid.GetPointer());
+    }
+
+    /// <summary>
+    /// This function loads all the files enforcing cache usage only. If the cache misses a file, it will not be downloaded.
+    /// </summary>
+    /// <param name="uuid"></param>
+    public static void LoadFilesFromCache(string uuid)
+    {
+        LoadFilesFromCache(uuid.GetPointer());
+    }
 }

src/Moss.NET.Sdk/UI/ScreenManager.cs

@@ -17,9 +17,7 @@ public static class ScreenManager
     private static extern void RegisterScreen(ulong keyPtr);
 
     [DllImport(Functions.DLL, EntryPoint = "_moss_pe_open_screen")]
-    private static extern ulong
-        OpenScreen(ulong keyPtr,
-            ulong initial_valuesPtr); // initial_valuesPtr = dict of values, could be any object that is non primitive
+    private static extern ulong OpenScreen(ulong keyPtr, ulong initial_valuesPtr); // initial_valuesPtr = dict of values, could be any object that is non primitive
 
     [DllImport(Functions.DLL, EntryPoint = "moss_pe_get_screen_value")]
     private static extern ulong GetScreenValue(ulong keyPtr); //-> ConfigGet
@@ -61,10 +59,9 @@ public static void Register<T>()
         var instance = Activator.CreateInstance<T>();
         var screen = new FFI.Dto.Screen(instance.Name, "ext_event_screen_loop", "ext_event_screen_preloop",
             "ext_event_screen_postloop");
-        var screenPtr = Utils.Serialize(screen, JsonContext.Default.Screen);
-        Screens.TryAdd(instance.Name, instance);
 
-        RegisterScreen(screenPtr);
+        Screens.TryAdd(instance.Name, instance);
+        RegisterScreen(screen.GetPointer());
     }
 
     public static void Open<T>(Dictionary<string, object> values)

src/SamplePlugin/SampleExtension.cs

@@ -24,24 +24,25 @@ public static void Main()
 
     public override ExtensionInfo Register(MossState state)
     {
-        Pdk.Log(LogLevel.Info, "registered sample extension");
-
         Config.Set("theme", "dark");
         Defaults.SetDefaultValue("OUTLINE_COLOR", Color.Blue);
         Theme.Apply(new DarkTheme());
 
         Assets.Add("Assets/swap.svg");
 
-        var md = Storage.GetDocumentMetadata("0ba3df9c-8ca0-4347-8d7c-07471101baad");
+        var md = Storage.GetApiDocumentMetadata("0ba3df9c-8ca0-4347-8d7c-07471101baad");
         Pdk.Log(LogLevel.Info, $"Metadata: {md.VisibleName} with {md.Hash}");
 
         var nid = Storage.DuplicateDocument("0ba3df9c-8ca0-4347-8d7c-07471101baad");
 
         //var testUuid = Storage.NewNotebook("test notebook");
         //Storage.NewPdf("test pdf", Array.Empty<byte>());
-      //  Storage.NewEpub("test ebook", Array.Empty<byte>());
-
+        //Storage.NewEpub("test ebook", Array.Empty<byte>());
+        InternalFunctions.ExportDocument("0ba3df9c-8ca0-4347-8d7c-07471101baad");
         Storage.UnloadFiles("0ba3df9c-8ca0-4347-8d7c-07471101baad");
+        Storage.EnsureDownload("0ba3df9c-8ca0-4347-8d7c-07471101baad");
+        Storage.LoadFilesFromCache("0ba3df9c-8ca0-4347-8d7c-07471101baad");
+
         Storage.RandomizeUUIDs("0ba3df9c-8ca0-4347-8d7c-07471101baad");
 
         Moss.NET.Sdk.Moss.RegisterExtensionButton(
@@ -74,7 +75,7 @@ public override void ExtensionLoop(MossState state)
 
         GUI.InvertIcon("swap", "swap_inverted");
 
-        InternalFunctions.ExportStatisticalData();
+        RunOnce.Execute(InternalFunctions.ExportStatisticalData);
     }
 
     public override void Unregister()