remove team button support

client/web/src/repo/blob/own/FileOwnershipEntry.tsx

@@ -26,7 +26,6 @@ interface Props {
     makeOwnerButton?: React.ReactElement
     repoID: string
     filePath: string
-    userID?: string
     setRemoveOwnerError: any
     refetch: any
 }
@@ -43,7 +42,6 @@ export const FileOwnershipEntry: React.FunctionComponent<Props> = ({
     makeOwnerButton,
     repoID,
     filePath,
-    userID,
     refetch,
     setRemoveOwnerError,
 }) => {
@@ -57,6 +55,9 @@ export const FileOwnershipEntry: React.FunctionComponent<Props> = ({
         return owner.email
     }
 
+    const userID = owner.__typename === 'Person' && owner.user?.__typename === 'User' ? owner.user.id : undefined
+    const teamID = owner.__typename === 'Team' ? owner.id : undefined
+
     const email = findEmail()
 
     const assignedOwnerReasons: AssignedOwnerFields[] = reasons
@@ -119,7 +120,8 @@ export const FileOwnershipEntry: React.FunctionComponent<Props> = ({
                                 onError={setRemoveOwnerError}
                                 repoId={repoID}
                                 path={filePath}
-                                userId={userID}
+                                userID={userID}
+                                teamID={teamID}
                                 isDirectAssigned={isDirectAssigned}
                             />
                         ))}

client/web/src/repo/blob/own/OwnerList.tsx

@@ -149,6 +149,8 @@ export const OwnerList: React.FunctionComponent<OwnerListProps> = ({
                                 )
                             )
                             .map((ownership, index) => {
+                                if (ownership.owner.__typename === 'Team') {
+                                }
                                 const userId =
                                     ownership.owner.__typename === 'Person' &&
                                     ownership.owner.user?.__typename === 'User'
@@ -162,7 +164,6 @@ export const OwnerList: React.FunctionComponent<OwnerListProps> = ({
                                         <FileOwnershipEntry
                                             refetch={refetch}
                                             owner={ownership.owner}
-                                            userID={userId}
                                             repoID={repoID}
                                             filePath={filePath}
                                             reasons={ownership.reasons}
@@ -211,7 +212,6 @@ export const OwnerList: React.FunctionComponent<OwnerListProps> = ({
                                             owner={ownership.owner}
                                             reasons={ownership.reasons}
                                             makeOwnerButton={makeOwnerButton?.(userId)}
-                                            userID={userId}
                                             repoID={repoID}
                                             filePath={filePath}
                                             refetch={refetch}

client/web/src/repo/blob/own/RemoveOwnerButton.tsx

@@ -6,16 +6,22 @@ import { ErrorLike, asError } from '@sourcegraph/common'
 import { useMutation } from '@sourcegraph/http-client'
 import { Button, Icon, Tooltip } from '@sourcegraph/wildcard'
 
-import { RemoveAssignedOwnerResult, RemoveAssignedOwnerVariables } from '../../../graphql-operations'
+import {
+    RemoveAssignedOwnerResult,
+    RemoveAssignedOwnerVariables,
+    RemoveAssignedTeamResult,
+    RemoveAssignedTeamVariables,
+} from '../../../graphql-operations'
 
-import { REMOVE_ASSIGNED_OWNER } from './grapqlQueries'
+import { REMOVE_ASSIGNED_OWNER, REMOVE_ASSIGNED_TEAM } from './grapqlQueries'
 
 export interface RemoveOwnerButtonProps {
     onSuccess: () => Promise<any>
     onError: (e: Error) => void
     repoId: string
     path: string
-    userId?: string
+    userID?: string
+    teamID?: string
     isDirectAssigned: boolean
 }
 
@@ -24,35 +30,44 @@ export const RemoveOwnerButton: React.FC<RemoveOwnerButtonProps> = ({
     onError,
     repoId,
     path,
-    userId,
+    userID,
+    teamID,
     isDirectAssigned,
 }) => {
     const tooltipContent = !isDirectAssigned
         ? 'Ownership can only be modified at the same direct path as it was assigned.'
         : 'Remove ownership'
 
-    const [removeAssignedOwner, { loading }] = useMutation<RemoveAssignedOwnerResult, RemoveAssignedOwnerVariables>(
-        REMOVE_ASSIGNED_OWNER,
-        {}
-    )
+    const [removeAssignedOwner, { loading: removeLoading }] = useMutation<
+        RemoveAssignedOwnerResult,
+        RemoveAssignedOwnerVariables
+    >(REMOVE_ASSIGNED_OWNER, {})
+    const [removeAssignedTeam, { loading: removeTeamLoading }] = useMutation<
+        RemoveAssignedTeamResult,
+        RemoveAssignedTeamVariables
+    >(REMOVE_ASSIGNED_TEAM, {})
+
+    const createInputObject = (id: string) => ({
+        variables: {
+            input: {
+                absolutePath: path,
+                assignedOwnerID: id,
+                repoID: repoId,
+            },
+        },
+        onCompleted: async () => {
+            await onSuccess()
+        },
+        onError: (errors: ErrorLike) => {
+            onError(asError(errors))
+        },
+    })
 
     const removeOwner: () => Promise<void> = async () => {
-        if (userId) {
-            await removeAssignedOwner({
-                variables: {
-                    input: {
-                        absolutePath: path,
-                        assignedOwnerID: userId,
-                        repoID: repoId,
-                    },
-                },
-                onCompleted: async () => {
-                    await onSuccess()
-                },
-                onError: (errors: ErrorLike) => {
-                    onError(asError(errors))
-                },
-            })
+        if (userID) {
+            await removeAssignedOwner(createInputObject(userID))
+        } else if (teamID) {
+            await removeAssignedTeam(createInputObject(teamID))
         }
     }
 
@@ -67,7 +82,11 @@ export const RemoveOwnerButton: React.FC<RemoveOwnerButtonProps> = ({
                 size="sm"
                 disabled={!isDirectAssigned}
             >
-                <Icon color="white" aria-hidden={true} svgPath={loading ? mdiLoading : mdiDelete} />
+                <Icon
+                    color="white"
+                    aria-hidden={true}
+                    svgPath={removeLoading || removeTeamLoading ? mdiLoading : mdiDelete}
+                />
                 Remove owner
             </Button>
         </Tooltip>

client/web/src/repo/blob/own/grapqlQueries.ts

@@ -20,6 +20,7 @@ export const OWNER_FIELDS = gql`
             }
         }
         ... on Team {
+            id
             name
             teamDisplayName: displayName
             avatarURL
@@ -199,3 +200,11 @@ export const REMOVE_ASSIGNED_OWNER = gql`
         }
     }
 `
+
+export const REMOVE_ASSIGNED_TEAM = gql`
+    mutation RemoveAssignedTeam($input: AssignOwnerOrTeamInput!) {
+        removeAssignedTeam(input: $input) {
+            alwaysNil
+        }
+    }
+`

enterprise/cmd/frontend/internal/own/resolvers/resolvers.go

@@ -797,6 +797,7 @@ func (r *ownResolver) computeAssignedTeams(ctx context.Context, db edb.Enterpris
 
 	fetchedTeams := make(map[int32]*types.Team)
 
+	isDirectMatch := false
 	for _, summary := range assignedTeamSummaries {
 		var team *types.Team
 		teamID := summary.OwnerTeamID
@@ -810,14 +811,17 @@ func (r *ownResolver) computeAssignedTeams(ctx context.Context, db edb.Enterpris
 			team = teamFromDB
 			fetchedTeams[teamID] = teamFromDB
 		}
+		if blob.Path() == summary.FilePath {
+			isDirectMatch = true
+		}
 		res := ownershipResolver{
 			db: db,
 			resolvedOwner: &codeowners.Team{
 				Team: team,
 			},
 			reasons: []*ownershipReasonResolver{
 				{
-					&assignedOwner{},
+					&assignedOwner{directMatch: isDirectMatch},
 				},
 			},
 		}