mirror of
https://github.com/mudler/LocalAI.git
synced 2026-11-06 12:33:16 -05:00
Ettore Di Giacinto
d537695895
Add the zip format of a package (a digest manifest plus the files of the OCI package for one platform) and an addonpkg zip subcommand that writes it from a verified layout, with a sha256 file. The zip comes from the same layout the registry package comes from, so there is one build path. Version comparison moves to contract.CompareVersions so the installer can use it for upgrades and downgrades. Assisted-by: Claude:claude-sonnet-5-5 go gofmt
65 lines
2.1 KiB
Go
65 lines
2.1 KiB
Go
// SPDX-License-Identifier: MIT
|
|
package packaging
|
|
|
|
import (
|
|
"bytes"
|
|
"crypto/sha256"
|
|
"errors"
|
|
"flag"
|
|
"fmt"
|
|
"os"
|
|
"path/filepath"
|
|
|
|
"github.com/mudler/LocalAI/pkg/addons/zipformat"
|
|
)
|
|
|
|
// cmdZip writes the zip of a verified platform layout, and its sha256 file.
|
|
// The files are those of the OCI package, so the zip is not a second build
|
|
// path: whatever the layout passed in verify is what the zip holds.
|
|
func cmdZip(args []string) error {
|
|
fs := flag.NewFlagSet("zip", flag.ContinueOnError)
|
|
dir := fs.String("layout", "", "platform OCI layout")
|
|
platform := fs.String("platform", "", "expected os/arch")
|
|
version := fs.String("version", "", "expected version")
|
|
config := fs.String("config", "addons/tables/package/platforms.yaml", "platform matrix file")
|
|
outDir := fs.String("out", "", "directory that receives <id>-<version>-<os>-<arch>.zip and its .sha256 file")
|
|
if err := fs.Parse(args); err != nil {
|
|
return err
|
|
}
|
|
if *outDir == "" || *dir == "" {
|
|
return errors.New("--layout and --out are required")
|
|
}
|
|
cfg, err := loadPlatforms(*config)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
v, err := verifyLeg(*dir, *platform, *version, cfg)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
entries := make([]zipformat.Entry, 0, len(v.Files))
|
|
for _, f := range v.Files {
|
|
entries = append(entries, zipformat.Entry{Name: f.name, Data: f.data, Exec: f.exec})
|
|
}
|
|
var buf bytes.Buffer
|
|
m := v.Manifest
|
|
if err := zipformat.Write(&buf, m.ID, m.Version, m.OS, m.Arch, m.Executable, entries); err != nil {
|
|
return err
|
|
}
|
|
if err := os.MkdirAll(*outDir, 0o750); err != nil {
|
|
return err
|
|
}
|
|
name := zipformat.FileName(m.ID, m.Version, m.OS, m.Arch)
|
|
zipPath := filepath.Join(*outDir, name)
|
|
if err := os.WriteFile(zipPath, buf.Bytes(), 0o644); err != nil { // #nosec G306 -- a distributable file
|
|
return err
|
|
}
|
|
sum := sha256.Sum256(buf.Bytes())
|
|
line := fmt.Sprintf("%x %s\n", sum, name)
|
|
if err := os.WriteFile(zipPath+".sha256", []byte(line), 0o644); err != nil { // #nosec G306 -- a distributable file
|
|
return err
|
|
}
|
|
_, _ = fmt.Fprintf(stdout, "zip=%s\nsha256=%x\nsize=%d\n", zipPath, sum, buf.Len())
|
|
return nil
|
|
}
|