vikunja-api/pkg/modules/migration/create_from_structure.go

// Vikunja is a todo-list application to facilitate your life.
// Copyright 2018-2020 Vikunja and contributors. All rights reserved.
//
// 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.
//
// 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.
//
// You should have received a copy of the GNU General Public License
// along with this program.  If not, see <https://www.gnu.org/licenses/>.

package migration

import (
	"bytes"
	"code.vikunja.io/api/pkg/models"
	"code.vikunja.io/api/pkg/user"
	"io/ioutil"
)

// InsertFromStructure takes a fully nested Vikunja data structure and a user and then creates everything for this user
// (Namespaces, tasks, etc. Even attachments and relations.)
func InsertFromStructure(str []*models.NamespaceWithLists, user *user.User) (err error) {

	// Create all namespaces
	for _, n := range str {
		err = n.Create(user)
		if err != nil {
			return
		}

		// Create all lists
		for _, l := range n.Lists {
			// The tasks slice is going to be reset during the creation of the list so we rescue it here to be able
			// to still loop over the tasks aftere the list was created.
			tasks := l.Tasks

			l.NamespaceID = n.ID
			err = l.Create(user)
			if err != nil {
				return
			}

			// Create all tasks
			for _, t := range tasks {
				t.ListID = l.ID
				err = t.Create(user)
				if err != nil {
					return
				}

				// Create all relation for each task
				for kind, tasks := range t.RelatedTasks {
					// First create the related tasks if they does not exist
					for _, rt := range tasks {
						if rt.ID == 0 {
							err = rt.Create(user)
							if err != nil {
								return
							}
						}

						// Then create the relation
						taskRel := &models.TaskRelation{
							TaskID:       rt.ID,
							OtherTaskID:  t.ID,
							RelationKind: kind,
						}
						err = taskRel.Create(user)
						if err != nil {
							return
						}
					}
				}

				// Create all attachments for each task
				for _, a := range t.Attachments {
					// Check if we have a file to create
					if len(a.File.FileContent) > 0 {
						a.TaskID = t.ID
						fr := ioutil.NopCloser(bytes.NewReader(a.File.FileContent))
						err = a.NewAttachment(fr, a.File.Name, a.File.Size, user)
						if err != nil {
							return
						}
					}
				}
			}
		}
	}

	return nil
}