MonoRail/NewGenerator/Castle.NewGenerator.CLI/Program.cs

   1 namespace Castle.NewGenerator.CLI
   2 {
   3         using System;
   4         using System.ComponentModel;
   5         using System.Reflection;
   6         using Castle.Components.Common.TemplateEngine;
   7         using Castle.Components.Common.TemplateEngine.NVelocityTemplateEngine;
   8         using Castle.NewGenerator.Core.MR;
   9         using Core;
  10         using Mono.GetOptions;
  11
  12         public class GeneratorDriver
  13         {
  14                 static void Main(string[] args)
  15                 {
  16                         AppDomain.CurrentDomain.AssemblyResolve += new ResolveEventHandler(CurrentDomain_AssemblyResolve);
  17
  18                         MainOptions opts = new MainOptions();
  19                         opts.ProcessArgs(args);
  20
  21                         if (args.Length == 0 || args[0].StartsWith("-"))
  22                         {
  23                                 ShowUsage();
  24                         }
  25
  26                         Console.WriteLine("Processing {0} {1}", args[0], opts.verbose);
  27
  28                         IGenerator generator = null;
  29
  30                         if (args[0] == "controller")
  31                         {
  32                                 generator = new NewController();
  33                         }
  34                         else if (args[0] == "project")
  35                         {
  36                                 generator = new NewProject();
  37                         }
  38                         else
  39                         {
  40                                 Console.Error.WriteLine("Not supported");
  41                                 return;
  42                         }
  43
  44                         Configure(generator, args);
  45
  46                         string workingDir = AppDomain.CurrentDomain.BaseDirectory;
  47                         string templateDir = @"C:\dev\DotNet\castle\svn\trunk\MonoRail\NewGenerator\GeneratorTemplates\";
  48
  49                         GeneratorContext context = new GeneratorContext(workingDir, templateDir);
  50
  51                         ITemplateEngine engine = new NVelocityTemplateEngine(templateDir);
  52
  53                         if (engine is ISupportInitialize)
  54                         {
  55                                 ((ISupportInitialize)engine).BeginInit();
  56                         }
  57
  58                         generator.Generate(context, new DefaultGeneratorService(context, engine));
  59                 }
  60
  61                 static Assembly CurrentDomain_AssemblyResolve(object sender, ResolveEventArgs args)
  62                 {
  63                         // C:\dev\castle\svn\trunk\build\net-2.0\debug
  64                         return Assembly.LoadFile(@"C:\dev\DotNet\castle\svn\trunk\build\net-2.0\debug\" + args.Name);
  65                 }
  66
  67                 private static void ShowUsage()
  68                 {
  69                         Console.WriteLine("CastleGen project|controller -property name=value");
  70                 }
  71
  72                 private static void Configure(IGenerator generator, string[] args)
  73                 {
  74                         Type type = generator.GetType();
  75
  76                         object[] attrs = type.GetCustomAttributes(typeof(GeneratorOptionsAttribute), false);
  77
  78                         if (attrs.Length == 1)
  79                         {
  80                                 GeneratorOptionsAttribute attr = (GeneratorOptionsAttribute) attrs[0];
  81
  82                                 Type configurerType = typeof(GeneratorConfigurer<>).MakeGenericType(type);
  83
  84                                 object configurer = Activator.CreateInstance(attr.OptionClass);
  85
  86                                 MethodInfo method = configurerType.GetMethod("Configure", BindingFlags.Instance | BindingFlags.Public);
  87
  88                                 method.Invoke(configurer, new object[] { generator, args });
  89                         }
  90                 }
  91
  92                 public class MainOptions : Options
  93                 {
  94                         [Option(1, "Verbose", 'v')]
  95                         public bool verbose;
  96                 }
  97         }
  98 }