From 84f35cab5e6f21a706ffb53cdfdb7398cdae1949 Mon Sep 17 00:00:00 2001
From: David Fowler <davidfowl@gmail.com>
Date: Sat, 25 Jan 2014 05:31:46 -0800
Subject: [PATCH] Handle defines and unsafe code.

---
 build/_k-generate-projects.shade | 17 +++++++++++++++--
 build/k10.txt                    |  8 ++++----
 build/net45.txt                  |  8 ++++----
 3 files changed, 23 insertions(+), 10 deletions(-)

diff --git a/build/_k-generate-projects.shade b/build/_k-generate-projects.shade
index 100205073a..533ce7265d 100644
--- a/build/_k-generate-projects.shade
+++ b/build/_k-generate-projects.shade
@@ -150,6 +150,7 @@ functions
                 var d = serializer.DeserializeObject(jsonText) as IDictionary<string, object>;
                 var configs = GetObject(d, "configurations") ?? new Dictionary<string, object>();
                 var dependencies = GetObject(d, "dependencies") ?? new Dictionary<string, object>();
+                var compilationOptions = GetObject(d, "compilationOptions") ?? new Dictionary<string, object>();
 
                 if(configs.Count == 0)
                 {
@@ -173,6 +174,11 @@ functions
                                                     .Select(r => r.Key)
                                                     .ToList();
 
+                var defines = Get<IEnumerable<object>>(compilationOptions, "define") ?? new object[0];
+                object unsafeValue = Get<object>(compilationOptions, "allowUnsafe");
+                bool allowUnsafeCode = unsafeValue == null ? false : (bool)unsafeValue;
+
+                string extraProperties = allowUnsafeCode ? "\n<AllowUnsafeBlocks>true</AllowUnsafeBlocks>" : "";
 
                 // HACK: Assume the packages folder is 2 up from the projectDir
                 string packagesDir = Path.GetFullPath(Path.Combine(projectDir, "..", "..", "packages"));
@@ -205,6 +211,8 @@ functions
                     var template = templates[targetFramework]
                                        .Replace("{ProjectGuid}", id)
                                        .Replace("{Name}", projectName)
+                                       .Replace("{Defines}", String.Join(";", defines.Select(def => def.ToString())))
+                                       .Replace("{ExtraProperties}", extraProperties)
                                        .Replace("{Files}", filesString)
                                        .Replace("{ProjectReferences}", BuildProjectReferences(projectReferences, targetFramework, projectMapping))
                                        .Replace("{References}", BuildReferences(packageReferences, gacReferences, packagesDir, targetFramework, GetCandidates(targetFramework)));
@@ -340,15 +348,20 @@ functions
                 return sb.ToString();
             }
 
-            private static IDictionary<string, object> GetObject(IDictionary<string, object> obj, string key)
+            private static T Get<T>(IDictionary<string, object> obj, string key) where T : class
             {
                 object value;
                 if (obj.TryGetValue(key, out value))
                 {
-                    return value as IDictionary<string, object>;
+                    return value as T;
                 }
 
                 return null;
             }
+
+            private static IDictionary<string, object> GetObject(IDictionary<string, object> obj, string key)
+            {
+                return Get<IDictionary<string, object>>(obj, key);
+            }
         }
     }
diff --git a/build/k10.txt b/build/k10.txt
index 98924c1a4a..060f9f89e6 100644
--- a/build/k10.txt
+++ b/build/k10.txt
@@ -19,18 +19,18 @@
     <DebugType>full</DebugType>
     <Optimize>false</Optimize>
     <OutputPath>bin\Debug\K</OutputPath>
-    <DefineConstants>DEBUG;TRACE;K10</DefineConstants>
+    <DefineConstants>DEBUG;TRACE;K10;{Defines}</DefineConstants>
     <ErrorReport>prompt</ErrorReport>
-    <WarningLevel>4</WarningLevel>
+    <WarningLevel>4</WarningLevel>{ExtraProperties}
   </PropertyGroup>
   <PropertyGroup Condition=" '$(Configuration)|$(Platform)' == 'Release|AnyCPU' ">
     <PlatformTarget>AnyCPU</PlatformTarget>
     <DebugType>pdbonly</DebugType>
     <Optimize>true</Optimize>
     <OutputPath>bin\Release\K</OutputPath>
-    <DefineConstants>TRACE;K10</DefineConstants>
+    <DefineConstants>TRACE;K10;{Defines}</DefineConstants>
     <ErrorReport>prompt</ErrorReport>
-    <WarningLevel>4</WarningLevel>
+    <WarningLevel>4</WarningLevel>{ExtraProperties}
   </PropertyGroup>
   <ItemGroup>
     {Files}
diff --git a/build/net45.txt b/build/net45.txt
index 8d2e271f7b..c24abbc2d8 100644
--- a/build/net45.txt
+++ b/build/net45.txt
@@ -19,18 +19,18 @@
     <DebugType>full</DebugType>
     <Optimize>false</Optimize>
     <OutputPath>bin\Debug\net45</OutputPath>
-    <DefineConstants>DEBUG;TRACE;NET45</DefineConstants>
+    <DefineConstants>DEBUG;TRACE;NET45;{Defines}</DefineConstants>
     <ErrorReport>prompt</ErrorReport>
-    <WarningLevel>4</WarningLevel>
+    <WarningLevel>4</WarningLevel>{ExtraProperties}
   </PropertyGroup>
   <PropertyGroup Condition=" '$(Configuration)|$(Platform)' == 'Release|AnyCPU' ">
     <PlatformTarget>AnyCPU</PlatformTarget>
     <DebugType>pdbonly</DebugType>
     <Optimize>true</Optimize>
     <OutputPath>bin\Release\net45</OutputPath>
-    <DefineConstants>TRACE;NET45</DefineConstants>
+    <DefineConstants>TRACE;NET45;{Defines}</DefineConstants>
     <ErrorReport>prompt</ErrorReport>
-    <WarningLevel>4</WarningLevel>
+    <WarningLevel>4</WarningLevel>{ExtraProperties}
   </PropertyGroup>
   <ItemGroup>
     <Reference Include="mscorlib" />