internal static string GetInitialCatalog(IVsDataProviderManager dataProviderManager, IVsDataConnection dataConnection)
{
string initialCatalog = null;
var invariantName = GetProviderInvariantName(dataProviderManager, dataConnection.Provider);
var props = GetConnectionProperties(dataProviderManager, dataConnection);
if (props != null)
{
if (props.ContainsKey(LocalDataUtil.CONNECTION_PROPERTY_ATTACH_DB_FILENAME)
&&
!string.IsNullOrEmpty(props[LocalDataUtil.CONNECTION_PROPERTY_ATTACH_DB_FILENAME] as string))
{
//sql client with "AttachDbFileName" parameter in the connection string.
object o = null;
props.TryGetValue(LocalDataUtil.CONNECTION_PROPERTY_ATTACH_DB_FILENAME, out o);
initialCatalog = o as String;
if (initialCatalog != null)
{
initialCatalog = Path.GetFileNameWithoutExtension(initialCatalog);
}
}
else if (LocalDataUtil.IsSqlMobileConnectionString(invariantName))
{
// sql CE
object o = null;
props.TryGetValue(LocalDataUtil.CONNECTION_PROPERTY_DATA_SOURCE, out o);
initialCatalog = o as String;
if (initialCatalog != null)
{
initialCatalog = Path.GetFileNameWithoutExtension(initialCatalog);
}
}
else
{
object o = null;
props.TryGetValue("Database", out o);
initialCatalog = o as String;
}
}
// save the default catalog
if (string.IsNullOrEmpty(initialCatalog))
{
var sourceInformation = dataConnection.GetService(typeof(IVsDataSourceInformation)) as IVsDataSourceInformation;
Debug.Assert(
sourceInformation != null,
"Could not find the IVsDataSourceInformation for this IVsDataConnection to determine the default catalog");
if (sourceInformation != null)
{
initialCatalog = sourceInformation["DefaultCatalog"] as string;
// Note: it is valid for initialCatalog to be null for certain providers which do not support that concept
}
}
return(initialCatalog);
}
private string ConvertConnectionStringToNewPath(
string providerInvariantName, string filePathKey, string oldConnectionString, string newFilePath, bool useDataDirectoryMacro)
{
if (string.IsNullOrEmpty(filePathKey))
{
Debug.Fail("requires non-null, non-empty filePathKey");
return(oldConnectionString);
}
if (LocalDataUtil.IsSqlMobileConnectionString(providerInvariantName))
{
// DbConnectionString does not support SQL Mobile
return(GenerateNewSqlMobileConnectionString(oldConnectionString, newFilePath, useDataDirectoryMacro));
}
var dbConnectionStringBuilder = new DbConnectionStringBuilder();
dbConnectionStringBuilder.ConnectionString = oldConnectionString;
object filePathObject;
dbConnectionStringBuilder.TryGetValue(filePathKey, out filePathObject);
var filePath = filePathObject as string;
if (string.IsNullOrEmpty(filePath))
{
Debug.Fail("could not find filePath for filePathKey=" + filePathKey);
return(oldConnectionString);
}
// replace old path with new one
dbConnectionStringBuilder.Remove(filePathKey);
if (useDataDirectoryMacro)
{
dbConnectionStringBuilder.Add(filePathKey, DataDirectoryMacro + Path.DirectorySeparatorChar + Path.GetFileName(newFilePath));
}
else
{
dbConnectionStringBuilder.Add(filePathKey, newFilePath);
}
return(dbConnectionStringBuilder.ConnectionString);
}
internal static string GetInitialCatalog(string providerName, string providerConnectionString)
{
var dbsb = new DbConnectionStringBuilder();
dbsb.ConnectionString = providerConnectionString;
var initialCatalog = String.Empty;
if (dbsb.ContainsKey(LocalDataUtil.CONNECTION_PROPERTY_ATTACH_DB_FILENAME)
&&
!string.IsNullOrEmpty(dbsb[LocalDataUtil.CONNECTION_PROPERTY_ATTACH_DB_FILENAME] as string))
{
//sql client with "AttachDbFileName" parameter in the connection string.
object o = null;
dbsb.TryGetValue(LocalDataUtil.CONNECTION_PROPERTY_ATTACH_DB_FILENAME, out o);
initialCatalog = o as String;
if (initialCatalog != null)
{
initialCatalog = Path.GetFileNameWithoutExtension(initialCatalog);
}
}
else if (LocalDataUtil.IsSqlMobileConnectionString(providerName))
{
// sql CE
object o = null;
dbsb.TryGetValue(LocalDataUtil.CONNECTION_PROPERTY_DATA_SOURCE, out o);
initialCatalog = o as String;
if (initialCatalog != null)
{
initialCatalog = Path.GetFileNameWithoutExtension(initialCatalog);
}
}
else
{
object o = null;
dbsb.TryGetValue("Initial Catalog", out o);
initialCatalog = o as String;
}
return(initialCatalog);
}
public override void OnActivated()
{
base.OnActivated();
_onWorkflowCleanup = false;
Debug.Assert(
!Wizard.MovingNext || _workflowInstance == null,
"Possible memory leak: We should have destroyed the old workflow instance when activating WizardPageDbGenSummary");
if (_workflowInstance == null)
{
if (LocalDataUtil.IsSqlMobileConnectionString(Wizard.ModelBuilderSettings.DesignTimeProviderInvariantName))
{
_ddlFileExtension = DatabaseGenerationEngine._sqlceFileExtension;
}
else
{
_ddlFileExtension = DatabaseGenerationEngine._ddlFileExtension;
}
// Add in the DbConfig page before if we have found a connection
if (Wizard.Mode == ModelBuilderWizardForm.WizardMode.PerformDBGenSummaryOnly &&
_addedDbConfigPage == false)
{
Wizard.InsertPageBefore(Id, new WizardPageDbConfig(Wizard));
_addedDbConfigPage = true;
}
// Display the default path for the DDL
var artifactProjectItem = VsUtils.GetProjectItemForDocument(
Wizard.ModelBuilderSettings.Artifact.Uri.LocalPath, PackageManager.Package);
if (artifactProjectItem != null)
{
txtSaveDdlAs.Text = DatabaseGenerationEngine.CreateDefaultDdlFileName(artifactProjectItem) + _ddlFileExtension;
}
// Disable all buttons except for Previous and Cancel
Wizard.EnableButton(ButtonType.Previous, true);
Wizard.EnableButton(ButtonType.Next, false);
Wizard.EnableButton(ButtonType.Finish, false);
Wizard.EnableButton(ButtonType.Cancel, true);
// Display a status message
ShowStatus(Properties.Resources.DbGenSummary_StatusDeterminingDDL);
// Extract the XML from the EDMX file and convert it into an EdmItemCollection for the workflow
EdmItemCollection edm = null;
using (new VsUtils.HourglassHelper())
{
IList <EdmSchemaError> schemaErrors;
edm = Wizard.ModelBuilderSettings.Artifact.GetEdmItemCollectionFromArtifact(out schemaErrors);
Debug.Assert(
edm != null && schemaErrors.Count == 0,
"EdmItemCollection schema errors found; we should have performed validation on the EdmItemCollection before instantiating the wizard.");
}
var existingSsdl = Wizard.ModelBuilderSettings.Artifact.GetSsdlAsString();
var existingMsl = Wizard.ModelBuilderSettings.Artifact.GetMslAsString();
// Attempt to get the workflow path, template path, and database schema name from the artifact. If we don't find them, we'll use defaults.
var workflowPath = DatabaseGenerationEngine.GetWorkflowPathFromArtifact(Wizard.ModelBuilderSettings.Artifact);
var templatePath = DatabaseGenerationEngine.GetTemplatePathFromArtifact(Wizard.ModelBuilderSettings.Artifact);
var databaseSchemaName = DatabaseGenerationEngine.GetDatabaseSchemaNameFromArtifact(Wizard.ModelBuilderSettings.Artifact);
// Save off the SynchronizationContext so we can post methods to the UI event queue when
// responding to workflow events (since they are executed in a separate thread)
_synchronizationContext = SynchronizationContext.Current;
// Invoke the Pipeline/Workflow. The Workflow engine will automatically wrap this in a background thread
try
{
using (new VsUtils.HourglassHelper())
{
var resolvedWorkflowFileInfo = DatabaseGenerationEngine.ResolveAndValidateWorkflowPath(
Wizard.Project,
workflowPath);
var resolvedDefaultPath = VsUtils.ResolvePathWithMacro(
null, DatabaseGenerationEngine.DefaultWorkflowPath,
new Dictionary <string, string>
{
{ ExtensibleFileManager.EFTOOLS_USER_MACRONAME, ExtensibleFileManager.UserEFToolsDir.FullName },
{ ExtensibleFileManager.EFTOOLS_VS_MACRONAME, ExtensibleFileManager.VSEFToolsDir.FullName }
});
// Display a security warning if the workflow path specified is different from the default
if (!resolvedWorkflowFileInfo.FullName.Equals(
Path.GetFullPath(resolvedDefaultPath), StringComparison.OrdinalIgnoreCase))
{
var displayCustomWorkflowWarning = true;
try
{
var customWorkflowWarningString = EdmUtils.GetUserSetting(RegKeyNameCustomWorkflowWarning);
if (false == String.IsNullOrEmpty(customWorkflowWarningString) &&
false == Boolean.TryParse(customWorkflowWarningString, out displayCustomWorkflowWarning))
{
displayCustomWorkflowWarning = true;
}
if (displayCustomWorkflowWarning)
{
var cancelledDuringCustomWorkflowWarning = DismissableWarningDialog
.ShowWarningDialogAndSaveDismissOption(
Resources.DatabaseCreation_CustomWorkflowWarningTitle,
Resources.DatabaseCreation_WarningCustomWorkflow,
RegKeyNameCustomWorkflowWarning,
DismissableWarningDialog.ButtonMode.OkCancel);
if (cancelledDuringCustomWorkflowWarning)
{
HandleError(
String.Format(
CultureInfo.CurrentCulture, Resources.DatabaseCreation_CustomWorkflowCancelled,
resolvedWorkflowFileInfo.FullName), false);
return;
}
}
}
catch (SecurityException e)
{
// We should at least alert the user of why this is failing so they can take steps to fix it.
VsUtils.ShowMessageBox(
Services.ServiceProvider,
String.Format(
CultureInfo.CurrentCulture, Resources.ErrorReadingWritingUserSetting,
RegKeyNameCustomWorkflowWarning, e.Message),
OLEMSGBUTTON.OLEMSGBUTTON_OK,
OLEMSGDEFBUTTON.OLEMSGDEFBUTTON_FIRST,
OLEMSGICON.OLEMSGICON_WARNING);
}
}
_workflowInstance = DatabaseGenerationEngine.CreateDatabaseScriptGenerationWorkflow(
_synchronizationContext,
Wizard.Project,
Wizard.ModelBuilderSettings.Artifact.Uri.LocalPath,
resolvedWorkflowFileInfo,
templatePath,
edm,
existingSsdl,
existingMsl,
databaseSchemaName,
Wizard.ModelBuilderSettings.InitialCatalog,
Wizard.ModelBuilderSettings.RuntimeProviderInvariantName,
Wizard.ModelBuilderSettings.AppConfigConnectionString,
Wizard.ModelBuilderSettings.ProviderManifestToken,
Wizard.ModelBuilderSettings.Artifact.SchemaVersion,
_workflowInstance_WorkflowCompleted,
_workflowInstance_UnhandledException);
}
Wizard.ModelBuilderSettings.WorkflowInstance = _workflowInstance;
_workflowInstance.Run();
}
catch (Exception e)
{
HandleError(e.Message, true);
if (_workflowInstance != null)
{
CleanupWorkflow();
}
}
}
}
