forked from readium/readium-lcp-server
-
Notifications
You must be signed in to change notification settings - Fork 0
/
webrepository.go
79 lines (63 loc) · 2.07 KB
/
webrepository.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
// Copyright (c) 2020 Readium Foundation
// Use of this source code is governed by a BSD-style license
// that can be found in the LICENSE file exposed on Github (readium) in the project repository.
package webrepository
import (
"errors"
"io/ioutil"
"os"
"path"
"github.com/fubaydullaev/readium-lcp-server/config"
)
// ErrNotFound error trown when repository is not found
var ErrNotFound = errors.New("Repository not found")
// WebRepository interface for repository db interaction
type WebRepository interface {
GetMasterFile(name string) (RepositoryFile, error)
GetMasterFiles() func() (RepositoryFile, error)
}
// RepositoryFile struct defines a file stored in a repository
type RepositoryFile struct {
Name string `json:"name"`
Path string
}
// RepositoryManager contains all repository definitions
type RepositoryManager struct {
MasterRepositoryPath string
EncryptedRepositoryPath string
}
// GetMasterFile returns a specific repository file
func (repManager RepositoryManager) GetMasterFile(name string) (RepositoryFile, error) {
var filePath = path.Join(repManager.MasterRepositoryPath, name)
if _, err := os.Stat(filePath); err == nil {
// File exists
var repFile RepositoryFile
repFile.Name = name
repFile.Path = filePath
return repFile, err
}
return RepositoryFile{}, ErrNotFound
}
// GetMasterFiles returns all filenames from the master repository
func (repManager RepositoryManager) GetMasterFiles() func() (RepositoryFile, error) {
files, err := ioutil.ReadDir(repManager.MasterRepositoryPath)
var fileIndex int
if err != nil {
return func() (RepositoryFile, error) { return RepositoryFile{}, err }
}
return func() (RepositoryFile, error) {
var repFile RepositoryFile
for fileIndex < len(files) {
file := files[fileIndex]
repFile.Name = file.Name()
fileIndex++
return repFile, err
}
return repFile, ErrNotFound
}
}
// Init returns a WebPublication interface (db interaction)
func Init(config config.FrontendServerInfo) (i WebRepository, err error) {
i = RepositoryManager{config.MasterRepository, config.EncryptedRepository}
return
}