Added endpoint for fetching all notebook jobs

src/Microsoft.SqlTools.ServiceLayer/Agent/AgentService.cs

@@ -8,6 +8,7 @@
 using System.Data;
 using System.IO;
 using System.Threading.Tasks;
+using System.Data.SqlClient;
 using System.Xml;
 using Microsoft.SqlServer.Management.Common;
 using Microsoft.SqlServer.Management.Smo;
@@ -122,6 +123,8 @@ public void InitializeService(ServiceHost serviceHost)
             this.ServiceHost.SetRequestHandler(UpdateAgentScheduleRequest.Type, HandleUpdateAgentScheduleRequest);
             this.ServiceHost.SetRequestHandler(DeleteAgentScheduleRequest.Type, HandleDeleteAgentScheduleRequest);
 
+            // Notebook request handlers
+            this.ServiceHost.SetRequestHandler(AgentNotebooksRequest.Type, HandleAgentNotebooksRequest);
         }
 
         #region "Jobs Handlers"
@@ -1211,6 +1214,88 @@ private Tuple<SqlConnectionInfo, DataTable, ServerConnection> CreateSqlConnectio
             return new Tuple<SqlConnectionInfo, DataTable, ServerConnection>(sqlConnInfo, dt, serverConnection);
         }
 
+        internal async Task HandleAgentNotebooksRequest(AgentNotebooksParams parameters, RequestContext<AgentNotebooksResult> requestContext)
+        {
+            await Task.Run(async () =>
+            {
+                try
+                {
+                    var result = new AgentNotebooksResult();
+                    ConnectionInfo connInfo;
+                    ConnectionServiceInstance.TryFindConnection(
+                        parameters.OwnerUri,
+                        out connInfo);
+
+                    if (connInfo != null)
+                    {
+                        using (SqlConnection connection = new SqlConnection(ConnectionService.BuildConnectionString(connInfo.ConnectionDetails)))
+                        {
+                            connection.Open();
+                            /*
+                            This query fetches all the notebook Jobs across all the databases accessible by the user.
+                             */
+                            string getJobIdsFromDatabaseQueryString = 
+                                "DECLARE @script as varchar(MAX)\n" +
+                                "SET @script =\n" + 
+                                "'USE [?];\n" +
+                                "IF EXISTS\n" + 
+                                "(SELECT * FROM INFORMATION_SCHEMA.TABLES WHERE TABLE_NAME = N''nb_template'')\n"+
+                                "BEGIN\n" + 
+                                "select dbo.nb_template.*,\n" +
+                                "DB_NAME() as db_name,\n" +
+                                "msdb.dbo.sysjobs.*\n" +
+                                "from [?].dbo.nb_template\n" +
+                                "LEFT JOIN\n" + 
+                                "msdb.dbo.sysjobs \n" +
+                                "ON \n" + 
+                                "[?].dbo.nb_template.job_id = msdb.dbo.sysjobs.job_id\n" +
+                                "END '\n" +  
+                                "EXEC sp_MSforeachdb @script\n";
+                            var agentnotebooks = new List<NotebookJobInfo>();
+                            using( SqlCommand getJobIdsFromDatabaseQueryCommand = new SqlCommand(getJobIdsFromDatabaseQueryString, connection))
+                            {
+                                SqlDataAdapter jobIdsAdapter = new SqlDataAdapter(getJobIdsFromDatabaseQueryCommand);
+                                DataSet jobIdsDataSet = new DataSet();
+                                jobIdsAdapter.Fill(jobIdsDataSet);
+                                var serverConnection = ConnectionService.OpenServerConnection(connInfo);
+                                var fetcher = new JobFetcher(serverConnection);
+                                var filter = new JobActivityFilter();
+                                // Fetching all the jobs
+                                var jobs = fetcher.FetchJobs(filter);
+                                // Creating Dictionary of jobs for effcient access later on.
+                                Dictionary<Guid, JobProperties> allJobsHashTable = new Dictionary<Guid, JobProperties>();
+                                if(jobs != null)
+                                {
+                                    foreach (var job in jobs.Values)
+                                    {
+                                        allJobsHashTable.Add(job.JobID, job);
+                                    }
+                                }                                
+                                foreach (DataTable templateTable in jobIdsDataSet.Tables)
+                                {
+                                    foreach (DataRow templateRow in templateTable.Rows){
+                                        NotebookJobInfo notebookJob = new NotebookJobInfo();
+                                        notebookJob.Template = (string)templateRow["json"];
+                                        notebookJob.TargetDatabase = (string)templateRow["db_name"];
+                                        // Setting Job using the already fetched Jobs
+                                        notebookJob.Job = AgentUtilities.ConvertToAgentJobInfo(allJobsHashTable[(Guid)templateRow["job_id"]]);
+                                        agentnotebooks.Add(notebookJob);
+                                    }
+                                }
+                            }
+                            result.Success = true;
+                            result.Notebooks = agentnotebooks.ToArray();
+                        }
+                    }
+                    await requestContext.SendResult(result);
+                }
+                catch (Exception e)
+                {
+                    await requestContext.SendError(e);
+                }
+            });
+        }
+
         #endregion // "Helpers"
     }
 }

src/Microsoft.SqlTools.ServiceLayer/Agent/Contracts/AgentNotebookRequest.cs

@@ -0,0 +1,51 @@
+//
+// Copyright (c) Microsoft. All rights reserved.
+// Licensed under the MIT license. See LICENSE file in the project root for full license information.
+//
+
+using Microsoft.SqlTools.Hosting.Protocol.Contracts;
+using Microsoft.SqlTools.ServiceLayer.TaskServices;
+using Microsoft.SqlTools.ServiceLayer.Utility;
+using Microsoft.SqlTools.Utility;
+
+namespace Microsoft.SqlTools.ServiceLayer.Agent.Contracts
+{
+    /// <summary>
+    /// SQL Agent Notebooks activity parameters
+    /// </summary>
+    public class AgentNotebooksParams : GeneralRequestDetails
+    {
+        public string OwnerUri {get; set;}
+
+    }
+    /// <summary>
+    /// SQL Agent Notebook activity result
+    /// </summary>
+    public class AgentNotebooksResult : ResultStatus
+    {
+        public NotebookJobInfo[] Notebooks {get; set;}
+    }
+    /// <summary>
+    /// SQL Agent Notebook request type
+    /// </summary>
+    public class AgentNotebooksRequest
+    {
+        /// <summary>
+        /// Request definition
+        /// </summary>
+        public static readonly
+        RequestType<AgentNotebooksParams, AgentNotebooksResult> Type = 
+        RequestType<AgentNotebooksParams, AgentNotebooksResult>.Create("agent/notebooks");
+    }
+    /// <summary>
+    /// SQL Agent Notebook Job Info Class
+    /// </summary>
+    public class NotebookJobInfo
+    {
+        public AgentJobInfo Job {get; set;}
+        public string Template {get; set;}
+        public string TargetDatabase {get; set;}
+    }
+
+
+}