parent
6f9cb0434e
commit
f415963f24
@ -1,5 +0,0 @@
|
|||||||
module snap-seed-glue
|
|
||||||
|
|
||||||
go 1.22.1
|
|
||||||
|
|
||||||
require github.com/snapcore/snapd v0.0.0-20240328101726-fdc222fc37a0
|
|
||||||
@ -1,289 +0,0 @@
|
|||||||
package main
|
|
||||||
|
|
||||||
// Copyright (C) 2024 Simon Quigley <tsimonq2@ubuntu.com>
|
|
||||||
//
|
|
||||||
// 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.
|
|
||||||
|
|
||||||
import (
|
|
||||||
"flag"
|
|
||||||
"fmt"
|
|
||||||
"io/ioutil"
|
|
||||||
"log"
|
|
||||||
"os"
|
|
||||||
"os/exec"
|
|
||||||
"path/filepath"
|
|
||||||
"strings"
|
|
||||||
|
|
||||||
"github.com/snapcore/snapd/snap"
|
|
||||||
"github.com/snapcore/snapd/interfaces/builtin"
|
|
||||||
"gopkg.in/yaml.v2"
|
|
||||||
)
|
|
||||||
|
|
||||||
type seed struct {
|
|
||||||
Snaps []struct {
|
|
||||||
Name string `yaml:"name"`
|
|
||||||
Channel string `yaml:"channel"`
|
|
||||||
File string `yaml:"file"`
|
|
||||||
} `yaml:"snaps"`
|
|
||||||
}
|
|
||||||
|
|
||||||
func main() {
|
|
||||||
snap.SanitizePlugsSlots = builtin.SanitizePlugsSlots
|
|
||||||
|
|
||||||
var seed_directory string
|
|
||||||
flag.StringVar(&seed_directory, "seed", "/var/lib/snapd/seed", "Specify the seed directory")
|
|
||||||
flag.Parse()
|
|
||||||
|
|
||||||
snap_set := make(map[string]bool)
|
|
||||||
|
|
||||||
snaps_dir := filepath.Join(seed_directory, "snaps")
|
|
||||||
assertions_dir := filepath.Join(seed_directory, "assertions")
|
|
||||||
seed_yaml := filepath.Join(seed_directory, "seed.yaml")
|
|
||||||
|
|
||||||
ensure_seed_yaml(seed_yaml)
|
|
||||||
|
|
||||||
existing_snaps_in_yaml := load_existing_snaps(seed_yaml)
|
|
||||||
|
|
||||||
for _, snap_info := range flag.Args() {
|
|
||||||
parts := strings.SplitN(snap_info, "=", 2)
|
|
||||||
snap_name := parts[0]
|
|
||||||
channel := "stable" // Default to stable if no channel is specified
|
|
||||||
if len(parts) == 2 {
|
|
||||||
channel = parts[1]
|
|
||||||
}
|
|
||||||
process_snap_with_prereqs(snap_name, channel, &snap_set, snaps_dir, assertions_dir, seed_yaml, existing_snaps_in_yaml)
|
|
||||||
}
|
|
||||||
|
|
||||||
essentialSnaps := []string{"snapd", "bare"}
|
|
||||||
for _, snapName := range essentialSnaps {
|
|
||||||
if !existing_snaps_in_yaml[snapName] {
|
|
||||||
process_snap_with_prereqs(snapName, "stable", &snap_set, snaps_dir, assertions_dir, seed_yaml, existing_snaps_in_yaml)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
update_seed_yaml(snaps_dir, seed_yaml, snap_set, existing_snaps_in_yaml)
|
|
||||||
|
|
||||||
remove_state_json(filepath.Join(seed_directory, "..", "state.json"))
|
|
||||||
ensure_assertions(assertions_dir)
|
|
||||||
validate_seed(seed_yaml)
|
|
||||||
}
|
|
||||||
|
|
||||||
func ensure_seed_yaml(seed_yaml string) {
|
|
||||||
if _, err := os.Stat(seed_yaml); os.IsNotExist(err) {
|
|
||||||
file, err := os.Create(seed_yaml)
|
|
||||||
if err != nil {
|
|
||||||
log.Fatalf("Failed to create seed.yaml: %v", err)
|
|
||||||
}
|
|
||||||
defer file.Close()
|
|
||||||
file.WriteString("snaps:\n")
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func load_existing_snaps(seed_yaml string) map[string]bool {
|
|
||||||
file, err := ioutil.ReadFile(seed_yaml)
|
|
||||||
if err != nil {
|
|
||||||
log.Fatalf("Failed to read seed.yaml: %v", err)
|
|
||||||
}
|
|
||||||
|
|
||||||
var seed_data seed
|
|
||||||
if err := yaml.Unmarshal(file, &seed_data); err != nil {
|
|
||||||
log.Fatalf("Failed to parse seed.yaml: %v", err)
|
|
||||||
}
|
|
||||||
|
|
||||||
existing := make(map[string]bool)
|
|
||||||
for _, snap := range seed_data.Snaps {
|
|
||||||
existing[snap.Name] = true
|
|
||||||
}
|
|
||||||
return existing
|
|
||||||
}
|
|
||||||
|
|
||||||
func update_seed_yaml(snaps_dir, seed_yaml string, snap_set map[string]bool, existing_snaps map[string]bool) {
|
|
||||||
seed_data := load_seed_data(seed_yaml)
|
|
||||||
|
|
||||||
for snap_name := range snap_set {
|
|
||||||
if !existing_snaps[snap_name] {
|
|
||||||
snap_files, err := filepath.Glob(filepath.Join(snaps_dir, fmt.Sprintf("%s_*.snap", snap_name)))
|
|
||||||
if err != nil || len(snap_files) == 0 {
|
|
||||||
log.Printf("No snap file found for %s", snap_name)
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
snap_file := filepath.Base(snap_files[0])
|
|
||||||
log.Printf(snap_file)
|
|
||||||
|
|
||||||
// FIXME: should put the real name of the channel in here
|
|
||||||
seed_data.Snaps = append(seed_data.Snaps, struct {
|
|
||||||
Name string `yaml:"name"`
|
|
||||||
Channel string `yaml:"channel"`
|
|
||||||
File string `yaml:"file"`
|
|
||||||
}{snap_name, "latest/stable", snap_file})
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// Marshal to YAML and write back to file
|
|
||||||
data, err := yaml.Marshal(&seed_data)
|
|
||||||
if err != nil {
|
|
||||||
log.Fatalf("Failed to marshal seed data to YAML: %v", err)
|
|
||||||
}
|
|
||||||
|
|
||||||
if err := ioutil.WriteFile(seed_yaml, data, 0644); err != nil {
|
|
||||||
log.Fatalf("Failed to write updated seed.yaml: %v", err)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func load_seed_data(seed_yaml string) seed {
|
|
||||||
file, err := ioutil.ReadFile(seed_yaml)
|
|
||||||
if err != nil {
|
|
||||||
log.Fatalf("Failed to read seed.yaml: %v", err)
|
|
||||||
}
|
|
||||||
|
|
||||||
var seed_data seed
|
|
||||||
if err := yaml.Unmarshal(file, &seed_data); err != nil {
|
|
||||||
log.Fatalf("Failed to parse seed.yaml: %v", err)
|
|
||||||
}
|
|
||||||
return seed_data
|
|
||||||
}
|
|
||||||
|
|
||||||
func remove_state_json(state_json_path string) {
|
|
||||||
if _, err := os.Stat(state_json_path); err == nil {
|
|
||||||
os.Remove(state_json_path)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func validate_seed(seed_yaml string) {
|
|
||||||
cmd := exec.Command("snap", "debug", "validate-seed", seed_yaml)
|
|
||||||
if err := cmd.Run(); err != nil {
|
|
||||||
log.Printf("Error validating seed: %v", err)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func process_snap_with_prereqs(snap_name, channel string, snap_set *map[string]bool, snaps_dir, assertions_dir, seed_yaml string, existing_snaps_in_yaml map[string]bool) {
|
|
||||||
if (*snap_set)[snap_name] {
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
// Download the snap if not already processed or listed in seed.yaml
|
|
||||||
if !existing_snaps_in_yaml[snap_name] {
|
|
||||||
cmd := exec.Command("snap", "download", snap_name, "--channel="+channel, "--target-directory="+snaps_dir)
|
|
||||||
if err := cmd.Run(); err != nil {
|
|
||||||
log.Printf("Error downloading snap %s from channel %s: %v", snap_name, channel, err)
|
|
||||||
return
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
snap_files, err := filepath.Glob(filepath.Join(snaps_dir, fmt.Sprintf("%s_*.snap", snap_name)))
|
|
||||||
if err != nil || len(snap_files) == 0 {
|
|
||||||
log.Printf("No snap file found for %s in channel %s", snap_name, channel)
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
snap_file := snap_files[0]
|
|
||||||
|
|
||||||
cmd := exec.Command("unsquashfs", "-n", "-d", filepath.Join(snaps_dir, fmt.Sprintf("%s_meta", snap_name)), snap_file, "meta/snap.yaml")
|
|
||||||
if err := cmd.Run(); err != nil {
|
|
||||||
log.Printf("Error extracting meta/snap.yaml from snap %s: %v", snap_name, err)
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
yaml_data, err := ioutil.ReadFile(filepath.Join(snaps_dir, fmt.Sprintf("%s_meta/meta/snap.yaml", snap_name)))
|
|
||||||
if err != nil {
|
|
||||||
log.Printf("Error reading snap.yaml file for %s: %v", snap_name, err)
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
info, err := snap.InfoFromSnapYaml(yaml_data)
|
|
||||||
if err != nil {
|
|
||||||
log.Printf("Error parsing snap.yaml data for %s: %v", snap_name, err)
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
(*snap_set)[snap_name] = true
|
|
||||||
|
|
||||||
tracker := snap.SimplePrereqTracker{}
|
|
||||||
missing_provider_content_tags := tracker.MissingProviderContentTags(info, nil)
|
|
||||||
for provider_snap := range missing_provider_content_tags {
|
|
||||||
if !(*snap_set)[provider_snap] {
|
|
||||||
process_snap_with_prereqs(provider_snap, "stable", snap_set, snaps_dir, assertions_dir, seed_yaml, existing_snaps_in_yaml)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
if info.Base != "" && !(*snap_set)[info.Base] {
|
|
||||||
process_snap_with_prereqs(info.Base, "stable", snap_set, snaps_dir, assertions_dir, seed_yaml, existing_snaps_in_yaml)
|
|
||||||
}
|
|
||||||
|
|
||||||
assert_files, err := filepath.Glob(filepath.Join(snaps_dir, "*.assert"))
|
|
||||||
for _, file := range assert_files {
|
|
||||||
target_path := filepath.Join(assertions_dir, filepath.Base(file))
|
|
||||||
err := os.Rename(file, target_path)
|
|
||||||
if err != nil {
|
|
||||||
log.Printf("Failed to move %s to %s: %v", file, assertions_dir, err)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
os.RemoveAll(filepath.Join(snaps_dir, fmt.Sprintf("%s_meta", snap_name)))
|
|
||||||
}
|
|
||||||
|
|
||||||
func ensure_assertions(assertions_dir string) {
|
|
||||||
model := "generic-classic"
|
|
||||||
brand := "generic"
|
|
||||||
series := "16"
|
|
||||||
|
|
||||||
model_assertion_path := filepath.Join(assertions_dir, "model")
|
|
||||||
account_key_assertion_path := filepath.Join(assertions_dir, "account-key")
|
|
||||||
account_assertion_path := filepath.Join(assertions_dir, "account")
|
|
||||||
|
|
||||||
// Check and generate model assertion
|
|
||||||
if _, err := os.Stat(model_assertion_path); os.IsNotExist(err) {
|
|
||||||
output, err := exec.Command("snap", "known", "--remote", "model", "series="+series, "model="+model, "brand-id="+brand).CombinedOutput()
|
|
||||||
if err != nil {
|
|
||||||
log.Fatalf("Failed to fetch model assertion: %v, Output: %s", err, string(output))
|
|
||||||
}
|
|
||||||
ioutil.WriteFile(model_assertion_path, output, 0644)
|
|
||||||
}
|
|
||||||
|
|
||||||
// Generate account-key assertion if not exists
|
|
||||||
if _, err := os.Stat(account_key_assertion_path); os.IsNotExist(err) {
|
|
||||||
signKeySha3 := grep_pattern(model_assertion_path, "sign-key-sha3-384: ")
|
|
||||||
output, err := exec.Command("snap", "known", "--remote", "account-key", "public-key-sha3-384="+signKeySha3).CombinedOutput()
|
|
||||||
if err != nil {
|
|
||||||
log.Fatalf("Failed to fetch account-key assertion: %v, Output: %s", err, string(output))
|
|
||||||
}
|
|
||||||
ioutil.WriteFile(account_key_assertion_path, output, 0644)
|
|
||||||
}
|
|
||||||
|
|
||||||
// Generate account assertion if not exists
|
|
||||||
if _, err := os.Stat(account_assertion_path); os.IsNotExist(err) {
|
|
||||||
accountId := grep_pattern(account_key_assertion_path, "account-id: ")
|
|
||||||
output, err := exec.Command("snap", "known", "--remote", "account", "account-id="+accountId).CombinedOutput()
|
|
||||||
if err != nil {
|
|
||||||
log.Fatalf("Failed to fetch account assertion: %v, Output: %s", err, string(output))
|
|
||||||
}
|
|
||||||
ioutil.WriteFile(account_assertion_path, output, 0644)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func grep_pattern(filePath, pattern string) string {
|
|
||||||
content, err := ioutil.ReadFile(filePath)
|
|
||||||
if err != nil {
|
|
||||||
log.Fatalf("Failed to read from file %s: %v", filePath, err)
|
|
||||||
}
|
|
||||||
lines := strings.Split(string(content), "\n")
|
|
||||||
for _, line := range lines {
|
|
||||||
if strings.Contains(line, pattern) {
|
|
||||||
parts := strings.SplitN(line, ":", 2)
|
|
||||||
if len(parts) == 2 {
|
|
||||||
return strings.TrimSpace(parts[1])
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
log.Fatalf("Pattern %s not found in file %s", pattern, filePath)
|
|
||||||
return ""
|
|
||||||
}
|
|
||||||
Loading…
Reference in new issue