sqltoolsservice/src/Microsoft.SqlTools.ServiceLayer/Agent/Jobs/JobStepsActions.cs

//
// Copyright (c) Microsoft. All rights reserved.
// Licensed under the MIT license. See LICENSE file in the project root for full license information.
//

using System;
using Microsoft.SqlServer.Management.Smo.Agent;
using Microsoft.SqlTools.ServiceLayer.Agent.Contracts;
using Microsoft.SqlTools.ServiceLayer.Management;

namespace Microsoft.SqlTools.ServiceLayer.Agent
{
    /// <summary>
    /// JobStepsActions
    /// </summary>
    internal class JobStepsActions : ManagementActionBase
    {
        private JobStepData data;
        private JobData jobData;
        private ConfigAction configAction;

        public JobStepsActions(
            CDataContainer dataContainer, 
            JobData jobData,
            AgentJobStepInfo stepInfo,
            ConfigAction configAction)
        {
            this.configAction = configAction;
            this.DataContainer = dataContainer;
            this.jobData = jobData;

            if (configAction == ConfigAction.Create)
            {
                this.data = new JobStepData(jobData.JobSteps);
            }
            else
            {
                JobStep jobStep = GetJobStep(this.jobData, stepInfo.StepName);
                this.data = new JobStepData(jobStep, jobData.JobSteps);
            }

            // load properties from AgentJobStepInfo
            this.data.ID = stepInfo.Id;
            this.data.Name = stepInfo.StepName;
            this.data.Command = stepInfo.Script;
        }

        protected override bool DoPreProcessExecution(RunType runType, out ExecutionMode executionResult)
        {
            base.DoPreProcessExecution(runType, out executionResult);
            if (this.configAction == ConfigAction.Drop)
            {
                if (this.data.JobStep != null)
                {
                    this.data.JobStep.DropIfExists();                    
                }
            }
            else
            {
                // Make sure the job step name is not blank.
                if (this.data.Name == null || this.data.Name.Length == 0)
                {
                    throw new Exception(SR.JobStepNameCannotBeBlank);
                }

                // Check to make sure that the user has not entered a job step name that already exists.
                for (int stepIndex = 0; stepIndex < this.data.Parent.Steps.Count; ++stepIndex)
                {
                    // don't compare if the id's are the same.
                    if (data.ID != ((JobStepData)this.data.Parent.Steps[stepIndex]).ID && data.Name == ((JobStepData)this.data.Parent.Steps[stepIndex]).Name)
                    {
                        // Throw an error if the job step name already exists
                        throw new Exception(SR.JobStepNameAlreadyExists(this.data.Name));
                    }
                }

                if (runType == RunType.RunNow)
                {                   
                    this.data.ApplyChanges(this.jobData.Job);
                }
            }

            // regular execution always takes place
            return true;
        }

        private JobStep GetJobStep(JobData jobData, string stepName)
        {
            JobStep jobStep = null;
            if (jobData.Job != null)
            {
                const string UrnFormatStr = "Server[@Name='{0}']/JobServer[@Name='{0}']/Job[@Name='{1}']/Step[@Name='{2}']";
                string serverName = this.DataContainer.Server.Name.ToUpper();
                string urn = string.Format(UrnFormatStr, serverName, jobData.Job.Name, stepName);
                jobStep = jobData.Job.Parent.Parent.GetSmoObject(urn) as JobStep;
            }
            return jobStep;
        }
    }
}