extended stored procedures

Author: csbernath

DataCommander/Providers/DataCommander.Providers.SqlServer/DataCommander.Providers.SqlServer.csproj

@@ -72,6 +72,8 @@
     <Prefer32Bit>false</Prefer32Bit>
   </PropertyGroup>
   <ItemGroup>
+    <Compile Include="ObjectExplorer\ExtendedStoreProcedureCollectionNode.cs" />
+    <Compile Include="ObjectExplorer\ExtendedStoreProcedureNode.cs" />
     <Compile Include="ObjectExplorer\SqlServerSystemType.cs" />
     <Compile Include="SqlServerConnectionStringBuilder.cs" />
     <Compile Include="DatabaseObjectMultipartName.cs" />

DataCommander/Providers/DataCommander.Providers.SqlServer/ObjectExplorer/ExtendedStoreProcedureCollectionNode.cs

@@ -0,0 +1,41 @@
+using System.Collections.Generic;
+using System.Windows.Forms;
+using Foundation.Data;
+using Foundation.Data.SqlClient;
+
+namespace DataCommander.Providers.SqlServer.ObjectExplorer
+{
+    internal sealed class ExtendedStoreProcedureCollectionNode : ITreeNode
+    {
+        private readonly DatabaseNode _database;
+
+        public ExtendedStoreProcedureCollectionNode(DatabaseNode database) => _database = database;
+
+        string ITreeNode.Name => "Extended Stored Procedures";
+        bool ITreeNode.IsLeaf => false;
+        bool ITreeNode.Sortable => false;
+        string ITreeNode.Query => null;
+        ContextMenuStrip ITreeNode.ContextMenu => null;
+
+        IEnumerable<ITreeNode> ITreeNode.GetChildren(bool refresh)
+        {
+            var executor = new SqlCommandExecutor(_database.Databases.Server.ConnectionString);
+            var commandText = @"select
+    s.name,
+    o.name
+from sys.all_objects o
+join sys.schemas s
+    on o.schema_id = s.schema_id
+where o.type = 'X'
+order by 1,2";
+            var request = new ExecuteReaderRequest(commandText);
+            var childNodes = executor.ExecuteReader(request, dataRecord =>
+            {
+                var schema = dataRecord.GetString(0);
+                var name = dataRecord.GetString(1);
+                return new ExtendedStoreProcedureNode(_database, schema, name);
+            });
+            return childNodes;
+        }
+    }
+}

DataCommander/Providers/DataCommander.Providers.SqlServer/ObjectExplorer/ExtendedStoreProcedureNode.cs

@@ -0,0 +1,27 @@
+using System;
+using System.Collections.Generic;
+using System.Windows.Forms;
+
+namespace DataCommander.Providers.SqlServer.ObjectExplorer
+{
+    internal sealed class ExtendedStoreProcedureNode : ITreeNode
+    {
+        private readonly DatabaseNode _database;
+        private readonly string _schema;
+        private readonly string _name;
+
+        public ExtendedStoreProcedureNode(DatabaseNode database, string schema, string name)
+        {
+            _database = database;
+            _schema = schema;
+            _name = name;
+        }
+
+        string ITreeNode.Name => $"{_schema}.{_name}";
+        bool ITreeNode.IsLeaf => true;
+        bool ITreeNode.Sortable => false;
+        string ITreeNode.Query => null;
+        ContextMenuStrip ITreeNode.ContextMenu => null;
+        IEnumerable<ITreeNode> ITreeNode.GetChildren(bool refresh) => throw new NotSupportedException();
+    }
+}

DataCommander/Providers/DataCommander.Providers.SqlServer/ObjectExplorer/ProgrammabilityNode.cs

@@ -14,12 +14,16 @@ internal sealed class ProgrammabilityNode : ITreeNode
 
         IEnumerable<ITreeNode> ITreeNode.GetChildren(bool refresh)
         {
-            return new ITreeNode[]
-            {
-                new StoredProcedureCollectionNode(_database, false),
-                new FunctionCollectionNode(_database),
-                new UserDefinedTableTypeCollectionNode(_database)
-            };
+            var childNodes = new List<ITreeNode>();
+            childNodes.Add(new StoredProcedureCollectionNode(_database, false));
+            childNodes.Add(new FunctionCollectionNode(_database));
+
+            if (_database.Name == "master")
+                childNodes.Add(new ExtendedStoreProcedureCollectionNode(_database));
+
+            childNodes.Add(new UserDefinedTableTypeCollectionNode(_database));
+
+            return childNodes;
         }
 
         bool ITreeNode.Sortable => false;