CommvaultPowerShell

0.3.4

.history/custom/Module_20241113221056.cs

using System;
using System.Collections.Generic;
using System.Net.Http;
using System.Net.Http.Headers;
using System.Security.Cryptography.X509Certificates;
using System.Text;
 
namespace Commvault.Powershell
{
    using static Commvault.Powershell.Runtime.Extensions;
 
    public partial class Module
    {
        partial void CustomInit()
        {
            // Add a step at the end of the pipeline to attach the API key
            // Add once for the regular pipeline
            this._pipeline.Append(AddAuthtoken);
 
            // Add once for the pipeline that supports a proxy
            this._pipelineWithProxy.Append(AddAuthtoken);
        }
 
        protected async System.Threading.Tasks.Task<System.Net.Http.HttpResponseMessage> AddAuthtoken(System.Net.Http.HttpRequestMessage request, Runtime.IEventListener callback, Runtime.ISendAsync next)
        {
            try
            {
                // Translate the request URI
                request.RequestUri = TranslateIncomingRequestToCommvaultAPI(request);
 
                // Check if the URL contains metallic.io
                bool isMetallic = request.RequestUri.AbsoluteUri.ToLower().Contains("metallic.io");
 
                // Get the authentication token
                var token = System.Environment.GetEnvironmentVariable("CVToken");
 
                if (string.IsNullOrEmpty(token))
                {
                    throw new Exception("CVToken environment variable is empty or not set.");
                }
 
                // Add the appropriate authentication header
                if (isMetallic)
                {
                    request.Headers.Add("Authorization", "Bearer " + token);
                }
                else
                {
                    request.Headers.Add("Authtoken", token);
                }
 
                // Add other necessary headers
                request.Headers.Add("Accept", "application/json");
 
                // Load the server certificate
                X509CertificateValidator validator = new X509CertificateValidator();
                X509Certificate2 serverCert = validator.GetServerCertificate(request.RequestUri.Host, request.RequestUri.Port);
 
                // Trust the server certificate
                if (serverCert != null)
                {
                    validator.TrustCertificate(serverCert);
                }
 
                // Create a new HttpClient with the custom certificate validation
                using (var client = new HttpClient(new HttpClientHandler
                {
                    ClientCertificateOptions = ClientCertificateOption.Manual,
                    ServerCertificateCustomValidationCallback = validator.ValidateCertificate
                }))
                {
                    // Let the request continue
                    return await client.SendAsync(request, HttpCompletionOption.ResponseContentRead);
                }
            }
            catch (Exception ex)
            {
                // Log or handle the exception as needed
                throw ex;
            }
        }
 
        private static System.Uri TranslateIncomingRequestToCommvaultAPI(System.Net.Http.HttpRequestMessage request)
        {
            string removeString = "api";
            int removePos = request.RequestUri.AbsoluteUri.IndexOf(removeString) + removeString.Length;
            string URL = System.Environment.GetEnvironmentVariable("WebServerURL") + request.RequestUri.AbsoluteUri.Remove(0, removePos);
 
            var _url = new global::System.Uri(global::System.Text.RegularExpressions.Regex.Replace(
                        URL, "\\?&*$|&*$|(\\?)&+|(&)&+", "$1$2"));
 
            return _url;
        }
    }
}