/// <summary>
/// Returns if the asset exists locally or in the cloud
/// </summary>
/// <param name="assetName"></param>
/// <param name="assetType"></param>
/// <param name="localWorkspacePath"></param>
/// <param name="automationApi"></param>
/// <param name="resourceGroupName"></param>
/// <param name="automationAccountName"></param>
/// <param name="encryptionCertThumbprint"></param>
/// <returns>null if the asset does not exist or else returns the asset</returns>
public static async Task <AutomationAsset> GetAsset(String assetName, String assetType, String localWorkspacePath, AutomationManagementClient automationApi, string resourceGroupName, string automationAccountName, string encryptionCertThumbprint, ICollection <ConnectionType> connectionTypes)
{
AutomationAsset automationAsset = null;
// Get local assets
LocalAssets localAssets = LocalAssetsStore.Get(localWorkspacePath, encryptionCertThumbprint, connectionTypes);
// Search for variables
CancellationTokenSource cts = new CancellationTokenSource();
cts.CancelAfter(TIMEOUT_MS);
if (assetType == Constants.AssetType.Variable)
{
// Check local asset store first
var localVariable = localAssets.Variables.Find(asset => asset.Name == assetName);
if (localVariable != null)
{
automationAsset = new AutomationVariable(localVariable);
}
else
{
try
{
// Check cloud. Catch exception if it doesn't exist
VariableGetResponse cloudVariable = await automationApi.Variables.GetAsync(resourceGroupName, automationAccountName, assetName, cts.Token);
automationAsset = new AutomationVariable(cloudVariable.Variable);
}
catch (Exception e)
{
// If the exception is not found, don't throw new exception as this is expected
if (e.HResult != -2146233088)
{
throw e;
}
}
}
}
// Search for credentials
else if (assetType == Constants.AssetType.Credential)
{
// Check local asset store first
var localCredential = localAssets.PSCredentials.Find(asset => asset.Name == assetName);
if (localCredential != null)
{
automationAsset = new AutomationCredential(localCredential);
}
else
{
try
{
// Check cloud. Catch execption if it doesn't exist
CredentialGetResponse cloudVariable = await automationApi.PsCredentials.GetAsync(resourceGroupName, automationAccountName, assetName, cts.Token);
automationAsset = new AutomationCredential(cloudVariable.Credential);
}
catch (Exception e)
{
// If the exception is not found, don't throw new exception as this is expected
if (e.HResult != -2146233088)
{
throw e;
}
}
}
}
// Search for connections
else if (assetType == Constants.AssetType.Connection)
{
// Check local asset store first
var localConnection = localAssets.Connections.Find(asset => asset.Name == assetName);
if (localConnection != null)
{
automationAsset = new AutomationConnection(localConnection);
}
else
{
try
{
// Check cloud. Catch exception if it doesn't exist
ConnectionGetResponse cloudConnection = await automationApi.Connections.GetAsync(resourceGroupName, automationAccountName, assetName, cts.Token);
cts = new CancellationTokenSource();
cts.CancelAfter(TIMEOUT_MS);
ConnectionTypeGetResponse connectionType = await automationApi.ConnectionTypes.GetAsync(resourceGroupName, automationAccountName,
cloudConnection.Connection.Properties.ConnectionType.Name, cts.Token);
automationAsset = new AutomationConnection(cloudConnection.Connection, connectionType.ConnectionType);
}
catch (Exception e)
{
// If the exception is not found, don't throw new exception as this is expected
if (e.HResult != -2146233088)
{
throw e;
}
}
}
}
// Search for certificates
else if (assetType == Constants.AssetType.Certificate)
{
// Check local asset store first
var localCertificate = localAssets.Certificate.Find(asset => asset.Name == assetName);
if (localCertificate != null)
{
automationAsset = new AutomationCertificate(localCertificate);
}
else
{
try
{
// Check cloud. Catch execption if it doesn't exist
CertificateGetResponse cloudCertificate = await automationApi.Certificates.GetAsync(resourceGroupName, automationAccountName, assetName, cts.Token);
automationAsset = new AutomationCertificate(cloudCertificate.Certificate);
}
catch (Exception e)
{
// If the exception is not found, don't throw new exception as this is expected
if (e.HResult != -2146233088)
{
throw e;
}
}
}
}
return(automationAsset);
}
public static async Task <ISet <AutomationAsset> > GetAll(String localWorkspacePath, AutomationManagementClient automationApi, string resourceGroupName, string automationAccountName, string encryptionCertThumbprint, ICollection <ConnectionType> connectionTypes)
{
CancellationTokenSource cts = new CancellationTokenSource();
cts.CancelAfter(TIMEOUT_MS);
VariableListResponse cloudVariables = await automationApi.Variables.ListAsync(resourceGroupName, automationAccountName, cts.Token);
cts = new CancellationTokenSource();
cts.CancelAfter(TIMEOUT_MS);
CredentialListResponse cloudCredentials = await automationApi.PsCredentials.ListAsync(resourceGroupName, automationAccountName, cts.Token);
cts = new CancellationTokenSource();
cts.CancelAfter(TIMEOUT_MS);
ConnectionListResponse cloudConnections = await automationApi.Connections.ListAsync(resourceGroupName, automationAccountName, cts.Token);
CertificateListResponse cloudCertificates = await automationApi.Certificates.ListAsync(resourceGroupName, automationAccountName, cts.Token);
// need to get connections one at a time to get each connection's values. Values currently come back as empty in list call
var connectionAssetsWithValues = new HashSet <Connection>();
foreach (var connection in cloudConnections.Connection)
{
cts = new CancellationTokenSource();
cts.CancelAfter(TIMEOUT_MS);
var connectionResponse = await automationApi.Connections.GetAsync(resourceGroupName, automationAccountName, connection.Name, cts.Token);
connectionAssetsWithValues.Add(connectionResponse.Connection);
}
LocalAssets localAssets = LocalAssetsStore.Get(localWorkspacePath, encryptionCertThumbprint, connectionTypes);
var automationAssets = new SortedSet <AutomationAsset>();
// Compare cloud variables to local
foreach (var cloudAsset in cloudVariables.Variables)
{
var localAsset = localAssets.Variables.Find(asset => asset.Name == cloudAsset.Name);
var automationAsset = (localAsset != null) ?
new AutomationVariable(localAsset, cloudAsset) :
new AutomationVariable(cloudAsset);
automationAssets.Add(automationAsset);
}
// Add remaining locally created variables
foreach (var localAsset in localAssets.Variables)
{
var automationAsset = new AutomationVariable(localAsset);
automationAssets.Add(automationAsset);
}
// Compare cloud credentials to local
foreach (var cloudAsset in cloudCredentials.Credentials)
{
var localAsset = localAssets.PSCredentials.Find(asset => asset.Name == cloudAsset.Name);
var automationAsset = (localAsset != null) ?
new AutomationCredential(localAsset, cloudAsset) :
new AutomationCredential(cloudAsset);
automationAssets.Add(automationAsset);
}
// Add remaining locally created credentials
foreach (var localAsset in localAssets.PSCredentials)
{
var automationAsset = new AutomationCredential(localAsset);
automationAssets.Add(automationAsset);
}
// Compare cloud connections to local
foreach (var cloudAsset in connectionAssetsWithValues)
{
ConnectionTypeGetResponse connectionType = await automationApi.ConnectionTypes.GetAsync(resourceGroupName, automationAccountName, cloudAsset.Properties.ConnectionType.Name);
var localAsset = localAssets.Connections.Find(asset => asset.Name == cloudAsset.Name);
var automationAsset = (localAsset != null) ?
new AutomationConnection(localAsset, cloudAsset) :
new AutomationConnection(cloudAsset, connectionType.ConnectionType);
automationAssets.Add(automationAsset);
}
// Add remaining locally created connections
foreach (var localAsset in localAssets.Connections)
{
var automationAsset = new AutomationConnection(localAsset);
automationAssets.Add(automationAsset);
}
// Compare cloud certificates to local
foreach (var cloudAsset in cloudCertificates.Certificates)
{
var localAsset = localAssets.Certificate.Find(asset => asset.Name == cloudAsset.Name);
var automationAsset = (localAsset != null) ?
new AutomationCertificate(localAsset, cloudAsset) :
new AutomationCertificate(cloudAsset);
automationAssets.Add(automationAsset);
}
// Add remaining locally created certificates
foreach (var localAsset in localAssets.Certificate)
{
var automationAsset = new AutomationCertificate(localAsset);
automationAssets.Add(automationAsset);
}
return(automationAssets);
}
