public override async Task <ActionResponse> ExecuteActionAsync(ActionRequest request)
{
var taskName = request.DataStore.GetValue("TaskName");
using (TaskService ts = new TaskService())
{
TaskCollection tasks = ts.RootFolder.GetTasks(new Regex(taskName));
try
{
if (tasks.Exists(taskName))
{
foreach (Task task in tasks)
{
if (task.Name.EqualsIgnoreCase(taskName))
{
if (task.State == TaskState.Running || task.State == TaskState.Queued)
{
task.Stop();
NTHelper.KillProcess("azurebcp");
}
ts.RootFolder.DeleteTask(taskName, false);
}
}
}
}
catch (Exception e)
{
return(new ActionResponse(ActionStatus.Failure, e));
}
return(new ActionResponse(ActionStatus.Success));
}
}
private static void DirectoryCopy(string sourceDirName, string destDirName, bool copySubDirs)
{
// Get the subdirectories for the specified directory.
DirectoryInfo dir = new DirectoryInfo(sourceDirName);
if (!dir.Exists)
{
throw new DirectoryNotFoundException("Source directory does not exist or could not be found: " + sourceDirName);
}
DirectoryInfo[] dirs = dir.GetDirectories();
// If the destination directory doesn't exist, create it.
if (!Directory.Exists(destDirName))
{
Directory.CreateDirectory(destDirName);
}
// Kill any running azurebcp
NTHelper.KillProcess("azurebcp");
// Get the files in the directory and copy them to the new location.
FileInfo[] files = dir.GetFiles();
Parallel.ForEach(files, (currentFile) =>
{
string temppath = Path.Combine(destDirName, currentFile.Name);
currentFile.CopyTo(temppath, true);
FileInfo destination = new FileInfo(temppath);
try
{
if (destination.AlternateDataStreamExists("Zone.Identifier"))
{
destination.DeleteAlternateDataStream("Zone.Identifier");
}
}
catch { }
});
// If copying subdirectories, copy them and their contents to new location.
if (copySubDirs)
{
foreach (DirectoryInfo subdir in dirs)
{
string temppath = Path.Combine(destDirName, subdir.Name);
DirectoryCopy(subdir.FullName, temppath, copySubDirs);
}
}
}
public override async Task <ActionResponse> ExecuteActionAsync(ActionRequest request)
{
var taskDescription = request.DataStore.GetValue("TaskDescription");
var taskFile = request.DataStore.GetValue("TaskFile");
var taskName = request.DataStore.GetValue("TaskName");
var taskParameters = request.DataStore.GetValue("TaskParameters");
var taskProgram = request.DataStore.GetValue("TaskProgram");
var taskStartTime = request.DataStore.GetValue("TaskStartTime");
var taskUsername = string.IsNullOrEmpty(request.DataStore.GetValue("ImpersonationUsername"))
? WindowsIdentity.GetCurrent().Name
: NTHelper.CleanDomain(request.DataStore.GetValue("ImpersonationDomain")) + "\\" + NTHelper.CleanUsername(request.DataStore.GetValue("ImpersonationUsername"));
var taskPassword = string.IsNullOrEmpty(request.DataStore.GetValue("ImpersonationPassword"))
? null
: request.DataStore.GetValue("ImpersonationPassword");
if (taskPassword == null)
{
return(new ActionResponse(ActionStatus.Failure, JsonUtility.GetEmptyJObject(), "CreateTaskPasswordMissing"));
}
string workingDirectory = request.DataStore.GetValue("TaskDirectory") == null
? FileUtility.GetLocalTemplatePath(request.Info.AppName)
: FileUtility.GetLocalPath(request.DataStore.GetValue("TaskDirectory"));
bool isPowerShell = taskProgram.EqualsIgnoreCase("powershell");
using (TaskService ts = new TaskService())
{
TaskCollection tasks = ts.RootFolder.GetTasks(new Regex(taskName));
foreach (Task task in tasks)
{
if (task.Name.EqualsIgnoreCase(taskName))
{
if (task.State == TaskState.Running || task.State == TaskState.Queued)
{
task.Stop();
NTHelper.KillProcess("azurebcp");
}
ts.RootFolder.DeleteTask(taskName);
}
}
TaskDefinition td = ts.NewTask();
td.RegistrationInfo.Description = taskDescription;
td.Settings.Compatibility = TaskCompatibility.V2_1;
td.RegistrationInfo.Author = taskUsername;
td.Principal.RunLevel = TaskRunLevel.LUA;
td.Settings.StartWhenAvailable = true;
td.Settings.RestartCount = 3;
td.Settings.RestartInterval = TimeSpan.FromMinutes(3);
td.Settings.MultipleInstances = TaskInstancesPolicy.IgnoreNew;
td.Triggers.Add(new DailyTrigger
{
DaysInterval = 1,
StartBoundary = DateTime.Parse(taskStartTime)
});
string optionalArguments = string.Empty;
if (isPowerShell)
{
optionalArguments = Path.Combine(workingDirectory, taskFile);
}
else
{
taskProgram = taskFile;
}
if (isPowerShell)
{
optionalArguments = $"-ExecutionPolicy Bypass -File \"{optionalArguments}\"";
}
if (!string.IsNullOrEmpty(taskParameters))
{
optionalArguments += " " + taskParameters;
}
td.Actions.Add(new ExecAction(taskProgram, optionalArguments, workingDirectory));
ts.RootFolder.RegisterTaskDefinition(taskName, td, TaskCreation.CreateOrUpdate, taskUsername, taskPassword, TaskLogonType.Password, null);
}
return(new ActionResponse(ActionStatus.Success));
}