omnata-plugin-devkit 0.10.15a107__tar.gz → 0.10.16a109__tar.gz

{omnata_plugin_devkit-0.10.15a107 → omnata_plugin_devkit-0.10.16a109}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: omnata-plugin-devkit
-Version: 0.10.15a107
+Version: 0.10.16a109
 Summary: 
 Author: James Weakley
 Author-email: james.weakley@omnata.com

{omnata_plugin_devkit-0.10.15a107 → omnata_plugin_devkit-0.10.16a109}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "omnata-plugin-devkit"
-version = "0.10.15-a107"
+version = "0.10.16-a109"
 description = ""
 authors = ["James Weakley <james.weakley@omnata.com>"]
 readme = "README.md"

{omnata_plugin_devkit-0.10.15a107 → omnata_plugin_devkit-0.10.16a109}/src/omnata_plugin_devkit/jinja_templates/TUNNEL_TEST.sql.jinja

@@ -48,6 +48,7 @@ def run(session,tunnel_type:str,other_secrets_name:str):
       url = f'https://{ngrok_endpoint_host}/'
       try:
          response=http.request('GET',url)
+         return_data['mtls_success'] = True
       except urllib3.exceptions.MaxRetryError as max_retry_error:
          if isinstance(max_retry_error.reason, urllib3.exceptions.SSLError):
             ssl_error:urllib3.exceptions.SSLError = max_retry_error.reason

{omnata_plugin_devkit-0.10.15a107 → omnata_plugin_devkit-0.10.16a109}/src/omnata_plugin_devkit/plugin_uploader.py

@@ -20,6 +20,7 @@ import sys
 import typing
 import zipfile
 import copy
+import uuid
 from pathlib import Path
 from typing import Any, Dict, List, Literal, Optional
 
@@ -619,6 +620,28 @@ $${}$$;
 FROM '/streamlit'
 MAIN_FILE = '/plugin_configuration.py';"""
             )
+            
+            setup_script.write("""
+WITH CREATE_TOKEN_FUNCTION AS PROCEDURE()
+RETURNS BOOLEAN
+LANGUAGE JAVASCRIPT
+AS
+$$
+    var uuidStatement = snowflake.createStatement({ sqlText: 
+        `select UUID_STRING();`
+    });
+    var uuidResult = uuidStatement.execute();
+    uuidResult.next();
+    var uuid = uuidResult.getColumnValue(1);
+    var twoDollars='$' + '$';
+    snowflake.createStatement({ sqlText: 
+        `create or replace function PLUGIN.TOKEN() returns string as ${twoDollars}'${uuid}'${twoDollars} `
+    }).execute();
+    return true;
+$$
+CALL CREATE_TOKEN_FUNCTION();
+GRANT USAGE ON FUNCTION PLUGIN.TOKEN() TO application role OMNATA_MANAGEMENT;
+""")
 
             setup_script.write(
                 """GRANT USAGE ON STREAMLIT UI."Plugin Configuration" TO application role OMNATA_MANAGEMENT;\n"""
@@ -653,6 +676,15 @@ MAIN_FILE = '/plugin_configuration.py';"""
                 }
             )
             manifest_file.write(content)
+            custom_manifest_privileges = os.path.join(plugin_directory, "custom_manifest_privileges.sql")
+            if os.path.exists(custom_manifest_privileges):
+                print("Adding manifest privileges from custom_manifest_privileges.sql")
+                with open(
+                    custom_manifest_privileges, "r", encoding="utf-8"
+                ) as custom_manifest_privileges_stream:
+                    setup_script.write(custom_manifest_privileges_stream.read() + "\n")
+
+        
         print("Uploading app manifest to stage")
         self.session.sql(
             (