CommvaultPowerShell

0.3.4

.history/custom/Module_20241113220304.cs

using System;
using System.Collections.Generic;
using System.Net.Http;
using System.Net.Http.Headers;
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");
 
            // Set up the HttpClientHandler to skip certificate verification
            var handler = new HttpClientHandler
            {
                ServerCertificateCustomValidationCallback = (message, cert, chain, errors) => true
            };
 
            // Create a new HttpClient with the custom handler
            using (var client = new HttpClient(handler))
            {
                // 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;
        }
    }
}