-
Notifications
You must be signed in to change notification settings - Fork 0
/
appstore.go
92 lines (78 loc) · 1.93 KB
/
appstore.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
80
81
82
83
84
85
86
87
88
89
90
91
92
// package builtin provides an abstraction for uploading files and manifests
// to a file repository.
package builtin
import (
"io"
"io/ioutil"
"os"
"path/filepath"
"strings"
"github.com/groob/plist"
"github.com/pkg/errors"
"github.com/micromdm/micromdm/mdm/appmanifest"
)
type Repo struct {
Path string
}
func (r *Repo) SaveFile(name string, f io.Reader) error {
fname := filepath.Join(r.Path, name)
file, err := os.Create(fname)
if err != nil {
return errors.Wrapf(err, "saving file %s", name)
}
defer file.Close()
_, err = io.Copy(file, f)
return err
}
func (r *Repo) Manifest(name string) (*appmanifest.Manifest, error) {
manifestName := name
if !strings.HasSuffix(name, ".plist") {
manifestName = name + ".plist"
}
fname := filepath.Join(r.Path, manifestName)
file, err := os.Open(fname)
if err != nil {
return nil, errors.Wrapf(err, "reading manifest %s", name)
}
defer file.Close()
var m appmanifest.Manifest
if err := plist.NewDecoder(file).Decode(&m); err != nil {
return nil, errors.Wrap(err, "decoding manifest file")
}
return &m, nil
}
func (r *Repo) Apps(name string) (map[string]appmanifest.Manifest, error) {
manifests := make(map[string]appmanifest.Manifest)
if name != "" {
mf, err := os.Open(filepath.Join(r.Path, name))
if err != nil {
return nil, err
}
var m appmanifest.Manifest
if err := plist.NewDecoder(mf).Decode(&m); err != nil {
return nil, err
}
manifests[name] = m
return manifests, nil
}
files, err := ioutil.ReadDir(r.Path)
if err != nil {
return nil, err
}
for _, file := range files {
manifestName := file.Name()
if file.IsDir() || filepath.Ext(manifestName) != ".plist" {
continue
}
mf, err := os.Open(filepath.Join(r.Path, manifestName))
if err != nil {
return nil, err
}
var m appmanifest.Manifest
if err := plist.NewDecoder(mf).Decode(&m); err != nil {
return nil, err
}
manifests[manifestName] = m
}
return manifests, nil
}