136 lines
3.2 KiB
Go
136 lines
3.2 KiB
Go
// Vikunja is a to-do list application to facilitate your life.
|
|
// Copyright 2018-2021 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 Affero General Public Licensee 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 Affero General Public Licensee for more details.
|
|
//
|
|
// You should have received a copy of the GNU Affero General Public Licensee
|
|
// along with this program. If not, see <https://www.gnu.org/licenses/>.
|
|
|
|
package vikunja_file
|
|
|
|
import (
|
|
"archive/zip"
|
|
"bytes"
|
|
"code.vikunja.io/api/pkg/db"
|
|
"code.vikunja.io/api/pkg/models"
|
|
"code.vikunja.io/api/pkg/modules/migration"
|
|
"code.vikunja.io/api/pkg/user"
|
|
"encoding/json"
|
|
"fmt"
|
|
"io"
|
|
"strings"
|
|
)
|
|
|
|
type VikunjaFileMigrator struct {
|
|
}
|
|
|
|
func (v *VikunjaFileMigrator) Name() string {
|
|
return "vikunja-file"
|
|
}
|
|
|
|
func (v *VikunjaFileMigrator) Migrate(user *user.User, file io.ReaderAt, size int64) error {
|
|
r, err := zip.NewReader(file, size)
|
|
if err != nil {
|
|
return fmt.Errorf("could not open import file: %s", err)
|
|
}
|
|
|
|
var dataFile *zip.File
|
|
var filterFile *zip.File
|
|
storedFiles := make(map[string]*zip.File)
|
|
for _, f := range r.File {
|
|
if strings.HasPrefix(f.Name, "files/") {
|
|
fname := strings.ReplaceAll(f.Name, "files/", "")
|
|
storedFiles[fname] = f
|
|
continue
|
|
}
|
|
if f.Name == "data.json" {
|
|
dataFile = f
|
|
continue
|
|
}
|
|
if f.Name == "filters.json" {
|
|
filterFile = f
|
|
}
|
|
}
|
|
|
|
// TODO: check we have all files
|
|
|
|
//////
|
|
// Import the bulk of Vikunja data
|
|
df, err := dataFile.Open()
|
|
if err != nil {
|
|
return fmt.Errorf("could not open data file: %s", err)
|
|
}
|
|
defer df.Close()
|
|
|
|
var bufData bytes.Buffer
|
|
if _, err := bufData.ReadFrom(df); err != nil {
|
|
return fmt.Errorf("could not read data file: %s", err)
|
|
}
|
|
|
|
namespaces := []*models.NamespaceWithListsAndTasks{}
|
|
if err := json.Unmarshal(bufData.Bytes(), &namespaces); err != nil {
|
|
return fmt.Errorf("could not read data: %s", err)
|
|
}
|
|
|
|
for _, n := range namespaces {
|
|
for _, l := range n.Lists {
|
|
for _, t := range l.Tasks {
|
|
if len(t.Labels) == 0 {
|
|
continue
|
|
}
|
|
for _, label := range t.Labels {
|
|
label.ID = 0
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
// Import files
|
|
// TODO
|
|
|
|
err = migration.InsertFromStructure(namespaces, user)
|
|
if err != nil {
|
|
return fmt.Errorf("could not insert data: %s", err)
|
|
}
|
|
|
|
///////
|
|
// Import filters
|
|
ff, err := filterFile.Open()
|
|
if err != nil {
|
|
return fmt.Errorf("could not open filters file: %s", err)
|
|
}
|
|
defer ff.Close()
|
|
|
|
var bufFilter bytes.Buffer
|
|
if _, err := bufFilter.ReadFrom(ff); err != nil {
|
|
return fmt.Errorf("could not read filters file: %s", err)
|
|
}
|
|
|
|
filters := []*models.SavedFilter{}
|
|
if err := json.Unmarshal(bufFilter.Bytes(), &filters); err != nil {
|
|
return fmt.Errorf("could not read filter data: %s", err)
|
|
}
|
|
|
|
s := db.NewSession()
|
|
defer s.Close()
|
|
|
|
for _, f := range filters {
|
|
f.ID = 0
|
|
err = f.Create(s, user)
|
|
if err != nil {
|
|
_ = s.Rollback()
|
|
return err
|
|
}
|
|
}
|
|
|
|
return s.Commit()
|
|
}
|