From ac14c3222b60d71bf667eb46641fe9323602d9e5 Mon Sep 17 00:00:00 2001
From: SonjaKhan <sonjak3@uw.edu>
Date: Wed, 29 Oct 2014 10:51:34 -0700
Subject: [PATCH] making page generator work for different libraries

---
 DiagnosticsPages.sln         |  2 +-
 src/PageGenerator/Program.cs | 22 +++++++++++++++++-----
 2 files changed, 18 insertions(+), 6 deletions(-)

diff --git a/DiagnosticsPages.sln b/DiagnosticsPages.sln
index 1850737a64..f3c669a4f8 100644
--- a/DiagnosticsPages.sln
+++ b/DiagnosticsPages.sln
@@ -1,7 +1,7 @@
 
 Microsoft Visual Studio Solution File, Format Version 12.00
 # Visual Studio 14
-VisualStudioVersion = 14.0.22129.1
+VisualStudioVersion = 14.0.22228.0
 MinimumVisualStudioVersion = 10.0.40219.1
 Project("{2150E333-8FDC-42A3-9474-1A3956D46DE8}") = "src", "src", "{509A6F36-AD80-4A18-B5B1-717D38DFF29D}"
 EndProject
diff --git a/src/PageGenerator/Program.cs b/src/PageGenerator/Program.cs
index fcc9b074bc..a4f84c784f 100644
--- a/src/PageGenerator/Program.cs
+++ b/src/PageGenerator/Program.cs
@@ -12,6 +12,8 @@ namespace PageGenerator
 {
     public class Program
     {
+        private const int NumArgs = 1;
+
         private readonly ILibraryManager _libraryManager;
 
         public Program(ILibraryManager libraryManager)
@@ -21,7 +23,17 @@ namespace PageGenerator
 
         public void Main(string[] args)
         {
-            var diagnosticsLibInfo = _libraryManager.GetLibraryInformation("Microsoft.AspNet.Diagnostics");
+            if (args.Length != NumArgs)
+            {
+                throw new ArgumentException(string.Format("Requires {0} argument (Library Name), {1} given", NumArgs, args.Length));
+            }
+            var diagnosticsLibInfo = _libraryManager.GetLibraryInformation(args[0]);
+            if (diagnosticsLibInfo == null)
+            {
+                throw new ArgumentException(string.Format(
+                    "Unable to open library {0}. Is it spelled correctly and listed as a dependency in project.json?", 
+                    args[0]));
+            }
             var viewBasePath = Path.Combine(Path.GetDirectoryName(diagnosticsLibInfo.Path), "Views");
 
             Console.WriteLine("Generating code files for views in {0}", viewBasePath);
@@ -33,7 +45,7 @@ namespace PageGenerator
             foreach (var fileName in cshtmlFiles)
             {
                 Console.WriteLine("  Generating code file for view {0}...", Path.GetFileName(fileName));
-                GenerateCodeFile(fileName);
+                GenerateCodeFile(fileName, string.Format("{0}.Views", args[0]));
                 Console.WriteLine("      Done!");
                 fileCount++;
             }
@@ -56,7 +68,7 @@ namespace PageGenerator
             return Directory.EnumerateFiles(path, "*.cshtml");
         }
 
-        private static void GenerateCodeFile(string cshtmlFilePath)
+        private static void GenerateCodeFile(string cshtmlFilePath, string rootNamespace)
         {
             var basePath = Path.GetDirectoryName(cshtmlFilePath);
             var fileName = Path.GetFileName(cshtmlFilePath);
@@ -65,13 +77,13 @@ namespace PageGenerator
             var host = new RazorEngineHost(codeLang);
             host.DefaultBaseClass = "Microsoft.AspNet.Diagnostics.Views.BaseView";
             var engine = new RazorTemplateEngine(host);
-            
+
             using (var fileStream = File.OpenText(cshtmlFilePath))
             {
                 var code = engine.GenerateCode(
                     input: fileStream,
                     className: fileNameNoExtension,
-                    rootNamespace: "Microsoft.AspNet.Diagnostics.Views",
+                    rootNamespace: rootNamespace,
                     sourceFileName: fileName);
 
                 var source = code.GeneratedCode;