From 5924096e0f9637e6c9740ccd647e30855b984737 Mon Sep 17 00:00:00 2001
From: viktorstrate <viktorstrate@gmail.com>
Date: Mon, 22 Feb 2021 19:11:48 +0100
Subject: [PATCH] Add UI for detaching faces from face group

---
 .../SingleFaceGroup/DetachImageFacesModal.js  | 96 +++++++++++++++++++
 .../SingleFaceGroup/FaceGroupTitle.js         |  9 +-
 2 files changed, 104 insertions(+), 1 deletion(-)
 create mode 100644 ui/src/Pages/PeoplePage/SingleFaceGroup/DetachImageFacesModal.js

diff --git a/ui/src/Pages/PeoplePage/SingleFaceGroup/DetachImageFacesModal.js b/ui/src/Pages/PeoplePage/SingleFaceGroup/DetachImageFacesModal.js
new file mode 100644
index 00000000..7a53d872
--- /dev/null
+++ b/ui/src/Pages/PeoplePage/SingleFaceGroup/DetachImageFacesModal.js
@@ -0,0 +1,96 @@
+import { gql, useMutation } from '@apollo/client'
+import PropTypes from 'prop-types'
+import React, { useEffect, useState } from 'react'
+import { useHistory } from 'react-router-dom'
+import { Button, Modal } from 'semantic-ui-react'
+import { MY_FACES_QUERY } from '../PeoplePage'
+import SelectImageFacesTable from './SelectImageFacesTable'
+
+const DETACH_IMAGE_FACES_MUTATION = gql`
+  mutation detachImageFaces($faceIDs: [ID!]!) {
+    detachImageFaces(imageFaceIDs: $faceIDs) {
+      id
+      label
+    }
+  }
+`
+
+const DetachImageFacesModal = ({ open, setOpen, faceGroup }) => {
+  const [selectedImageFaces, setSelectedImageFaces] = useState([])
+  let history = useHistory()
+
+  const [detachImageFacesMutation] = useMutation(DETACH_IMAGE_FACES_MUTATION, {
+    variables: {},
+    refetchQueries: [
+      {
+        query: MY_FACES_QUERY,
+      },
+    ],
+  })
+
+  useEffect(() => {
+    if (!open) {
+      setSelectedImageFaces([])
+    }
+  }, [open])
+
+  if (open == false) return null
+
+  const detachImageFaces = () => {
+    const faceIDs = selectedImageFaces.map(face => face.id)
+
+    detachImageFacesMutation({
+      variables: {
+        faceIDs,
+      },
+    }).then(({ data }) => {
+      setOpen(false)
+      history.push(`/people/${data.detachImageFaces.id}`)
+    })
+  }
+
+  const imageFaces = faceGroup?.imageFaces ?? []
+
+  return (
+    <Modal
+      onClose={() => setOpen(false)}
+      onOpen={() => setOpen(true)}
+      open={open}
+    >
+      <Modal.Header>Detach Image Faces</Modal.Header>
+      <Modal.Content scrolling>
+        <Modal.Description>
+          <p>
+            Detach selected images of this face group and move them to a new
+            face group
+          </p>
+          <SelectImageFacesTable
+            imageFaces={imageFaces}
+            selectedImageFaces={selectedImageFaces}
+            setSelectedImageFaces={setSelectedImageFaces}
+            title="Select images to detach"
+          />
+        </Modal.Description>
+      </Modal.Content>
+      <Modal.Actions>
+        <Button onClick={() => setOpen(false)}>Cancel</Button>
+        <Button
+          disabled={selectedImageFaces.length == 0}
+          content="Detach image faces"
+          labelPosition="right"
+          icon="checkmark"
+          onClick={() => detachImageFaces()}
+          positive
+        />
+      </Modal.Actions>
+    </Modal>
+  )
+}
+
+DetachImageFacesModal.propTypes = {
+  open: PropTypes.bool.isRequired,
+  setOpen: PropTypes.func.isRequired,
+  faceGroup: PropTypes.object,
+}
+
+export default DetachImageFacesModal
diff --git a/ui/src/Pages/PeoplePage/SingleFaceGroup/FaceGroupTitle.js b/ui/src/Pages/PeoplePage/SingleFaceGroup/FaceGroupTitle.js
index b399db05..ec30be59 100644
--- a/ui/src/Pages/PeoplePage/SingleFaceGroup/FaceGroupTitle.js
+++ b/ui/src/Pages/PeoplePage/SingleFaceGroup/FaceGroupTitle.js
@@ -4,6 +4,7 @@ import React, { useState, useEffect, createRef } from 'react'
 import { Dropdown, Input } from 'semantic-ui-react'
 import styled from 'styled-components'
 import { SET_GROUP_LABEL_MUTATION } from '../PeoplePage'
+import DetachImageFacesModal from './DetachImageFacesModal'
 import MergeFaceGroupsModal from './MergeFaceGroupsModal'
 import MoveImageFacesModal from './MoveImageFacesModal'
 
@@ -33,6 +34,7 @@ const FaceGroupTitle = ({ faceGroup }) => {
   const inputRef = createRef()
   const [mergeModalOpen, setMergeModalOpen] = useState(false)
   const [moveModalOpen, setMoveModalOpen] = useState(false)
+  const [detachModalOpen, setDetachModalOpen] = useState(false)
 
   const [setGroupLabel, { loading: setLabelLoading }] = useMutation(
     SET_GROUP_LABEL_MUTATION,
@@ -103,7 +105,7 @@ const FaceGroupTitle = ({ faceGroup }) => {
             <Dropdown.Item
               icon="object ungroup"
               text="Detach Faces"
-              onClick={() => setMergeModalOpen(true)}
+              onClick={() => setDetachModalOpen(true)}
             />
             <Dropdown.Item
               icon="clone"
@@ -146,6 +148,11 @@ const FaceGroupTitle = ({ faceGroup }) => {
         setOpen={setMoveModalOpen}
         faceGroup={faceGroup}
       />
+      <DetachImageFacesModal
+        open={detachModalOpen}
+        setOpen={setDetachModalOpen}
+        faceGroup={faceGroup}
+      />
     </>
   )
 }