From 0ef926e5b473213e3436170fc18ae5eaf52dd333 Mon Sep 17 00:00:00 2001
From: Varun Patil <varunpatil@ucla.edu>
Date: Tue, 11 Oct 2022 12:49:05 -0700
Subject: [PATCH] Add face merging (fix #65)

---
 src/components/FaceMergeModal.vue | 162 ++++++++++++++++++++++++++++++
 src/components/FaceTopMatter.vue  |  14 ++-
 src/components/Modal.vue          |   5 +-
 src/components/Tag.vue            |   8 +-
 4 files changed, 184 insertions(+), 5 deletions(-)
 create mode 100644 src/components/FaceMergeModal.vue

diff --git a/src/components/FaceMergeModal.vue b/src/components/FaceMergeModal.vue
new file mode 100644
index 00000000..0f945b06
--- /dev/null
+++ b/src/components/FaceMergeModal.vue
@@ -0,0 +1,162 @@
+<template>
+    <Modal @close="close" size="large">
+        <template #title>
+            {{ t('memories', 'Merge {name} with person', { name }) }}
+        </template>
+
+        <div class="outer">
+            <div class="photo" v-for="photo of detail" :key="photo.fileid" >
+                <Tag :data="photo" :rowHeight="115" :noNavigate="true" @open="clickFace" />
+            </div>
+
+            <div v-if="procesingTotal > 0" class="info-pad">
+                {{ t('memories', 'Processing … {n}/{m}', {
+                    n: processing,
+                    m: procesingTotal,
+                }) }}
+            </div>
+        </div>
+
+        <template #buttons>
+            <NcButton @click="close" class="button" type="error">
+                {{ t('memories', 'Cancel') }}
+            </NcButton>
+        </template>
+    </Modal>
+</template>
+
+<script lang="ts">
+import { Component, Mixins, Watch } from 'vue-property-decorator';
+import { NcButton, NcTextField } from '@nextcloud/vue';
+import { showError } from '@nextcloud/dialogs'
+import { IFileInfo, IPhoto, ITag } from '../types';
+import Tag from './Tag.vue';
+
+import Modal from './Modal.vue';
+import GlobalMixin from '../mixins/GlobalMixin';
+import * as dav from '../services/DavRequests';
+import client from '../services/DavClient';
+
+@Component({
+    components: {
+        NcButton,
+        NcTextField,
+        Modal,
+        Tag,
+    }
+})
+export default class FaceMergeModal extends Mixins(GlobalMixin) {
+    private user: string = "";
+    private name: string = "";
+    private detail: IPhoto[] = [];
+    private processing = 0;
+    private procesingTotal = 0;
+
+    @Watch('$route')
+    async routeChange(from: any, to: any) {
+        this.refreshParams();
+    }
+
+    mounted() {
+        this.refreshParams();
+    }
+
+    public async refreshParams() {
+        this.user = this.$route.params.user || '';
+        this.name = this.$route.params.name || '';
+        this.detail = [];
+        this.processing = 0;
+        this.procesingTotal = 0;
+
+        const data = await dav.getPeopleData();
+        let detail = data[0].detail;
+        detail.forEach((photo: IPhoto) => {
+            photo.flag = this.c.FLAG_IS_FACE | this.c.FLAG_IS_TAG;
+        });
+        detail = detail.filter((photo: ITag) => {
+            const pname = photo.name || photo.fileid.toString();
+            return photo.user_id !== this.user || pname !== this.name;
+        });
+
+        this.detail = detail;
+    }
+
+    public async clickFace(face: ITag) {
+        const newName = face.name || face.fileid.toString();
+        if (!confirm(this.t('memories', 'Are you sure you want to merge {name} with {newName}?', { name: this.name, newName}))) {
+            return;
+        }
+
+		try {
+            // Get all files for current face
+            let res = await client.getDirectoryContents(
+                `/recognize/${this.user}/faces/${this.name}`, { details: true }
+            ) as any;
+            let data: IFileInfo[] = res.data;
+            this.procesingTotal = data.length;
+
+            // Don't try too much
+            let failures = 0;
+
+            // Create move calls
+            const calls = data.map((p) => async () => {
+                // Short circuit if we have too many failures
+                if (failures === 10) {
+                    showError(this.t('memories', 'Too many failures, aborting'));
+                    failures++;
+                }
+                if (failures >= 10) return;
+
+                // Move to new face with webdav
+                try {
+                    await client.moveFile(
+                        `/recognize/${this.user}/faces/${this.name}/${p.basename}`,
+                        `/recognize/${face.user_id}/faces/${newName}/${p.basename}`
+                    )
+                } catch (e) {
+                    console.error(e);
+                    showError(this.t('memories', 'Error while moving {basename}', p));
+                    failures++;
+                } finally {
+                    this.processing++;
+                }
+            });
+            for await (const _ of dav.runInParallel(calls, 10)) {
+                // nothing to do
+            }
+
+            // Go to new face
+            if (failures === 0) {
+                this.$router.push({ name: 'people', params: { user: face.user_id, name: newName } });
+                this.close();
+            }
+		} catch (error) {
+            console.error(error);
+			showError(this.t('photos', 'Failed to move {name}.', {
+                name: this.name,
+            }));
+		}
+    }
+
+    public close() {
+        this.$emit('close');
+    }
+}
+</script>
+
+<style lang="scss" scoped>
+.outer {
+    margin-top: 15px;
+}
+.photo {
+    display: inline-block;
+    position: relative;
+    cursor: pointer;
+    vertical-align: top;
+    font-size: 0.8em;
+}
+.info-pad {
+    margin-top: 6px;
+    margin-bottom: 2px;
+}
+</style>
\ No newline at end of file
diff --git a/src/components/FaceTopMatter.vue b/src/components/FaceTopMatter.vue
index a548041c..ded8f6e9 100644
--- a/src/components/FaceTopMatter.vue
+++ b/src/components/FaceTopMatter.vue
@@ -11,11 +11,15 @@
 
         <div class="right-actions">
             <NcActions :inline="1">
-                <NcActionButton :aria-label="t('memories', 'Rename person')" @click="showEditModal=true">
+                <NcActionButton :aria-label="t('memories', 'Rename person')" @click="showEditModal=true" close-after-click>
                     {{ t('memories', 'Rename person') }}
                     <template #icon> <EditIcon :size="20" /> </template>
                 </NcActionButton>
-                <NcActionButton :aria-label="t('memories', 'Remove person')" @click="showDeleteModal=true">
+                <NcActionButton :aria-label="t('memories', 'Merge with different person')" @click="showMergeModal=true" close-after-click>
+                    {{ t('memories', 'Merge with different person') }}
+                    <template #icon> <MergeIcon :size="20" /> </template>
+                </NcActionButton>
+                <NcActionButton :aria-label="t('memories', 'Remove person')" @click="showDeleteModal=true" close-after-click>
                     {{ t('memories', 'Remove person') }}
                     <template #icon> <DeleteIcon :size="20" /> </template>
                 </NcActionButton>
@@ -24,6 +28,7 @@
 
         <FaceEditModal v-if="showEditModal" @close="showEditModal=false" />
         <FaceDeleteModal v-if="showDeleteModal" @close="showDeleteModal=false" />
+        <FaceMergeModal v-if="showMergeModal" @close="showMergeModal=false" />
     </div>
 </template>
 
@@ -34,9 +39,11 @@ import GlobalMixin from '../mixins/GlobalMixin';
 import { NcActions, NcActionButton } from '@nextcloud/vue';
 import FaceEditModal from './FaceEditModal.vue';
 import FaceDeleteModal from './FaceDeleteModal.vue';
+import FaceMergeModal from './FaceMergeModal.vue';
 import BackIcon from 'vue-material-design-icons/ArrowLeft.vue';
 import EditIcon from 'vue-material-design-icons/Pencil.vue';
 import DeleteIcon from 'vue-material-design-icons/Close.vue';
+import MergeIcon from 'vue-material-design-icons/Merge.vue';
 
 @Component({
     components: {
@@ -44,15 +51,18 @@ import DeleteIcon from 'vue-material-design-icons/Close.vue';
         NcActionButton,
         FaceEditModal,
         FaceDeleteModal,
+        FaceMergeModal,
         BackIcon,
         EditIcon,
         DeleteIcon,
+        MergeIcon,
     },
 })
 export default class FaceTopMatter extends Mixins(GlobalMixin) {
     private name: string = '';
     private showEditModal: boolean = false;
     private showDeleteModal: boolean = false;
+    private showMergeModal: boolean = false;
 
     @Watch('$route')
     async routeChange(from: any, to: any) {
diff --git a/src/components/Modal.vue b/src/components/Modal.vue
index a972e7e4..85dc181f 100644
--- a/src/components/Modal.vue
+++ b/src/components/Modal.vue
@@ -1,6 +1,6 @@
 <template>
     <NcModal
-        size="small"
+        :size="size"
         @close="$emit('close')"
         :outTransition="true">
         <div class="container">
@@ -18,7 +18,7 @@
 </template>
 
 <script lang="ts">
-import { Component, Vue } from 'vue-property-decorator';
+import { Component, Prop, Vue } from 'vue-property-decorator';
 import { NcModal } from '@nextcloud/vue';
 
 @Component({
@@ -27,6 +27,7 @@ import { NcModal } from '@nextcloud/vue';
     }
 })
 export default class Modal extends Vue {
+    @Prop({default: 'small'}) private size?: string;
 }
 </script>
 
diff --git a/src/components/Tag.vue b/src/components/Tag.vue
index 24007408..925dc098 100644
--- a/src/components/Tag.vue
+++ b/src/components/Tag.vue
@@ -57,6 +57,7 @@ interface IFaceDetection extends IPhoto {
 export default class Tag extends Mixins(GlobalMixin) {
     @Prop() data: ITag;
     @Prop() rowHeight: number;
+    @Prop() noNavigate: boolean;
 
     // Separate property because the one on data isn't reactive
     private previews: IPhoto[] = [];
@@ -112,6 +113,11 @@ export default class Tag extends Mixins(GlobalMixin) {
 
     /** Open tag */
     openTag() {
+        this.$emit('open', this.data);
+        if (this.noNavigate) {
+            return;
+        }
+
         if (this.isFace) {
             const name = this.data.name || this.data.fileid.toString();
             const user = this.data.user_id;
@@ -178,7 +184,7 @@ export default class Tag extends Mixins(GlobalMixin) {
     top: 50%; width: 100%;
     transform: translateY(-50%);
     color: white;
-    width: 100%;
+    width: 90%;
     padding: 0 5%;
     text-align: center;
     font-size: 1.2em;