From 257938228015277c20c2d03a84338c9d0980b82b Mon Sep 17 00:00:00 2001
From: McMatts <matt@documize.com>
Date: Wed, 4 Apr 2018 14:38:04 +0100
Subject: [PATCH] Copt document categories when copying document

---
 domain/document/document.go | 19 +++++++++++++++++++
 1 file changed, 19 insertions(+)

diff --git a/domain/document/document.go b/domain/document/document.go
index 4431f828..73a9308e 100644
--- a/domain/document/document.go
+++ b/domain/document/document.go
@@ -142,6 +142,25 @@ func CopyDocument(ctx domain.RequestContext, s domain.Store, documentID string)
 		}
 	}
 
+	cats, err := s.Category.GetDocumentCategoryMembership(ctx, documentID)
+	if err != nil {
+		err = errors.Wrap(err, "unable to add copied page")
+		return
+	}
+
+	for ci := range cats {
+		cm := category.Member{}
+		cm.DocumentID = newDocumentID
+		cm.CategoryID = cats[ci].RefID
+		cm.OrgID = ctx.OrgID
+		cm.RefID = uniqueid.Generate()
+		s.Category.AssociateDocument(ctx, cm)
+		if err != nil {
+			err = errors.Wrap(err, "unable to add copied page")
+			return
+		}
+	}
+
 	return
 }