VisualStudioProjectFileHook.cs

Source code

// %BANNER_BEGIN%
// ---------------------------------------------------------------------
// %COPYRIGHT_BEGIN%
// Copyright (c) (2024) Magic Leap, Inc. All Rights Reserved.
// Use of this file is governed by the Software License Agreement, located here: https://www.magicleap.com/software-license-agreement-ml2
// Terms and conditions applicable to third-party materials accompanying this distribution may also be found in the top-level NOTICE file appearing herein.
// %COPYRIGHT_END%
// ---------------------------------------------------------------------
// %BANNER_END%
using System.IO;
using System.Text;
using System.Xml.Linq;
using System.Text.RegularExpressions;
using System.Collections.Generic;
using UnityEditor;

namespace MagicLeap
{
    public class VisualStudioProjectFileHook : AssetPostprocessor
    {
        class Utf8StringWriter : StringWriter
        {
            public override Encoding Encoding => Encoding.UTF8;
        }

        public static string OnGeneratedCSProject(string path, string content)
        {
            // Only change for project files starting with "MagicLeap.SDK"
            // i.e. "MagicLeap.SDK.csproj" & "MagicLeap.SDK.Editor.csproj"
            if (!Path.GetFileName(path).StartsWith("MagicLeap.SDK"))
            {
                return content;
            }

            // Parse the document and make some changes
            XDocument document = XDocument.Parse(content);
            AdjustDocument(document);

            // Save the changes using the Utf8StringWriter
            Utf8StringWriter str = new Utf8StringWriter();
            document.Save(str);

            return str.ToString();
        }

        static void AdjustDocument(XDocument document)
        {
            // Get namespace of document
            XNamespace ns = document.Root.Name.Namespace;

            string[] elementsToConsider = { "Compile", "None" };

            foreach (string element in elementsToConsider)
            {
                // Get all elements
                IEnumerable<XElement> compileElements = document.Root.Descendants(ns + element);

                // Regex to find which part of Include attribute of Compile element to use for Link element value.
                Regex regex = new Regex(@"\\(Packages)\\.*\.(cs|asmdef|shader)$");

                // Add child Link element to each Compile element
                foreach (XElement el in compileElements)
                {
                    string fileName = el.Attribute("Include").Value;
                    Match match = regex.Match(fileName);
                    if (match.Success)
                    {
                        // Substr from 1 to exclude initial slash character
                        XElement link = new XElement(ns + "Link")
                        {
                            Value = match.Value.Substring(1)
                        };

                        el.Add(link);
                    }
                }
            }
        }
    }
}