CommvaultPowerShell

0.3.4

.history/custom/Module_20241113220249.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;
        }
    }
}