mirror of
https://github.com/documize/community.git
synced 2025-07-19 13:19:43 +02:00
252 lines
9.5 KiB
Go
252 lines
9.5 KiB
Go
// Copyright 2016 Documize Inc. <legal@documize.com>. All rights reserved.
|
|
//
|
|
// This software (Documize Community Edition) is licensed under
|
|
// GNU AGPL v3 http://www.gnu.org/licenses/agpl-3.0.en.html
|
|
//
|
|
// You can operate outside the AGPL restrictions by purchasing
|
|
// Documize Enterprise Edition and obtaining a commercial license
|
|
// by contacting <sales@documize.com>.
|
|
//
|
|
// https://documize.com
|
|
|
|
// Package mysql handles data persistence for both category definition
|
|
// and and document/category association.
|
|
package mysql
|
|
|
|
import (
|
|
"database/sql"
|
|
"fmt"
|
|
"time"
|
|
|
|
"github.com/documize/community/core/env"
|
|
"github.com/documize/community/domain"
|
|
"github.com/documize/community/domain/store/mysql"
|
|
"github.com/documize/community/model/category"
|
|
"github.com/pkg/errors"
|
|
)
|
|
|
|
// Scope provides data access to MySQL.
|
|
type Scope struct {
|
|
Runtime *env.Runtime
|
|
}
|
|
|
|
// Add inserts the given record into the category table.
|
|
func (s Scope) Add(ctx domain.RequestContext, c category.Category) (err error) {
|
|
c.Created = time.Now().UTC()
|
|
c.Revised = time.Now().UTC()
|
|
|
|
_, err = ctx.Transaction.Exec("INSERT INTO category (refid, orgid, labelid, category, created, revised) VALUES (?, ?, ?, ?, ?, ?)",
|
|
c.RefID, c.OrgID, c.LabelID, c.Category, c.Created, c.Revised)
|
|
|
|
if err != nil {
|
|
err = errors.Wrap(err, "unable to execute insert category")
|
|
}
|
|
|
|
return
|
|
}
|
|
|
|
// GetBySpace returns space categories accessible by user.
|
|
// Context is used to for user ID.
|
|
func (s Scope) GetBySpace(ctx domain.RequestContext, spaceID string) (c []category.Category, err error) {
|
|
err = s.Runtime.Db.Select(&c, `
|
|
SELECT id, refid, orgid, labelid, category, created, revised FROM category
|
|
WHERE orgid=? AND labelid=?
|
|
AND refid IN (SELECT refid FROM permission WHERE orgid=? AND location='category' AND refid IN (
|
|
SELECT refid from permission WHERE orgid=? AND who='user' AND (whoid=? OR whoid='0') AND location='category' UNION ALL
|
|
SELECT p.refid from permission p LEFT JOIN rolemember r ON p.whoid=r.roleid
|
|
WHERE p.orgid=? AND p.who='role' AND p.location='category' AND (r.userid=? OR r.userid='0')
|
|
))
|
|
ORDER BY category`, ctx.OrgID, spaceID, ctx.OrgID, ctx.OrgID, ctx.UserID, ctx.OrgID, ctx.UserID)
|
|
|
|
if err == sql.ErrNoRows {
|
|
err = nil
|
|
}
|
|
if err != nil {
|
|
err = errors.Wrap(err, fmt.Sprintf("unable to execute select categories for space %s", spaceID))
|
|
}
|
|
|
|
return
|
|
}
|
|
|
|
// GetAllBySpace returns all space categories.
|
|
func (s Scope) GetAllBySpace(ctx domain.RequestContext, spaceID string) (c []category.Category, err error) {
|
|
err = s.Runtime.Db.Select(&c, `
|
|
SELECT id, refid, orgid, labelid, category, created, revised FROM category
|
|
WHERE orgid=? AND labelid=?
|
|
AND labelid IN (SELECT refid FROM permission WHERE orgid=? AND location='space' AND refid IN (
|
|
SELECT refid from permission WHERE orgid=? AND who='user' AND (whoid=? OR whoid='0') AND location='space' AND action='view' UNION ALL
|
|
SELECT p.refid from permission p LEFT JOIN rolemember r ON p.whoid=r.roleid WHERE p.orgid=? AND p.who='role' AND p.location='space'
|
|
AND p.action='view' AND (r.userid=? OR r.userid='0')
|
|
))
|
|
ORDER BY category`, ctx.OrgID, spaceID, ctx.OrgID, ctx.OrgID, ctx.UserID, ctx.OrgID, ctx.UserID)
|
|
|
|
if err == sql.ErrNoRows || len(c) == 0 {
|
|
err = nil
|
|
c = []category.Category{}
|
|
}
|
|
if err != nil {
|
|
err = errors.Wrap(err, fmt.Sprintf("unable to execute select all categories for space %s", spaceID))
|
|
}
|
|
|
|
return
|
|
}
|
|
|
|
// Update saves category name change.
|
|
func (s Scope) Update(ctx domain.RequestContext, c category.Category) (err error) {
|
|
c.Revised = time.Now().UTC()
|
|
|
|
_, err = ctx.Transaction.NamedExec("UPDATE category SET category=:category, revised=:revised WHERE orgid=:orgid AND refid=:refid", c)
|
|
|
|
if err != nil {
|
|
err = errors.Wrap(err, fmt.Sprintf("unable to execute update for category %s", c.RefID))
|
|
}
|
|
|
|
return
|
|
}
|
|
|
|
// Get returns specified category
|
|
func (s Scope) Get(ctx domain.RequestContext, id string) (c category.Category, err error) {
|
|
err = s.Runtime.Db.Get(&c, "SELECT id, refid, orgid, labelid, category, created, revised FROM category WHERE orgid=? AND refid=?",
|
|
ctx.OrgID, id)
|
|
|
|
if err != nil {
|
|
err = errors.Wrap(err, fmt.Sprintf("unable to get category %s", id))
|
|
}
|
|
|
|
return
|
|
}
|
|
|
|
// Delete removes category from the store.
|
|
func (s Scope) Delete(ctx domain.RequestContext, id string) (rows int64, err error) {
|
|
b := mysql.BaseQuery{}
|
|
return b.DeleteConstrained(ctx.Transaction, "category", ctx.OrgID, id)
|
|
}
|
|
|
|
// AssociateDocument inserts category membership record into the category member table.
|
|
func (s Scope) AssociateDocument(ctx domain.RequestContext, m category.Member) (err error) {
|
|
m.Created = time.Now().UTC()
|
|
m.Revised = time.Now().UTC()
|
|
|
|
_, err = ctx.Transaction.Exec("INSERT INTO categorymember (refid, orgid, categoryid, labelid, documentid, created, revised) VALUES (?, ?, ?, ?, ?, ?, ?)",
|
|
m.RefID, m.OrgID, m.CategoryID, m.LabelID, m.DocumentID, m.Created, m.Revised)
|
|
|
|
if err != nil {
|
|
err = errors.Wrap(err, "unable to execute insert categorymember")
|
|
}
|
|
|
|
return
|
|
}
|
|
|
|
// DisassociateDocument removes document associatation from category.
|
|
func (s Scope) DisassociateDocument(ctx domain.RequestContext, categoryID, documentID string) (rows int64, err error) {
|
|
b := mysql.BaseQuery{}
|
|
|
|
sql := fmt.Sprintf("DELETE FROM categorymember WHERE orgid='%s' AND categoryid='%s' AND documentid='%s'",
|
|
ctx.OrgID, categoryID, documentID)
|
|
|
|
return b.DeleteWhere(ctx.Transaction, sql)
|
|
}
|
|
|
|
// RemoveCategoryMembership removes all category associations from the store.
|
|
func (s Scope) RemoveCategoryMembership(ctx domain.RequestContext, categoryID string) (rows int64, err error) {
|
|
b := mysql.BaseQuery{}
|
|
|
|
sql := fmt.Sprintf("DELETE FROM categorymember WHERE orgid='%s' AND categoryid='%s'",
|
|
ctx.OrgID, categoryID)
|
|
|
|
return b.DeleteWhere(ctx.Transaction, sql)
|
|
}
|
|
|
|
// RemoveSpaceCategoryMemberships removes all category associations from the store for the space.
|
|
func (s Scope) RemoveSpaceCategoryMemberships(ctx domain.RequestContext, spaceID string) (rows int64, err error) {
|
|
b := mysql.BaseQuery{}
|
|
|
|
sql := fmt.Sprintf("DELETE FROM categorymember WHERE orgid='%s' AND labelid='%s'",
|
|
ctx.OrgID, spaceID)
|
|
|
|
return b.DeleteWhere(ctx.Transaction, sql)
|
|
}
|
|
|
|
// RemoveDocumentCategories removes all document category associations from the store.
|
|
func (s Scope) RemoveDocumentCategories(ctx domain.RequestContext, documentID string) (rows int64, err error) {
|
|
b := mysql.BaseQuery{}
|
|
|
|
sql := fmt.Sprintf("DELETE FROM categorymember WHERE orgid='%s' AND documentid='%s'",
|
|
ctx.OrgID, documentID)
|
|
|
|
return b.DeleteWhere(ctx.Transaction, sql)
|
|
}
|
|
|
|
// DeleteBySpace removes all category and category associations for given space.
|
|
func (s Scope) DeleteBySpace(ctx domain.RequestContext, spaceID string) (rows int64, err error) {
|
|
b := mysql.BaseQuery{}
|
|
|
|
s1 := fmt.Sprintf("DELETE FROM categorymember WHERE orgid='%s' AND labelid='%s'", ctx.OrgID, spaceID)
|
|
b.DeleteWhere(ctx.Transaction, s1)
|
|
|
|
s2 := fmt.Sprintf("DELETE FROM category WHERE orgid='%s' AND labelid='%s'", ctx.OrgID, spaceID)
|
|
return b.DeleteWhere(ctx.Transaction, s2)
|
|
}
|
|
|
|
// GetSpaceCategorySummary returns number of documents and users for space categories.
|
|
func (s Scope) GetSpaceCategorySummary(ctx domain.RequestContext, spaceID string) (c []category.SummaryModel, err error) {
|
|
err = s.Runtime.Db.Select(&c, `
|
|
SELECT 'documents' as type, categoryid, COUNT(*) as count
|
|
FROM categorymember
|
|
WHERE orgid=? AND labelid=? GROUP BY categoryid, type
|
|
UNION ALL
|
|
SELECT 'users' as type, refid AS categoryid, count(*) AS count
|
|
FROM permission
|
|
WHERE orgid=? AND location='category'
|
|
AND refid IN (SELECT refid FROM category WHERE orgid=? AND labelid=?)
|
|
GROUP BY refid, type`,
|
|
ctx.OrgID, spaceID, ctx.OrgID, ctx.OrgID, spaceID /*, ctx.OrgID, ctx.OrgID, spaceID*/)
|
|
|
|
if err == sql.ErrNoRows || len(c) == 0 {
|
|
err = nil
|
|
c = []category.SummaryModel{}
|
|
}
|
|
if err != nil {
|
|
err = errors.Wrap(err, fmt.Sprintf("select category summary for space %s", spaceID))
|
|
}
|
|
|
|
return
|
|
}
|
|
|
|
// GetDocumentCategoryMembership returns all space categories associated with given document.
|
|
func (s Scope) GetDocumentCategoryMembership(ctx domain.RequestContext, documentID string) (c []category.Category, err error) {
|
|
err = s.Runtime.Db.Select(&c, `
|
|
SELECT id, refid, orgid, labelid, category, created, revised FROM category
|
|
WHERE orgid=? AND refid IN (SELECT categoryid FROM categorymember WHERE orgid=? AND documentid=?)`, ctx.OrgID, ctx.OrgID, documentID)
|
|
|
|
if err == sql.ErrNoRows {
|
|
err = nil
|
|
}
|
|
if err != nil {
|
|
err = errors.Wrap(err, fmt.Sprintf("unable to execute select categories for document %s", documentID))
|
|
}
|
|
|
|
return
|
|
}
|
|
|
|
// GetSpaceCategoryMembership returns category/document associations within space.
|
|
func (s Scope) GetSpaceCategoryMembership(ctx domain.RequestContext, spaceID string) (c []category.Member, err error) {
|
|
err = s.Runtime.Db.Select(&c, `
|
|
SELECT id, refid, orgid, labelid, categoryid, documentid, created, revised FROM categorymember
|
|
WHERE orgid=? AND labelid=?
|
|
AND labelid IN (SELECT refid FROM permission WHERE orgid=? AND location='space' AND refid IN (
|
|
SELECT refid from permission WHERE orgid=? AND who='user' AND (whoid=? OR whoid='0') AND location='space' AND action='view' UNION ALL
|
|
SELECT p.refid from permission p LEFT JOIN rolemember r ON p.whoid=r.roleid WHERE p.orgid=? AND p.who='role' AND p.location='space'
|
|
AND p.action='view' AND (r.userid=? OR r.userid='0')
|
|
))
|
|
ORDER BY documentid`, ctx.OrgID, spaceID, ctx.OrgID, ctx.OrgID, ctx.UserID, ctx.OrgID, ctx.UserID)
|
|
|
|
if err == sql.ErrNoRows {
|
|
err = nil
|
|
}
|
|
if err != nil {
|
|
err = errors.Wrap(err, fmt.Sprintf("select all category/document membership for space %s", spaceID))
|
|
}
|
|
|
|
return
|
|
}
|