vikunja-api/pkg/models/label_task.go

348 lines
11 KiB
Go
Raw Normal View History

2020-02-07 17:27:45 +01:00
// Vikunja is a to-do list application to facilitate your life.
2020-01-09 18:33:22 +01:00
// Copyright 2018-2020 Vikunja and contributors. All rights reserved.
2018-12-31 02:18:41 +01:00
//
// This program is free software: you can redistribute it and/or modify
// it under the terms of the GNU General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.
2018-12-31 02:18:41 +01:00
//
// This program is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
// GNU General Public License for more details.
2018-12-31 02:18:41 +01:00
//
// You should have received a copy of the GNU General Public License
// along with this program. If not, see <https://www.gnu.org/licenses/>.
2018-12-31 02:18:41 +01:00
package models
import (
"code.vikunja.io/api/pkg/timeutil"
"code.vikunja.io/api/pkg/user"
2018-12-31 02:18:41 +01:00
"code.vikunja.io/web"
"xorm.io/builder"
2018-12-31 02:18:41 +01:00
)
2019-01-10 00:08:12 +01:00
// LabelTask represents a relation between a label and a task
type LabelTask struct {
// The unique, numeric id of this label.
ID int64 `xorm:"int(11) autoincr not null unique pk" json:"-"`
2019-01-10 00:08:12 +01:00
TaskID int64 `xorm:"int(11) INDEX not null" json:"-" param:"listtask"`
// The label id you want to associate with a task.
LabelID int64 `xorm:"int(11) INDEX not null" json:"label_id" param:"label"`
// A timestamp when this task was created. You cannot change this value.
Created timeutil.TimeStamp `xorm:"created not null" json:"created"`
2019-01-10 00:08:12 +01:00
web.CRUDable `xorm:"-" json:"-"`
web.Rights `xorm:"-" json:"-"`
}
// TableName makes a pretty table name
func (LabelTask) TableName() string {
return "label_task"
}
2018-12-31 02:18:41 +01:00
// Delete deletes a label on a task
// @Summary Remove a label from a task
// @Description Remove a label from a task. The user needs to have write-access to the list to be able do this.
// @tags labels
// @Accept json
// @Produce json
2019-01-03 23:22:06 +01:00
// @Security JWTKeyAuth
2018-12-31 02:18:41 +01:00
// @Param task path int true "Task ID"
// @Param label path int true "Label ID"
2019-03-21 07:40:56 +01:00
// @Success 200 {object} models.Message "The label was successfully removed."
2018-12-31 02:18:41 +01:00
// @Failure 403 {object} code.vikunja.io/web.HTTPError "Not allowed to remove the label."
// @Failure 404 {object} code.vikunja.io/web.HTTPError "Label not found."
// @Failure 500 {object} models.Message "Internal error"
// @Router /tasks/{task}/labels/{label} [delete]
2019-01-10 00:08:12 +01:00
func (lt *LabelTask) Delete() (err error) {
_, err = x.Delete(&LabelTask{LabelID: lt.LabelID, TaskID: lt.TaskID})
2018-12-31 02:18:41 +01:00
return err
}
// Create adds a label to a task
// @Summary Add a label to a task
// @Description Add a label to a task. The user needs to have write-access to the list to be able do this.
// @tags labels
// @Accept json
// @Produce json
2019-01-03 23:22:06 +01:00
// @Security JWTKeyAuth
2018-12-31 02:18:41 +01:00
// @Param task path int true "Task ID"
2019-03-21 07:40:56 +01:00
// @Param label body models.LabelTask true "The label object"
// @Success 200 {object} models.LabelTask "The created label relation object."
2018-12-31 02:18:41 +01:00
// @Failure 400 {object} code.vikunja.io/web.HTTPError "Invalid label object provided."
// @Failure 403 {object} code.vikunja.io/web.HTTPError "Not allowed to add the label."
// @Failure 404 {object} code.vikunja.io/web.HTTPError "The label does not exist."
// @Failure 500 {object} models.Message "Internal error"
// @Router /tasks/{task}/labels [put]
2019-01-10 00:08:12 +01:00
func (lt *LabelTask) Create(a web.Auth) (err error) {
2018-12-31 02:18:41 +01:00
// Check if the label is already added
2019-01-10 00:08:12 +01:00
exists, err := x.Exist(&LabelTask{LabelID: lt.LabelID, TaskID: lt.TaskID})
2018-12-31 02:18:41 +01:00
if err != nil {
return err
}
if exists {
2019-01-10 00:08:12 +01:00
return ErrLabelIsAlreadyOnTask{lt.LabelID, lt.TaskID}
2018-12-31 02:18:41 +01:00
}
// Insert it
2019-01-10 00:08:12 +01:00
_, err = x.Insert(lt)
2019-05-22 19:48:48 +02:00
if err != nil {
return err
}
err = updateListByTaskID(lt.TaskID)
2019-01-10 00:08:12 +01:00
return
2018-12-31 02:18:41 +01:00
}
// ReadAll gets all labels on a task
// @Summary Get all labels on a task
// @Description Returns all labels which are assicociated with a given task.
// @tags labels
// @Accept json
// @Produce json
// @Param task path int true "Task ID"
2019-10-23 23:11:40 +02:00
// @Param page query int false "The page number. Used for pagination. If not provided, the first page of results is returned."
// @Param per_page query int false "The maximum number of items per page. Note this parameter is limited by the configured maximum of items per page."
2018-12-31 02:18:41 +01:00
// @Param s query string false "Search labels by label text."
2019-01-03 23:22:06 +01:00
// @Security JWTKeyAuth
2018-12-31 02:18:41 +01:00
// @Success 200 {array} models.Label "The labels"
// @Failure 500 {object} models.Message "Internal error"
// @Router /tasks/{task}/labels [get]
2019-10-23 23:11:40 +02:00
func (lt *LabelTask) ReadAll(a web.Auth, search string, page int, perPage int) (result interface{}, resultCount int, numberOfTotalItems int64, err error) {
2018-12-31 02:18:41 +01:00
// Check if the user has the right to see the task
2019-08-14 22:19:04 +02:00
task := Task{ID: lt.TaskID}
2019-03-24 13:35:50 +01:00
canRead, err := task.CanRead(a)
if err != nil {
2019-10-23 23:11:40 +02:00
return nil, 0, 0, err
2019-03-24 13:35:50 +01:00
}
if !canRead {
2019-10-23 23:11:40 +02:00
return nil, 0, 0, ErrNoRightToSeeTask{lt.TaskID, a.GetID()}
2018-12-31 02:18:41 +01:00
}
2019-01-10 00:08:12 +01:00
return getLabelsByTaskIDs(&LabelByTaskIDsOptions{
User: &user.User{ID: a.GetID()},
2019-01-10 00:08:12 +01:00
Search: search,
Page: page,
TaskIDs: []int64{lt.TaskID},
})
2018-12-31 02:18:41 +01:00
}
2019-01-10 00:08:12 +01:00
// Helper struct, contains the label + its task ID
2018-12-31 02:18:41 +01:00
type labelWithTaskID struct {
TaskID int64 `json:"-"`
2018-12-31 02:18:41 +01:00
Label `xorm:"extends"`
}
2019-01-10 00:08:12 +01:00
// LabelByTaskIDsOptions is a struct to not clutter the function with too many optional parameters.
type LabelByTaskIDsOptions struct {
User *user.User
Search string
Page int
2019-10-23 23:11:40 +02:00
PerPage int
TaskIDs []int64
GetUnusedLabels bool
GroupByLabelIDsOnly bool
2019-01-10 00:08:12 +01:00
}
2018-12-31 02:18:41 +01:00
// Helper function to get all labels for a set of tasks
// Used when getting all labels for one task as well when getting all lables
2019-10-23 23:11:40 +02:00
func getLabelsByTaskIDs(opts *LabelByTaskIDsOptions) (ls []*labelWithTaskID, resultCount int, totalEntries int64, err error) {
// We still need the task ID when we want to get all labels for a task, but because of this, we get the same label
// multiple times when it is associated to more than one task.
// Because of this whole thing, we need this extra switch here to only group by Task IDs if needed.
// Probably not the most ideal solution.
var groupBy = "labels.id,label_task.task_id"
if opts.GroupByLabelIDsOnly {
groupBy = "labels.id"
}
// Get all labels associated with these tasks
2018-12-31 02:18:41 +01:00
var labels []*labelWithTaskID
cond := builder.And(builder.In("label_task.task_id", opts.TaskIDs), builder.NotNull{"label_task.label_id"})
if opts.GetUnusedLabels {
cond = builder.Or(cond, builder.Eq{"labels.created_by_id": opts.User.ID})
}
2018-12-31 02:18:41 +01:00
err = x.Table("labels").
Select("labels.*, label_task.task_id").
Join("LEFT", "label_task", "label_task.label_id = labels.id").
Where(cond).
2019-01-10 00:08:12 +01:00
And("labels.title LIKE ?", "%"+opts.Search+"%").
GroupBy(groupBy).
2019-10-23 23:11:40 +02:00
Limit(getLimitFromPageIndex(opts.Page, opts.PerPage)).
2018-12-31 02:18:41 +01:00
Find(&labels)
if err != nil {
2019-10-23 23:11:40 +02:00
return nil, 0, 0, err
2018-12-31 02:18:41 +01:00
}
// Get all created by users
var userids []int64
for _, l := range labels {
userids = append(userids, l.CreatedByID)
}
users := make(map[int64]*user.User)
2018-12-31 02:18:41 +01:00
err = x.In("id", userids).Find(&users)
if err != nil {
2019-10-23 23:11:40 +02:00
return nil, 0, 0, err
2018-12-31 02:18:41 +01:00
}
2019-08-14 21:59:31 +02:00
// Obfuscate all user emails
for _, u := range users {
u.Email = ""
}
2018-12-31 02:18:41 +01:00
// Put it all together
for in, l := range labels {
labels[in].CreatedBy = users[l.CreatedByID]
}
2019-10-23 23:11:40 +02:00
// Get the total number of entries
condCount := builder.And(builder.In("label_task.task_id", opts.TaskIDs), builder.NotNull{"label_task.label_id"})
if opts.GetUnusedLabels {
condCount = builder.Or(cond, builder.Eq{"labels.created_by_id": opts.User.ID})
}
2019-10-23 23:11:40 +02:00
totalEntries, err = x.Table("labels").
Join("LEFT", "label_task", "label_task.label_id = labels.id").
Where(condCount).
2019-10-23 23:11:40 +02:00
And("labels.title LIKE ?", "%"+opts.Search+"%").
GroupBy(groupBy).
Count(&Label{})
if err != nil {
return nil, 0, 0, err
}
return labels, len(labels), totalEntries, err
2018-12-31 02:18:41 +01:00
}
2019-01-10 00:08:12 +01:00
// Create or update a bunch of task labels
2019-08-14 22:19:04 +02:00
func (t *Task) updateTaskLabels(creator web.Auth, labels []*Label) (err error) {
2019-01-10 00:08:12 +01:00
// If we don't have any new labels, delete everything right away. Saves us some hassle.
if len(labels) == 0 && len(t.Labels) > 0 {
_, err = x.Where("task_id = ?", t.ID).
Delete(LabelTask{})
return err
}
// If we didn't change anything (from 0 to zero) don't do anything.
if len(labels) == 0 && len(t.Labels) == 0 {
return nil
}
// Make a hashmap of the new labels for easier comparison
newLabels := make(map[int64]*Label, len(labels))
for _, newLabel := range labels {
newLabels[newLabel.ID] = newLabel
}
// Get old labels to delete
var found bool
var labelsToDelete []int64
oldLabels := make(map[int64]*Label, len(t.Labels))
allLabels := t.Labels
t.Labels = []*Label{} // We re-empty our labels struct here because we want it to be fully empty so we can put in all the actual labels.
for _, oldLabel := range allLabels {
found = false
if newLabels[oldLabel.ID] != nil {
found = true // If a new label is already in the list with old labels
}
// Put all labels which are only on the old list to the trash
if !found {
labelsToDelete = append(labelsToDelete, oldLabel.ID)
} else {
t.Labels = append(t.Labels, oldLabel)
}
// Put it in a list with all old labels, just using the loop here
oldLabels[oldLabel.ID] = oldLabel
}
// Delete all labels not passed
if len(labelsToDelete) > 0 {
_, err = x.In("label_id", labelsToDelete).
And("task_id = ?", t.ID).
Delete(LabelTask{})
if err != nil {
return err
}
}
// Loop through our labels and add them
for _, l := range labels {
// Check if the label is already added on the task and only add it if not
if oldLabels[l.ID] != nil {
// continue outer loop
continue
}
// Add the new label
label, err := getLabelByIDSimple(l.ID)
if err != nil {
return err
}
// Check if the user has the rights to see the label he is about to add
2019-03-24 13:35:50 +01:00
hasAccessToLabel, err := label.hasAccessToLabel(creator)
if err != nil {
return err
}
if !hasAccessToLabel {
user, _ := creator.(*user.User)
2019-01-10 00:08:12 +01:00
return ErrUserHasNoAccessToLabel{LabelID: l.ID, UserID: user.ID}
}
// Insert it
_, err = x.Insert(&LabelTask{LabelID: l.ID, TaskID: t.ID})
if err != nil {
return err
}
t.Labels = append(t.Labels, label)
}
2019-05-22 19:48:48 +02:00
err = updateListLastUpdated(&List{ID: t.ListID})
2019-01-10 00:08:12 +01:00
return
}
// LabelTaskBulk is a helper struct to update a bunch of labels at once
type LabelTaskBulk struct {
// All labels you want to update at once.
2019-01-10 00:08:12 +01:00
Labels []*Label `json:"labels"`
TaskID int64 `json:"-" param:"listtask"`
web.CRUDable `json:"-"`
web.Rights `json:"-"`
}
// Create updates a bunch of labels on a task at once
// @Summary Update all labels on a task.
// @Description Updates all labels on a task. Every label which is not passed but exists on the task will be deleted. Every label which does not exist on the task will be added. All labels which are passed and already exist on the task won't be touched.
2019-01-10 00:08:12 +01:00
// @tags labels
// @Accept json
// @Produce json
// @Security JWTKeyAuth
// @Param label body models.LabelTaskBulk true "The array of labels"
// @Param taskID path int true "Task ID"
// @Success 200 {object} models.LabelTaskBulk "The updated labels object."
// @Failure 400 {object} code.vikunja.io/web.HTTPError "Invalid label object provided."
// @Failure 500 {object} models.Message "Internal error"
// @Router /tasks/{taskID}/labels/bulk [post]
func (ltb *LabelTaskBulk) Create(a web.Auth) (err error) {
task, err := GetTaskByIDSimple(ltb.TaskID)
2019-01-10 00:08:12 +01:00
if err != nil {
return
}
labels, _, _, err := getLabelsByTaskIDs(&LabelByTaskIDsOptions{
TaskIDs: []int64{ltb.TaskID},
})
if err != nil {
return err
}
for _, l := range labels {
task.Labels = append(task.Labels, &l.Label)
}
2019-03-21 19:26:04 +01:00
return task.updateTaskLabels(a, ltb.Labels)
2019-01-10 00:08:12 +01:00
}