public void GenerateDataContextInProject(object sender, ExecutedRoutedEventArgs e)
{
var databaseInfo = ValidateMenuInfo(sender);
if (databaseInfo == null)
{
return;
}
var isDesktop = (bool)((MenuItem)sender).Tag;
if (package == null)
{
return;
}
var dte = package.GetServiceHelper(typeof(DTE)) as DTE;
if (dte == null)
{
return;
}
if (dte.Mode == vsIDEMode.vsIDEModeDebug)
{
EnvDteHelper.ShowError("Cannot generate code while debugging");
return;
}
var helper = Helpers.RepositoryHelper.CreateEngineHelper(databaseInfo.DatabaseInfo.DatabaseType);
if (!helper.IsV35DbProviderInstalled())
{
EnvDteHelper.ShowError("This feature requires the SQL Server Compact 3.5 SP2 DbProvider to be properly installed");
return;
}
var dteH = new EnvDteHelper();
var project = dteH.GetProject(dte);
if (project == null)
{
EnvDteHelper.ShowError("Please select a project in Solution Explorer, where you want the DataContext to be placed");
return;
}
if (!isDesktop && !dteH.AllowedWpProjectKinds.Contains(new Guid(project.Kind)))
{
EnvDteHelper.ShowError("The selected project type does not support Windows Phone (please let me know if I am wrong)");
return;
}
if (isDesktop && !dteH.AllowedProjectKinds.Contains(new Guid(project.Kind)))
{
EnvDteHelper.ShowError("The selected project type does not support LINQ to SQL (please let me know if I am wrong)");
return;
}
if (project.Properties.Item("TargetFrameworkMoniker") == null)
{
EnvDteHelper.ShowError("The selected project type does not support Windows Phone - missing TargetFrameworkMoniker");
return;
}
if (!isDesktop)
{
if (project.Properties.Item("TargetFrameworkMoniker").Value.ToString() == "Silverlight,Version=v4.0,Profile=WindowsPhone71" ||
project.Properties.Item("TargetFrameworkMoniker").Value.ToString() == "WindowsPhone,Version=v8.0" ||
project.Properties.Item("TargetFrameworkMoniker").Value.ToString() == "WindowsPhone,Version=v8.1"
)
{
}
else
{
EnvDteHelper.ShowError("The selected project type does not support Windows Phone 7.1/8.0 - wrong TargetFrameworkMoniker: " + project.Properties.Item("TargetFrameworkMoniker").Value);
return;
}
}
if (isDesktop && !project.Properties.Item("TargetFrameworkMoniker").Value.ToString().Contains(".NETFramework"))
{
EnvDteHelper.ShowError("The selected project type does not support .NET Desktop - wrong TargetFrameworkMoniker: " + project.Properties.Item("TargetFrameworkMoniker").Value);
return;
}
if (!isDesktop && databaseInfo.DatabaseInfo.DatabaseType != DatabaseType.SQLCE35)
{
EnvDteHelper.ShowError("Sorry, only version 3.5 databases are supported for now");
return;
}
var sqlMetalPath = ProbeSqlMetalRegPaths();
if (string.IsNullOrEmpty(sqlMetalPath))
{
EnvDteHelper.ShowError("Could not find SQLMetal file location");
return;
}
var sdfFileName = string.Empty;
try
{
using (var repository = Helpers.RepositoryHelper.CreateRepository(databaseInfo.DatabaseInfo))
{
var tables = repository.GetAllTableNames();
var pks = repository.GetAllPrimaryKeys();
var checkedTables = string.Empty;
foreach (var tableName in tables)
{
var pk = pks.Where(k => k.TableName == tableName).FirstOrDefault();
if (pk.TableName == null)
{
checkedTables += tableName + Environment.NewLine;
}
}
if (!string.IsNullOrEmpty(checkedTables))
{
var message = string.Format("The tables below do not have Primary Keys defined,{0}and will not be generated properly:{1}{2}", Environment.NewLine, Environment.NewLine, checkedTables);
EnvDteHelper.ShowError(message);
}
var dbInfo = repository.GetDatabaseInfo();
foreach (var kvp in dbInfo)
{
if (kvp.Key == "Database")
{
sdfFileName = kvp.Value;
break;
}
}
sdfFileName = Path.GetFileName(sdfFileName);
}
var fileNameWithoutExtension = Path.GetFileNameWithoutExtension(databaseInfo.DatabaseInfo.Caption);
if (fileNameWithoutExtension != null)
{
var model = fileNameWithoutExtension.Replace(" ", string.Empty).Replace("#", string.Empty).Replace(".", string.Empty).Replace("-", string.Empty);
model = model + "Context";
var dcDialog = new DataContextDialog();
dcDialog.ModelName = model;
dcDialog.IsDesktop = isDesktop;
dcDialog.ProjectName = project.Name;
dcDialog.NameSpace = project.Properties.Item("DefaultNamespace").Value.ToString();
if (EnvDteHelper.VbProject == new Guid(project.Kind))
{
dcDialog.CodeLanguage = "VB";
}
else
{
dcDialog.CodeLanguage = "C#";
}
var result = dcDialog.ShowModal();
if (!result.HasValue || result.Value != true || string.IsNullOrWhiteSpace(dcDialog.ModelName))
{
return;
}
if (dcDialog.AddRowversionColumns)
{
AddRowVersionColumns(databaseInfo);
}
var sdfPath = databaseInfo.DatabaseInfo.ConnectionString;
//If version 4.0, create a 3.5 schema sdf, and use that as connection string
if (isDesktop && databaseInfo.DatabaseInfo.DatabaseType == DatabaseType.SQLCE40)
{
var tempFile = Path.GetTempFileName();
using (var repository = Helpers.RepositoryHelper.CreateRepository(databaseInfo.DatabaseInfo))
{
var generator = DataConnectionHelper.CreateGenerator(repository, tempFile, databaseInfo.DatabaseInfo.DatabaseType);
generator.ScriptDatabaseToFile(Scope.Schema);
}
if (sdfFileName != null)
{
sdfPath = Path.Combine(Path.GetTempPath(), sdfFileName);
}
using (Stream stream = new MemoryStream(Resources.SqlCe35AddinStore))
{
// Create a FileStream object to write a stream to a file
using (var fileStream = File.Create(sdfPath, (int)stream.Length))
{
// Fill the bytes[] array with the stream data
var bytesInStream = new byte[stream.Length];
stream.Read(bytesInStream, 0, bytesInStream.Length);
// Use FileStream object to write to the specified file
fileStream.Write(bytesInStream, 0, bytesInStream.Length);
}
}
var info = new DatabaseInfo
{
ConnectionString = "Data Source=" + sdfPath,
DatabaseType = DatabaseType.SQLCE35
};
using (var repository = Helpers.RepositoryHelper.CreateRepository(info))
{
var script = File.ReadAllText(tempFile);
repository.ExecuteSql(script);
}
sdfPath = info.ConnectionString;
}
var versionNumber = GetVersionTableNumber(databaseInfo.DatabaseInfo, isDesktop);
model = dcDialog.ModelName;
var dcPath = Path.Combine(Path.GetTempPath(), model + ".cs");
if (dcDialog.CodeLanguage == "VB")
{
dcPath = Path.Combine(Path.GetTempPath(), model + ".vb");
}
var parameters = " /provider:SQLCompact /code:\"" + dcPath + "\"";
parameters += " /conn:\"" + sdfPath + "\"";
parameters += " /context:" + model;
if (dcDialog.Pluralize)
{
parameters += " /pluralize";
}
if (!string.IsNullOrWhiteSpace(dcDialog.NameSpace))
{
parameters += " /namespace:" + dcDialog.NameSpace;
}
var dcH = new DataContextHelper();
var sqlmetalResult = dcH.RunSqlMetal(sqlMetalPath, parameters);
if (!File.Exists(dcPath))
{
EnvDteHelper.ShowError("Error during SQL Metal run: " + sqlmetalResult);
return;
}
if (!isDesktop)
{
using (var repository = Helpers.RepositoryHelper.CreateRepository(databaseInfo.DatabaseInfo))
{
if (dcDialog.CodeLanguage == "VB")
{
DataContextHelper.FixDataContextVB(dcPath, model, dcDialog.NameSpace, sdfFileName, repository);
}
else
{
DataContextHelper.FixDataContextCS(dcPath, model, dcDialog.NameSpace, sdfFileName, repository);
}
}
}
if (dcDialog.MultipleFiles)
{
var classes = DataContextHelper.SplitIntoMultipleFiles(dcPath, dcDialog.NameSpace, model);
var projectPath = project.Properties.Item("FullPath").Value.ToString();
foreach (var item in classes)
{
var fileName = Path.Combine(projectPath, item.Key + ".cs");
if (File.Exists(fileName))
{
File.Delete(fileName);
}
File.WriteAllText(fileName, item.Value);
var classItem = dteH.GetProjectDataContextClass(project, fileName);
if (classItem != null)
{
classItem.Delete();
}
project.ProjectItems.AddFromFile(fileName);
}
}
else
{
var extension = ".cs";
if (dcDialog.CodeLanguage == "VB")
{
extension = ".vb";
}
var dcItem = dteH.GetProjectDc(project, model, extension);
if (dcItem == null)
{
project.ProjectItems.AddFromFileCopy(dcPath);
}
else
{
if (EnvDteHelper.ShowMessageBox("The Data Context class already exists in the project, do you wish to replace it?", OLEMSGBUTTON.OLEMSGBUTTON_YESNO, OLEMSGDEFBUTTON.OLEMSGDEFBUTTON_SECOND, OLEMSGICON.OLEMSGICON_QUERY) == System.Windows.Forms.DialogResult.Yes)
{
dcItem.Delete();
project.ProjectItems.AddFromFileCopy(dcPath);
}
}
}
EnvDteHelper.AddReference(project, "System.Data.Linq");
if (dcDialog.AddConnectionStringBuilder)
{
var projectPath = project.Properties.Item("FullPath").Value.ToString();
var fileName = "LocalDatabaseConnectionStringBuilder.cs";
var filePath = Path.Combine(projectPath, fileName);
if (File.Exists(filePath))
{
File.Delete(filePath);
}
using (Stream stream = new MemoryStream(Resources.LocalDatabaseConnectionStringBuilder))
{
// Create a FileStream object to write a stream to a file
using (var fileStream = File.Create(filePath, (int)stream.Length))
{
// Fill the bytes[] array with the stream data
var bytesInStream = new byte[stream.Length];
stream.Read(bytesInStream, 0, bytesInStream.Length);
// Use FileStream object to write to the specified file
fileStream.Write(bytesInStream, 0, bytesInStream.Length);
}
}
project.ProjectItems.AddFromFile(filePath);
}
// Creates __Version table and adds one row if desired
if (dcDialog.AddVersionTable)
{
using (var repository = Helpers.RepositoryHelper.CreateRepository(databaseInfo.DatabaseInfo))
{
var list = repository.GetAllTableNames();
if (!list.Contains("__VERSION"))
{
repository.ExecuteSql(string.Format(@"
CREATE TABLE [__VERSION] (
[SchemaVersion] int NOT NULL
, [DateUpdated] datetime NOT NULL DEFAULT (GETDATE())
);
GO
CREATE INDEX [IX_SchemaVersion] ON [__VERSION] ([SchemaVersion] DESC);
GO
INSERT INTO [__VERSION] ([SchemaVersion]) VALUES ({0});
GO", versionNumber));
}
}
}
DataConnectionHelper.LogUsage("DatabaseCreateDC");
}
}
catch (Exception ex)
{
DataConnectionHelper.SendError(ex, databaseInfo.DatabaseInfo.DatabaseType, false);
}
}