train-pwsh 0.1.5 → 0.1.6

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: f2abb2512b0e08cfc8c62921943decbd47cc7ba52320755483be12c7c74392e2
-  data.tar.gz: 6211655d9ae40fa13982f34f6e456f451b3e47a81d31f8cb7d791085b0dc41aa
+  metadata.gz: 9c0b97bc0ab701a473da4c28af195434cfc6fc374103614bb00ea02766c7d28f
+  data.tar.gz: 86f25bad8916379f13ae5769dc0a0ee89d30b1798c679496903a3b8a874f015f
 SHA512:
-  metadata.gz: 253d3d3fc36100ba90563ec290ffd81bf50670065abd029657277ab49feb4d279ddadf21625366fd8c5de27a871f7484a995c580ae43438df44ef4938f75fb50
-  data.tar.gz: 97952ce6382936fe1bd6c9457a979537b57ec91f418b9048c526a9e10d607ef4a5713b7ffbbafa75a5a5bdede35abc82476e4c008996e292666e6dcf8216f97a
+  metadata.gz: 1da78073262a500f458e9208d39850a778aa2b1993f8db53e2e8286bdb5850834e249124cee78b98a4be09b47cca338ad645f3b46fea2b45f8794b95e87672f1
+  data.tar.gz: e16e5aebad94903f7d7c7663eb5f41d9939029149f875fd1aa8af1fc1e2936cdd762440c47a8fe0632e362c2e69cf86f0f5f5e267d0504a2aa79ab336c12cfd0

data/lib/train-pwsh/connection.rb

@@ -55,6 +55,7 @@ module TrainPlugins
         #@pwsh_session_teams_pnp = @options.delete(:teams_pnp_session)
         @pwsh_session_graph_exchange = ::Pwsh::Manager.instance("#{@pwsh_path}", ['-NoLogo'])
         @pwsh_session_teams_pnp = ::Pwsh::Manager.instance("#{@pwsh_path}", [])
+        @pwsh_session_azure = ::Pwsh::Manager.instance("#{@pwsh_path}", ['-NoProfile'])
         @client_id = @options.delete(:client_id)
         @tenant_id = @options.delete(:tenant_id)
         @client_secret = @options.delete(:client_secret)
@@ -65,10 +66,13 @@ module TrainPlugins
         
         exit_status_graph_exchange = install_connect_graph_exchange()
         exit_status_teams_pnp = install_connect_teams_pnp()
+        exit_status_azure = install_azure()
         if exit_status_graph_exchange != 0
           return exit_status_graph_exchange
         elsif exit_status_teams_pnp != 0
           return exit_status_teams_pnp
+        elsif exit_status_azure != 0
+          return exit_status_azure
         end
         
       end
@@ -82,6 +86,8 @@ module TrainPlugins
           return run_script_in_graph_exchange(script)
         elsif session_type_hash.key?(:teams_pnp_session)
           return run_script_in_teams_pnp(script)
+        elsif session_type_hash.key?(:azure_session)
+          return run_script_in_azure(script)
         else
           return CommandResult.new("","",0)
         end
@@ -146,6 +152,24 @@ module TrainPlugins
         return pwsh_teams_pnp_install_connect_result[:exitcode]
       end
 
+      def install_azure()
+        pwsh_azure_install_connect = %{
+          #Collect designated inputs required for Graph, Exchange, and PnP connections
+          $client_id = '#{@client_id}'
+          $tenantid = '#{@tenant_id}'
+          $certificate_password = '#{@certificate_password}'
+          $certificate_path = '#{@certificate_path}'
+          $sharepoint_admin_url = '#{@sharepoint_admin_url}'
+
+          #Connect to Teams module
+          If ($null -eq (Get-Module -ListAvailable -Name "Az")) {Install-Module Az -Force -AllowClobber}
+          If ($null -eq (Get-Module -Name "Az")) {Import-Module Az}
+          Connect-AzAccount
+        }
+        pwsh_azure_install_connect_result = @pwsh_session_azure.execute(pwsh_azure_install_connect)
+        return pwsh_azure_install_connect_result[:exitcode]
+      end
+
       def run_script_in_graph_exchange(script)
         result = @pwsh_session_graph_exchange.execute(script)
         if result[:stdout].nil?
@@ -171,6 +195,19 @@ module TrainPlugins
         end
         return CommandResult.new(result[:stdout],result[:stderr],result[:exitcode])
       end
+
+      def run_script_in_azure(script)
+        result = @pwsh_session_azure.execute(script)
+        if result[:stdout].nil?
+          result[:stdout] = ""
+        end
+        if !result[:stdout].empty? && result[:stdout].match?(/is not recognized|session is not established/i)
+          result[:stderr] = result[:stdout]
+          result[:stdout] = ""
+          result[:exitcode] = -1
+        end
+        return CommandResult.new(result[:stdout],result[:stderr],result[:exitcode])
+      end
     end
   end
 end

data/lib/train-pwsh/version.rb

@@ -2,6 +2,6 @@
 
 module Train
   module Pwsh
-    VERSION = "0.1.5"
+    VERSION = "0.1.6"
   end
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: train-pwsh
 version: !ruby/object:Gem::Version
-  version: 0.1.5
+  version: 0.1.6
 platform: ruby
 authors:
 - Sujay Kandwal
 autorequire:
 bindir: exe
 cert_chain: []
-date: 2024-12-26 00:00:00.000000000 Z
+date: 2025-02-03 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: train