add migrations folder

Author: ramsessanchez

src/Migrations/Migrations.md

@@ -0,0 +1,20 @@
+# Migrations
+
+This directory contains common [AutoREST.PowerShell](https://github.com/Azure/autorest.powershell) configurations for Migrations v1.0 and/or beta modules.
+
+## AutoRest Configuration
+
+> see <https://aka.ms/autorest>
+
+``` yaml
+require:
+  - $(this-folder)/../readme.graph.md
+```
+
+### Directives
+
+> see https://github.com/Azure/autorest/blob/master/docs/powershell/directives.md
+
+``` yaml
+# Directives go here!
+```

src/Migrations/beta/.gitattributes

@@ -0,0 +1 @@
+* text=auto

src/Migrations/beta/.gitignore

@@ -0,0 +1,16 @@
+bin
+obj
+.vs
+generated
+internal
+exports
+tools
+custom/*.psm1
+custom/autogen-model-cmdlets
+test/*-TestResults.xml
+/*.ps1
+/*.ps1xml
+/*.psm1
+/*.snk
+/*.csproj
+/*.nuspec

src/Migrations/beta/custom/EventExtensions.cs

@@ -0,0 +1,38 @@
+// ------------------------------------------------------------------------------
+//  Copyright (c) Microsoft Corporation.  All Rights Reserved.  Licensed under the MIT License.  See License in the project root for license information.
+// ------------------------------------------------------------------------------
+
+namespace Microsoft.Graph.Beta.PowerShell
+{
+    using System;
+    using System.Text.RegularExpressions;
+    using System.Threading;
+    using System.Threading.Tasks;
+    using Microsoft.Graph.Beta.PowerShell.Runtime;
+
+    public static class EventExtensions
+    {
+        /// <summary>
+        /// Print event details to the provided stream
+        /// </summary>
+        /// <param name="getEventData">The event data to print</param>
+        /// <param name="signal">The delegate for signaling events to the runtime</param>
+        /// <param name="token">The cancellation token for the request</param>
+        /// <param name="streamName">The name of the stream to print data to</param>
+        /// <param name="eventName">The name of the event to be printed</param>
+        public static async void Print(this Func<EventArgs> getEventData, Func<string, CancellationToken, Func<EventArgs>, Task> signal, CancellationToken token, string streamName, string eventName)
+        {
+            var eventDisplayName = EventFactory.SplitPascalCase(eventName).ToUpperInvariant();
+            var data = EventDataConverter.ConvertFrom(getEventData()); // also, we manually use our TypeConverter to return an appropriate type
+            if (data.Id != Events.Verbose && data.Id != Events.Warning && data.Id != Events.Debug && data.Id != Events.Information && data.Id != Events.Error)
+            {
+                await signal(streamName, token, () => EventFactory.CreateLogEvent($"{eventDisplayName} The contents are '{data?.Id}' and '{data?.Message}'"));
+                if (data != null)
+                {
+                    await signal(streamName, token, () => EventFactory.CreateLogEvent($"{eventDisplayName} Parameter: '{data.Parameter}'\n{eventDisplayName} RequestMessage '{data.RequestMessage}'\n{eventDisplayName} Response: '{data.ResponseMessage}'\n{eventDisplayName} Value: '{data.Value}'"));
+                    await signal(streamName, token, () => EventFactory.CreateLogEvent($"{eventDisplayName} ExtendedData Type: '{data.ExtendedData?.GetType()}'\n{eventDisplayName} ExtendedData '{data.ExtendedData}'"));
+                }
+            }
+        }
+    }
+}

src/Migrations/beta/custom/EventFactory.cs

@@ -0,0 +1,72 @@
+// ------------------------------------------------------------------------------
+//  Copyright (c) Microsoft Corporation.  All Rights Reserved.  Licensed under the MIT License.  See License in the project root for license information.
+// ------------------------------------------------------------------------------
+
+using System;
+using System.Text.RegularExpressions;
+using System.Threading.Tasks;
+using Microsoft.Graph.Beta.PowerShell.Runtime;
+
+namespace Microsoft.Graph.Beta.PowerShell
+{
+    public static class EventFactory
+    {
+        /// <summary>
+        /// Create a tracing event containing a string message
+        /// </summary>
+        /// <param name="message">The string message to include in event data</param>
+        /// <returns>Valid EventData containing the message</returns>
+        public static EventData CreateLogEvent(Task<string> message)
+        {
+            return new EventData
+            {
+                Id = Guid.NewGuid().ToString(),
+                Message = message.Result
+            };
+        }
+
+        /// <summary>
+        /// Create a new debug message event
+        /// </summary>
+        /// <param name="message">The message</param>
+        /// <returns>An event containing the debug message</returns>
+        public static EventData CreateDebugEvent(string message)
+        {
+            return new EventData
+            {
+                Id = Events.Debug,
+                Message = message
+            };
+        }
+
+        /// <summary>
+        /// Create a new debug message event
+        /// </summary>
+        /// <param name="message">The message</param>
+        /// <returns>An event containing the debug message</returns>
+        public static EventData CreateWarningEvent(string message)
+        {
+            return new EventData
+            {
+                Id = Events.Warning,
+                Message = message
+            };
+        }
+        public static string SplitPascalCase(string word)
+        {
+            var regex = new Regex("([a-z]+)([A-Z])");
+            var output = regex.Replace(word, "$1 $2");
+            regex = new Regex("([A-Z])([A-Z][a-z])");
+            return regex.Replace(output, "$1 $2");
+        }
+
+        public static EventArgs CreateLogEvent(string message)
+        {
+            return new EventData
+            {
+                Id = Guid.NewGuid().ToString(),
+                Message = message
+            };
+        }
+    }
+}

src/Migrations/beta/custom/FileUploadCmdlet.cs

@@ -0,0 +1,44 @@
+// ------------------------------------------------------------------------------
+//  Copyright (c) Microsoft Corporation.  All Rights Reserved.  Licensed under the MIT License.  See License in the project root for license information.
+// ------------------------------------------------------------------------------
+namespace Microsoft.Graph.Beta.PowerShell.Cmdlets.Custom
+{
+    using Microsoft.Graph.PowerShell.Authentication.Common;
+    using System.IO;
+    using System.Management.Automation;
+
+    public partial class FileUploadCmdlet : PSCmdlet
+    {
+        /// <summary>Backing field for <see cref="InFile" /> property.</summary>
+        private string _inFile;
+
+        /// <summary>The path to the file to upload. This SHOULD include the file name and extension.</summary>
+        [Parameter(Mandatory = true, HelpMessage = "The path to the file to upload. This should include a path and file name. If you omit the path, the current location will be used.")]
+        [Runtime.Info(
+        Required = true,
+        ReadOnly = false,
+        Description = @"The path to the file to upload. This should include a path and file name. If you omit the path, the current location will be used.",
+        PossibleTypes = new[] { typeof(string) })]
+        [ValidateNotNullOrEmpty()]
+        [Category(ParameterCategory.Runtime)]
+        public string InFile { get => this._inFile; set => this._inFile = value; }
+
+        /// <summary>
+        /// Creates a file stream from the provided input file.
+        /// </summary>
+        /// <returns>A file stream.</returns>
+        internal Stream GetFileAsStream()
+        {
+            if (MyInvocation.BoundParameters.ContainsKey(nameof(InFile)))
+            {
+                string resolvedFilePath = this.GetProviderPath(InFile, true);
+                var fileProvider = ProtectedFileProvider.CreateFileProvider(resolvedFilePath, FileProtection.SharedRead, new DiskDataStore());
+                return fileProvider.Stream;
+            }
+            else
+            {
+                return null;
+            }
+        }
+    }
+}

src/Migrations/beta/custom/HttpMessageLogFormatter.cs

@@ -0,0 +1,193 @@
+// ------------------------------------------------------------------------------
+//  Copyright (c) Microsoft Corporation.  All Rights Reserved.  Licensed under the MIT License.  See License in the project root for license information.
+// ------------------------------------------------------------------------------
+
+namespace Microsoft.Graph.Beta.PowerShell
+{
+    using Microsoft.Graph.Beta.PowerShell.Models;
+    using Newtonsoft.Json;
+    using System;
+    using System.Collections.Generic;
+    using System.IO;
+    using System.Linq;
+    using System.Net.Http;
+    using System.Net.Http.Headers;
+    using System.Text;
+    using System.Text.RegularExpressions;
+    using System.Threading.Tasks;
+    using System.Xml.Linq;
+
+    public static class HttpMessageLogFormatter
+    {
+        internal static async Task<HttpRequestMessage> CloneAsync(this HttpRequestMessage originalRequest)
+        {
+            var newRequest = new HttpRequestMessage(originalRequest.Method, originalRequest.RequestUri);
+
+            // Copy requestClone headers.
+            foreach (var header in originalRequest.Headers)
+                newRequest.Headers.TryAddWithoutValidation(header.Key, header.Value);
+
+            // Copy requestClone properties.
+            foreach (var property in originalRequest.Properties)
+                newRequest.Properties.Add(property);
+
+            // Set Content if previous requestClone had one.
+            if (originalRequest.Content != null)
+            {
+                // Try cloning request content; otherwise, skip due to https://github.com/dotnet/corefx/pull/19082 in .NET 4.x.
+                try
+                {
+                    // HttpClient doesn't rewind streams and we have to explicitly do so.
+                    var ms = new MemoryStream();
+                    await originalRequest.Content.CopyToAsync(ms).ConfigureAwait(false);
+                    ms.Position = 0;
+                    newRequest.Content = new StreamContent(ms);
+                    // Attempt to copy request content headers with a single retry.
+                    // HttpHeaders dictionary is not thread-safe when targeting anything below .NET 7. For more information, see https://github.com/dotnet/runtime/issues/61798.
+                    int retryCount = 0;
+                    int maxRetryCount = 2;
+                    while (retryCount < maxRetryCount)
+                    {
+                        try
+                        {
+                            originalRequest.Content?.Headers?.ToList().ForEach(header => newRequest.Content?.Headers.TryAddWithoutValidation(header.Key, header.Value));
+                            retryCount = maxRetryCount;
+                        }
+                        catch (InvalidOperationException)
+                        {
+                            retryCount++;
+                        }
+                    }
+                }
+                catch
+                {
+                }
+            }
+            return newRequest;
+        }
+
+        public static async Task<string> GetHttpRequestLogAsync(HttpRequestMessage request)
+        {
+            if (request == null) return string.Empty;
+            var requestClone = await request.CloneAsync().ConfigureAwait(false);
+            string body = string.Empty;
+            try
+            {
+                if (requestClone.Content != null)
+                {
+                    body = FormatString(await requestClone.Content.ReadAsStringAsync());
+                }
+                else if (requestClone.Content == null && request.Content != null)
+                {
+                    body = "Skipped: Content body was disposed before the logger could access it.";
+                }
+            }
+            catch { }
+
+            StringBuilder stringBuilder = new StringBuilder();
+            stringBuilder.AppendLine($"============================ HTTP REQUEST ============================{Environment.NewLine}");
+            stringBuilder.AppendLine($"HTTP Method:{Environment.NewLine}{requestClone.Method}{Environment.NewLine}");
+            stringBuilder.AppendLine($"Absolute Uri:{Environment.NewLine}{requestClone.RequestUri}{Environment.NewLine}");
+            stringBuilder.AppendLine($"Headers:{Environment.NewLine}{HeadersToString(requestClone.Headers)}{Environment.NewLine}");
+            stringBuilder.AppendLine($"Body:{Environment.NewLine}{SanitizeBody(body)}{Environment.NewLine}");
+            return stringBuilder.ToString();
+        }
+
+        public static async Task<string> GetHttpResponseLogAsync(HttpResponseMessage response)
+        {
+            if (response == null) return string.Empty;
+
+            string body = string.Empty;
+            try
+            {
+                body = (response.Content == null) ? string.Empty : FormatString(await response.Content.ReadAsStringAsync());
+            }
+            catch { }
+
+            StringBuilder stringBuilder = new StringBuilder();
+            stringBuilder.AppendLine($"============================ HTTP RESPONSE ============================{Environment.NewLine}");
+            stringBuilder.AppendLine($"Status Code:{Environment.NewLine}{response.StatusCode}{Environment.NewLine}");
+            stringBuilder.AppendLine($"Headers:{Environment.NewLine}{HeadersToString(response.Headers)}{Environment.NewLine}");
+            stringBuilder.AppendLine($"Body:{Environment.NewLine}{SanitizeBody(body)}{Environment.NewLine}");
+            return stringBuilder.ToString();
+        }
+
+        public static async Task<string> GetErrorLogAsync(HttpResponseMessage response, IMicrosoftGraphODataErrorsMainError odataError)
+        {
+            if (response == null) return string.Empty;
+
+            StringBuilder stringBuilder = new StringBuilder();
+            stringBuilder.AppendLine($"{odataError?.Message}{Environment.NewLine}");
+            stringBuilder.AppendLine($"Status: {((int)response.StatusCode)} ({response.StatusCode})");
+            stringBuilder.AppendLine($"ErrorCode: {odataError?.Code}");
+            stringBuilder.AppendLine($"Date: {odataError?.InnerError?.Date}{Environment.NewLine}");
+            stringBuilder.AppendLine($"Headers:{Environment.NewLine}{HeadersToString(response.Headers)}{Environment.NewLine}");
+            return stringBuilder.ToString();
+        }
+
+        internal static string HeadersToString(HttpHeaders headers)
+        {
+            return HeadersToString(ConvertHttpHeadersToCollection(headers));
+        }
+
+        private static readonly Regex regexPattern = new Regex("(\\s*\"access_token\"\\s*:\\s*)\"[^\"]+\"", RegexOptions.Compiled);
+        private static object SanitizeBody(string body)
+        {
+            IList<Regex> regexList = new List<Regex>();
+            // Remove access_token:*  instances in body.
+            regexList.Add(regexPattern);
+
+            foreach (Regex matcher in regexList)
+            {
+                body = matcher.Replace(body, "$1\"<redacted>\"");
+            }
+
+            return body;
+        }
+
+        private static IDictionary<string, IEnumerable<string>> ConvertHttpHeadersToCollection(HttpHeaders headers)
+        {
+            headers.Remove("Authorization");
+            return headers.ToDictionary(a => a.Key, a => a.Value);
+        }
+
+        private static string HeadersToString(IDictionary<string, IEnumerable<string>> headers)
+        {
+            StringBuilder stringBuilder = headers.Aggregate(new StringBuilder(),
+                (sb, kvp) => sb.AppendLine(string.Format("{0,-30}: {1}", kvp.Key, String.Join(",", kvp.Value.ToArray()))));
+
+            if (stringBuilder.Length > 0)
+                stringBuilder.Remove(stringBuilder.Length - 2, 2);
+
+            return stringBuilder.ToString();
+        }
+
+        private static string FormatString(string content)
+        {
+            try
+            {
+                content = content.Trim();
+                if ((content.StartsWith("{") && content.EndsWith("}")) || // object
+                    (content.StartsWith("[") && content.EndsWith("]"))) // array
+                {
+                    return JsonConvert.SerializeObject(JsonConvert.DeserializeObject(content), Formatting.Indented);
+                }
+                if (content.StartsWith("<"))
+                {
+                    return XDocument.Parse(content).ToString();
+                }
+            }
+            catch
+            {
+                return content;
+            }
+
+            if (content.Length > Microsoft.Graph.PowerShell.Authentication.Constants.MaxContentLength)
+            {
+                return content.Substring(0, Microsoft.Graph.PowerShell.Authentication.Constants.MaxContentLength) + "\r\nDATA TRUNCATED DUE TO SIZE\r\n";
+            }
+
+            return content;
+        }
+    }
+}