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
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
|
package managers
import (
"bytes"
"fmt"
"io"
"os"
"path/filepath"
"strings"
incus "github.com/lxc/incus/v6/shared/util"
"github.com/lxc/distrobuilder/shared"
)
type apt struct {
common
}
func (m *apt) load() error {
m.commands = managerCommands{
clean: "apt-get",
install: "apt-get",
refresh: "apt-get",
remove: "apt-get",
update: "apt-get",
}
m.flags = managerFlags{
clean: []string{
"clean",
},
global: []string{
"-y",
},
install: []string{
"install",
},
remove: []string{
"remove", "--auto-remove",
},
refresh: []string{
"update",
},
update: []string{
"dist-upgrade",
},
}
return nil
}
func (m *apt) manageRepository(repoAction shared.DefinitionPackagesRepository) error {
var targetFile string
if repoAction.Name == "sources.list" {
targetFile = filepath.Join("/etc/apt", repoAction.Name)
} else {
targetFile = filepath.Join("/etc/apt/sources.list.d", repoAction.Name)
if !strings.HasSuffix(targetFile, ".list") {
targetFile = fmt.Sprintf("%s.list", targetFile)
}
}
if !incus.PathExists(filepath.Dir(targetFile)) {
err := os.MkdirAll(filepath.Dir(targetFile), 0o755)
if err != nil {
return fmt.Errorf("Failed to create directory %q: %w", filepath.Dir(targetFile), err)
}
}
f, err := os.OpenFile(targetFile, os.O_CREATE|os.O_RDWR, 0o644)
if err != nil {
return fmt.Errorf("Failed to open file %q: %w", targetFile, err)
}
defer f.Close()
content, err := io.ReadAll(f)
if err != nil {
return fmt.Errorf("Failed to read from file %q: %w", targetFile, err)
}
// Truncate file if it's not generated by distrobuilder
if !strings.HasPrefix(string(content), "# Generated by distrobuilder\n") {
err = f.Truncate(0)
if err != nil {
return fmt.Errorf("Failed to truncate %q: %w", targetFile, err)
}
_, err = f.Seek(0, 0)
if err != nil {
return fmt.Errorf("Failed to seek on file %q: %w", targetFile, err)
}
_, err = f.WriteString("# Generated by distrobuilder\n")
if err != nil {
return fmt.Errorf("Failed to write to file %q: %w", targetFile, err)
}
}
_, err = f.WriteString(repoAction.URL)
if err != nil {
return fmt.Errorf("Failed to write to file %q: %w", targetFile, err)
}
// Append final new line if missing
if !strings.HasSuffix(repoAction.URL, "\n") {
_, err = f.WriteString("\n")
if err != nil {
return fmt.Errorf("Failed to write to file %q: %w", targetFile, err)
}
}
if repoAction.Key != "" {
var reader io.Reader
if strings.HasPrefix(repoAction.Key, "-----BEGIN PGP PUBLIC KEY BLOCK-----") {
reader = strings.NewReader(repoAction.Key)
} else {
// If only key ID is provided, we need gpg to be installed early.
err := shared.RunCommand(m.ctx, nil, nil, "gpg", "--recv-keys", repoAction.Key)
if err != nil {
return fmt.Errorf("Failed to receive GPG keys: %w", err)
}
var buf bytes.Buffer
err = shared.RunCommand(m.ctx, nil, &buf, "gpg", "--export", "--armor", repoAction.Key)
if err != nil {
return fmt.Errorf("Failed to export GPG keys: %w", err)
}
reader = &buf
}
signatureFilePath := filepath.Join("/etc/apt/trusted.gpg.d", fmt.Sprintf("%s.asc", repoAction.Name))
f, err := os.Create(signatureFilePath)
if err != nil {
return fmt.Errorf("Failed to create file %q: %w", signatureFilePath, err)
}
defer f.Close()
_, err = io.Copy(f, reader)
if err != nil {
return fmt.Errorf("Failed to copy file: %w", err)
}
}
return nil
}
|
