From 69786fa46fa18090e7641e086cd2aed70a530748 Mon Sep 17 00:00:00 2001
From: Andy Robbins <arobbins@specterops.io>
Date: Tue, 18 Apr 2023 08:10:35 -0700
Subject: [PATCH] feat: Add new edges to edge filter

---
 .../SearchContainer/EdgeFilter/EdgeFilter.jsx | 28 +++++++++++++++++++
 .../EdgeFilter/EdgeFilter.module.css          |  2 +-
 src/index.js                                  | 10 +++++++
 3 files changed, 39 insertions(+), 1 deletion(-)

diff --git a/src/components/SearchContainer/EdgeFilter/EdgeFilter.jsx b/src/components/SearchContainer/EdgeFilter/EdgeFilter.jsx
index 16d4b7dd3..90a22af3f 100644
--- a/src/components/SearchContainer/EdgeFilter/EdgeFilter.jsx
+++ b/src/components/SearchContainer/EdgeFilter/EdgeFilter.jsx
@@ -144,6 +144,11 @@ const EdgeFilter = ({ open }) => {
                             'AZAddOwner',
                             'AZManagedIdentity',
                             'AZKeyVaultContributor',
+                            'AZNodeResourceGroup',
+                            'AZWebsiteContributor',
+                            'AZLogicAppContributor',
+                            'AZAutomationContributor',
+                            'AZAKSContributor',
                         ]}
                         sectionName='azure'
                     />
@@ -176,6 +181,29 @@ const EdgeFilter = ({ open }) => {
                     <EdgeFilterCheck name='AZAddOwner' />
                     <EdgeFilterCheck name='AZManagedIdentity' />
                     <EdgeFilterCheck name='AZKeyVaultContributor' />
+                    <EdgeFilterCheck name='AZNodeResourceGroup' />
+                    <EdgeFilterCheck name='AZWebsiteContributor' />
+                    <EdgeFilterCheck name='AZLogicAppContributor' />
+                    <EdgeFilterCheck name='AZAutomationContributor' />
+                    <EdgeFilterCheck name='AZAKSContributor' />
+                </div>
+                <div>
+                    <EdgeFilterSection
+                        title='MS Graph App Roles'
+                        edges={[
+                            'AZMGAddSecret',
+                            'AZMGAddOwner',
+                            'AZMGAddMember',
+                            'AZMGGrantAppRoles',
+                            'AZMGGrantRole',
+                        ]}
+                        sectionName='msgraphapproles'
+                    />
+                    <EdgeFilterCheck name='AZMGAddSecret' />
+                    <EdgeFilterCheck name='AZMGAddOwner' />
+                    <EdgeFilterCheck name='AZMGAddMember' />
+                    <EdgeFilterCheck name='AZMGGrantAppRoles' />
+                    <EdgeFilterCheck name='AZMGGrantRole' />
                 </div>
             </div>
         </motion.div>
diff --git a/src/components/SearchContainer/EdgeFilter/EdgeFilter.module.css b/src/components/SearchContainer/EdgeFilter/EdgeFilter.module.css
index d988e8c5f..b46118a70 100644
--- a/src/components/SearchContainer/EdgeFilter/EdgeFilter.module.css
+++ b/src/components/SearchContainer/EdgeFilter/EdgeFilter.module.css
@@ -35,7 +35,7 @@
 
 .container {
     display: grid;
-    grid-template-columns: 1fr 1fr;
+    grid-template-columns: 1fr 1fr 1fr;
     margin-left: 8px;
     margin-right: 8px;
     white-space: nowrap;
diff --git a/src/index.js b/src/index.js
index f6afa725d..69abeffc0 100644
--- a/src/index.js
+++ b/src/index.js
@@ -447,6 +447,16 @@ if (typeof conf.get('edgeincluded') === 'undefined') {
         HasSIDHistory: true,
         CanPSRemote: true,
         SyncLAPSPassword: true,
+        AZMGGrantRole: true,
+        AZMGAddSecret: true,
+        AZMGAddOwner: true,
+        AZMGAddMember: true,
+        AZMGGrantAppRoles: true,
+        AZNodeResourceGroup: true,
+        AZWebsiteContributor: true,
+        AZLogicAppContributo: true,
+        AZAutomationContributor: true,
+        AZAKSContributor: true,
     });
 }