Merge pull request #49 from kashifsoofi/master

#44 Display a friendly message to indicate possible problem.
kontent-ai · Jan 19, 2018 · 0b9671f · 0b9671f
2 parents b06c614 + fa2ce75
commit 0b9671f
Showing 1 changed file with 24 additions and 15 deletions.
diff --git a/src/CloudModelGenerator/CodeGenerator.cs b/src/CloudModelGenerator/CodeGenerator.cs
@@ -36,13 +36,19 @@ public void GenerateContentTypeModels(bool structuredModel = false)
             Directory.CreateDirectory(_options.OutputDir);
 
             var classCodeGenerators = GetClassCodeGenerators(structuredModel);
+            if (classCodeGenerators.Count() > 0)
+            {
+                foreach (var codeGenerator in classCodeGenerators)
+                {
+                    SaveToFile(codeGenerator.GenerateCode(), codeGenerator.ClassFilename, codeGenerator.OverwriteExisting);
+                }
 
-            foreach (var codeGenerator in classCodeGenerators)
+                Console.WriteLine($"{classCodeGenerators.Count()} content type models were successfully created.");
+            }
+            else
             {
-                SaveToFile(codeGenerator.GenerateCode(_options.ContentManagementApi), codeGenerator.ClassFilename, codeGenerator.OverwriteExisting);
+                Console.WriteLine($"No content type available for the project ({_options.ProjectId}). Please make sure you've enabled the delivery API.");
             }
-
-            Console.WriteLine($"{classCodeGenerators.Count()} content type models were successfully created.");
         }
 
         public void GenerateTypeProvider()
@@ -51,22 +57,25 @@ public void GenerateTypeProvider()
             Directory.CreateDirectory(_options.OutputDir);
 
             var classCodeGenerators = GetClassCodeGenerators();
-            var typeProviderCodeGenerator = new TypeProviderCodeGenerator(_options.Namespace);
-
-            foreach (var codeGenerator in classCodeGenerators)
+            if (classCodeGenerators.Count() > 0)
             {
-                typeProviderCodeGenerator.AddContentType(codeGenerator.ClassDefinition.Codename, codeGenerator.ClassDefinition.ClassName);
-            }
+                var typeProviderCodeGenerator = new TypeProviderCodeGenerator(_options.Namespace);
 
-            var typeProviderCode = typeProviderCodeGenerator.GenerateCode();
-            if (!string.IsNullOrEmpty(typeProviderCode))
-            {
-                SaveToFile(typeProviderCode, TypeProviderCodeGenerator.CLASS_NAME);
-                Console.WriteLine($"{TypeProviderCodeGenerator.CLASS_NAME} class was successfully created.");
+                foreach (var codeGenerator in classCodeGenerators)
+                {
+                    typeProviderCodeGenerator.AddContentType(codeGenerator.ClassDefinition.Codename, codeGenerator.ClassDefinition.ClassName);
+                }
+
+                var typeProviderCode = typeProviderCodeGenerator.GenerateCode();
+                if (!string.IsNullOrEmpty(typeProviderCode))
+                {
+                    SaveToFile(typeProviderCode, TypeProviderCodeGenerator.CLASS_NAME);
+                    Console.WriteLine($"{TypeProviderCodeGenerator.CLASS_NAME} class was successfully created.");
+                }
             }
             else
             {
-                Console.WriteLine($"{TypeProviderCodeGenerator.CLASS_NAME} class was not created.");
+                Console.WriteLine($"No content type available for the project ({_options.ProjectId}). Please make sure you've enabled the delivery API.");
             }
         }