bjoernager/forgejo
bjoernager/forgejo
1
Fork 0
Code Activity

Add support for indexing arch files

Browse source
This commit is contained in:
dragon 2024-12-06 14:51:49 +08:00
parent c287a15996
commit 37a300b105
5 changed files with 96 additions and 29 deletions
Download patch file Download diff file Expand all files Collapse all files

25
modules/packages/arch/metadata.go
View file

@ -4,6 +4,7 @@
package arch
import (
"archive/tar"
"bufio"
"bytes"
"encoding/hex"
@ -25,7 +26,9 @@ import (
// https://man.archlinux.org/man/PKGBUILD.5
const (
PropertyDescription = "arch.description"
PropertyDescription = "arch.description"
PropertyFiles = "arch.files"
PropertyArch = "arch.architecture"
PropertyDistribution = "arch.distribution"
@ -85,6 +88,8 @@ type FileMetadata struct {
Packager string `json:"packager"`
Arch string `json:"arch"`
PgpSigned string `json:"pgp"`
Files []string `json:"files,omitempty"`
}
// ParsePackage Function that receives arch package archive data and returns it's metadata.
@ -127,6 +132,8 @@ func ParsePackage(r *packages.HashedBuffer) (*Package, error) {
var pkg *Package
var mTree bool
files := make([]string, 0)
for {
f, err := tarball.Read()
if err == io.EOF {
@ -135,6 +142,11 @@ func ParsePackage(r *packages.HashedBuffer) (*Package, error) {
if err != nil {
return nil, err
}
// ref:https://gitlab.archlinux.org/pacman/pacman/-/blob/91546004903eea5d5267d59898a6029ba1d64031/lib/libalpm/add.c#L529-L533
if !strings.HasPrefix(f.Name(), ".") {
files = append(files, (f.Header.(*tar.Header)).Name)
}
switch f.Name() {
case ".PKGINFO":
pkg, err = ParsePackageInfo(tarballType, f)
@ -155,7 +167,7 @@ func ParsePackage(r *packages.HashedBuffer) (*Package, error) {
if !mTree {
return nil, util.NewInvalidArgumentErrorf(".MTREE file not found")
}
pkg.FileMetadata.Files = files
pkg.FileMetadata.CompressedSize = r.Size()
pkg.FileMetadata.MD5 = hex.EncodeToString(md5)
pkg.FileMetadata.SHA256 = hex.EncodeToString(sha256)
@ -339,3 +351,12 @@ func (p *Package) Desc() string {
}
return buf.String()
}
func (p *Package) Files() string {
var buf bytes.Buffer
buf.WriteString("%FILES%\n")
for _, item := range p.FileMetadata.Files {
_, _ = fmt.Fprintf(&buf, "%s\n", item)
}
return buf.String()
}

14
modules/packages/arch/metadata_test.go
View file

@ -344,8 +344,8 @@ func TestValidatePackageSpec(t *testing.T) {
})
}
func TestDescString(t *testing.T) {
const pkgdesc = `%FILENAME%
func TestDescAndFileString(t *testing.T) {
const pkgDesc = `%FILENAME%
zstd-1.5.5-1-x86_64.pkg.tar.zst
%NAME%
@ -415,6 +415,12 @@ ninja
dummy5
dummy6
`
const pkgFiles = `%FILES%
usr/
usr/bin/
usr/bin/zstd
`
md := &Package{
@ -441,7 +447,9 @@ dummy6
BuildDate: 1681646714,
Packager: "Jelle van der Waa <jelle@archlinux.org>",
Arch: "x86_64",
Files: []string{"usr/", "usr/bin/", "usr/bin/zstd"},
},
}
require.Equal(t, pkgdesc, md.Desc())
require.Equal(t, pkgDesc, md.Desc())
require.Equal(t, pkgFiles, md.Files())
}