public static void GenerateSetupInTester(ZTWPackage pkg)
{
string name = pkg.Name;
if (pkg == null)
{
throw new ArgumentNullException("pkg");
}
string testerPath = SetupController.TesterPath;
if (!SetupController.CheckTesterPathDetermined())
{
throw new Exception("Tester path is null. It must be specified by going to 'Enter Package Creator -> Locate Tester Path'.");
}
if (!testerPath.IsDirectory() && Path.GetExtension(testerPath) == "csproj" || testerPath.IsDirectory())
{
throw new Exception("The stored Tester isn't a csproj file.");
}
// First, we will create the template && store it on a file
// Declare the Hello world expression
CodeSnippetExpression helloWorld = new CodeSnippetExpression(@"Console.WriteLine(""Hello world!"")");
// Declare the class && the "OnSetup" method with its expression
var c = new CodeTypeDeclaration(pkg.SetupClass)
{
Attributes = MemberAttributes.Public,
IsClass = true,
Members =
{
new CodeMemberMethod()
{
Name = PackageConsts.OnSetupMethod,
Attributes = MemberAttributes.Public | MemberAttributes.Static,
Statements ={ new CodeExpressionStatement(helloWorld) }
},
new CodeMemberMethod()
{
Name = PackageConsts.OnFinishMethod,
Attributes = MemberAttributes.Public | MemberAttributes.Static,
Statements ={ new CodeExpressionStatement(helloWorld) }
}
},
StartDirectives =
{
new CodeRegionDirective(CodeRegionMode.Start, "\nstatic")
},
EndDirectives =
{
new CodeRegionDirective(CodeRegionMode.End, string.Empty)
}
};
// Specify and add Namespace
var ns = new CodeNamespace(pkg.SetupNamespace)
{
Types = { c }
};
// Create && add "System" import into existing namespace
ns.Imports.Add(new CodeNamespaceImport("System"));
// Then, create the unit && add everything into current namespace
var cu = new CodeCompileUnit()
{
Namespaces = { ns }
};
// Specify the language
var provider = CodeDomProvider.CreateProvider("CSharp");
// Output generated code to string Builder
var sb = new StringBuilder();
using (var sourceWriter = new StringWriter(sb))
provider.GenerateCodeFromCompileUnit(cu, sourceWriter, new CodeGeneratorOptions());
var text = sb.ToString();
string testerFolderPath = Path.GetDirectoryName(testerPath),
saveFilePath = Path.Combine(testerFolderPath, "Setups", pkg.PrettyName + ".cs"),
saveFolderPath = Path.GetDirectoryName(saveFilePath);
if (!Directory.Exists(saveFolderPath))
{
Directory.CreateDirectory(saveFolderPath);
}
// Overwrite confirmation
bool overwrite = true;
if (File.Exists(saveFilePath))
{
overwrite = SmartInput.NextConfirm("Do you want to overwrite the current file?");
}
if (overwrite)
{
File.WriteAllText(saveFilePath, text);
}
// Then, we will add the item && save it on the csproj
RoslynHelper.AddItem(testerPath, testerFolderPath, saveFilePath);
}