train-pwsh 0.1.5 → 1.0.7

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: f2abb2512b0e08cfc8c62921943decbd47cc7ba52320755483be12c7c74392e2
-  data.tar.gz: 6211655d9ae40fa13982f34f6e456f451b3e47a81d31f8cb7d791085b0dc41aa
+  metadata.gz: da66dd3bfae044e714689303864eed1b206c158282c49b0023a2847edcc9466d
+  data.tar.gz: de4f87e4206d4d54bd1fd0ca908ffb5bdaed81573c92046d71d9c64ea832b85e
 SHA512:
-  metadata.gz: 253d3d3fc36100ba90563ec290ffd81bf50670065abd029657277ab49feb4d279ddadf21625366fd8c5de27a871f7484a995c580ae43438df44ef4938f75fb50
-  data.tar.gz: 97952ce6382936fe1bd6c9457a979537b57ec91f418b9048c526a9e10d607ef4a5713b7ffbbafa75a5a5bdede35abc82476e4c008996e292666e6dcf8216f97a
+  metadata.gz: 6c78670c60457b39db97f0f83011c353841be259b8c2b0fdc1ab4967f878ab2aecfb5ba4707f101e180615fde9407bebfa137cdecd46be4523d72e64b135890f
+  data.tar.gz: d9cb83c1ef7f82c51d3f26b583b2496e8f57c9ede8c28006efedd4848cfc3a491a9868c5044c2ca1b652034ac4fe618ec14f3c887d73258bd8b4f56215a49faf

data/LICENSE.md

@@ -1,4 +1,4 @@
-Licensed under the apache-2.0 license, except as noted below.  
+Licensed under the Apache-2.0 license, except as noted below.  
 
 Redistribution and use in source and binary forms, with or without modification, are permitted provided that the following conditions are met:
 

data/README.md

@@ -12,6 +12,31 @@ A train-pwsh connection has eight fields that are needed for authentication, whi
 
 These fields need to be defined in the config file stored at this directory: `~/.inspec/config.json`. Particularly, under the `credentials` key of the json file, create a `pwsh` key with the value being another dictionary. This dictionary should have a key named `pwsh-options` with the value being another dictionary. This dictionary should contain the names of the eight fields above as well as their values. Please refer to this [link](https://origin.inspec.io/docs/reference/config/) for more detailed instructions.
 
+As mentioned in the link above documentation, to run train-pwsh, a `config.json` file is needed at the following directory: `~/.inspec/config.json`. An example json file is shown below; the values will need to be populated with your actual data. 
+
+```json
+{
+    "version": "1.1",
+    "cli_options": {
+        "color": "true"
+    },
+    "credentials": {
+        "pwsh": {
+            "pwsh-options": {
+                "client_id": "INSERT CLIENT ID",
+                "tenant_id": "INSERT TENANT ID",
+                "client_secret": "INSERT CLIENT SECRET",
+                "certificate_path": "INSERT CERTIFICATE PATH",
+                "certificate_password": "INSERT CERTIFICATE PASSWORD",
+                "organization": "INSERT ORGANIZATION",
+                "sharepoint_admin_url": "INSERT SHAREPOINT ADMIN URL",
+                "pwsh_path": "INSERT POWERSHELL PATH"
+            }
+        }
+    }
+}
+```
+
 On top of this, environment variables may need to be defined for some of these eight fields if they are to be used elsewhere in the profile as inputs. The README for the profile will specify which ones need to be stored as environment variables. 
 
 To set an environment variable on Mac, go to the `zschrc` file located at `~/.zschrc` and enter in the following syntax: `export VARIABLE_NAME='insert_value'`

data/Rakefile

@@ -1,4 +1,13 @@
 # frozen_string_literal: true
 
 require "bundler/gem_tasks"
+require 'github_changelog_generator/task'
+
 task default: %i[]
+
+GitHubChangelogGenerator::RakeTask.new :changelog do |config|
+  config.user = 'skandwal'
+  config.project = 'train-pwsh'
+  config.since_tag = '1.0.7'
+  config.future_release = '2.0.0'
+end

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 = "1.0.7"
   end
 end

data/train-pwsh.gemspec

@@ -8,17 +8,15 @@ Gem::Specification.new do |spec|
   spec.authors = ["Sujay Kandwal"]
   spec.email = ["skandwal@mitre.org"]
   spec.files = Dir['lib/**/*.rb'] + Dir['bin/*']+ ["README.md", "LICENSE.md", "NOTICE.md", "CHANGELOG.md", "CODE_OF_CONDUCT.md"]
-  spec.summary = "Enabling continuous Powershell connection over Inspec"
-  #spec.description = "TODO: Write a longer description or delete this line."
-  #spec.homepage = "TODO: Put your gem's website or public repo URL here."
-  spec.license = "MIT"
+  spec.summary = "Enabling continuous Powershell connection over Inspec."
+  spec.description = "Enabling continuous Powershell connection over Inspec."
+  spec.homepage = "https://rubygems.org/gems/train-pwsh"
+  spec.license = "Apache-2.0"
   spec.required_ruby_version = ">= 2.6.0"
 
-  #spec.metadata["allowed_push_host"] = "TODO: Set to your gem server 'https://example.com'"
-
-  #spec.metadata["homepage_uri"] = spec.homepage
-  #spec.metadata["source_code_uri"] = "TODO: Put your gem's public repo URL here."
-  #spec.metadata["changelog_uri"] = "TODO: Put your gem's CHANGELOG.md URL here."
+  spec.metadata["homepage_uri"] = spec.homepage
+  spec.metadata["source_code_uri"] = "https://github.com/mitre/train-pwsh/"
+  spec.metadata["changelog_uri"] = spec.metadata['source_code_uri'] + "blob/main/CHANGELOG.md"
 
   # Specify which files should be added to the gem when it is released.
   # The `git ls-files -z` loads the files in the RubyGem that have been added into git.

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: train-pwsh
 version: !ruby/object:Gem::Version
-  version: 0.1.5
+  version: 1.0.7
 platform: ruby
 authors:
 - Sujay Kandwal
 autorequire:
 bindir: exe
 cert_chain: []
-date: 2024-12-26 00:00:00.000000000 Z
+date: 2025-02-04 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: train
@@ -38,7 +38,7 @@ dependencies:
     - - ">="
       - !ruby/object:Gem::Version
         version: '0'
-description:
+description: Enabling continuous Powershell connection over Inspec.
 email:
 - skandwal@mitre.org
 executables: []
@@ -59,10 +59,13 @@ files:
 - lib/train-pwsh/version.rb
 - sig/train/pwsh.rbs
 - train-pwsh.gemspec
-homepage:
+homepage: https://rubygems.org/gems/train-pwsh
 licenses:
-- MIT
-metadata: {}
+- Apache-2.0
+metadata:
+  homepage_uri: https://rubygems.org/gems/train-pwsh
+  source_code_uri: https://github.com/mitre/train-pwsh/
+  changelog_uri: https://github.com/mitre/train-pwsh/blob/main/CHANGELOG.md
 post_install_message:
 rdoc_options: []
 require_paths:
@@ -81,5 +84,5 @@ requirements: []
 rubygems_version: 3.3.27
 signing_key:
 specification_version: 4
-summary: Enabling continuous Powershell connection over Inspec
+summary: Enabling continuous Powershell connection over Inspec.
 test_files: []