123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282 |
- // Copyright 2022 The Gitea Authors. All rights reserved.
- // SPDX-License-Identifier: MIT
-
- package container
-
- import (
- "context"
- "strings"
- "time"
-
- "code.gitea.io/gitea/models/db"
- "code.gitea.io/gitea/models/packages"
- user_model "code.gitea.io/gitea/models/user"
- container_module "code.gitea.io/gitea/modules/packages/container"
- "code.gitea.io/gitea/modules/util"
-
- "xorm.io/builder"
- )
-
- var ErrContainerBlobNotExist = util.NewNotExistErrorf("container blob does not exist")
-
- type BlobSearchOptions struct {
- OwnerID int64
- Image string
- Digest string
- Tag string
- IsManifest bool
- Repository string
- }
-
- func (opts *BlobSearchOptions) toConds() builder.Cond {
- var cond builder.Cond = builder.Eq{
- "package.type": packages.TypeContainer,
- }
-
- if opts.OwnerID != 0 {
- cond = cond.And(builder.Eq{"package.owner_id": opts.OwnerID})
- }
- if opts.Image != "" {
- cond = cond.And(builder.Eq{"package.lower_name": strings.ToLower(opts.Image)})
- }
- if opts.Tag != "" {
- cond = cond.And(builder.Eq{"package_version.lower_version": strings.ToLower(opts.Tag)})
- }
- if opts.IsManifest {
- cond = cond.And(builder.Eq{"package_file.lower_name": ManifestFilename})
- }
- if opts.Digest != "" {
- var propsCond builder.Cond = builder.Eq{
- "package_property.ref_type": packages.PropertyTypeFile,
- "package_property.name": container_module.PropertyDigest,
- "package_property.value": opts.Digest,
- }
-
- cond = cond.And(builder.In("package_file.id", builder.Select("package_property.ref_id").Where(propsCond).From("package_property")))
- }
- if opts.Repository != "" {
- var propsCond builder.Cond = builder.Eq{
- "package_property.ref_type": packages.PropertyTypePackage,
- "package_property.name": container_module.PropertyRepository,
- "package_property.value": opts.Repository,
- }
-
- cond = cond.And(builder.In("package.id", builder.Select("package_property.ref_id").Where(propsCond).From("package_property")))
- }
-
- return cond
- }
-
- // GetContainerBlob gets the container blob matching the blob search options
- // If multiple matching blobs are found (manifests with the same digest) the first (according to the database) is selected.
- func GetContainerBlob(ctx context.Context, opts *BlobSearchOptions) (*packages.PackageFileDescriptor, error) {
- pfds, err := getContainerBlobsLimit(ctx, opts, 1)
- if err != nil {
- return nil, err
- }
- if len(pfds) != 1 {
- return nil, ErrContainerBlobNotExist
- }
-
- return pfds[0], nil
- }
-
- // GetContainerBlobs gets the container blobs matching the blob search options
- func GetContainerBlobs(ctx context.Context, opts *BlobSearchOptions) ([]*packages.PackageFileDescriptor, error) {
- return getContainerBlobsLimit(ctx, opts, 0)
- }
-
- func getContainerBlobsLimit(ctx context.Context, opts *BlobSearchOptions, limit int) ([]*packages.PackageFileDescriptor, error) {
- pfs := make([]*packages.PackageFile, 0, limit)
- sess := db.GetEngine(ctx).
- Join("INNER", "package_version", "package_version.id = package_file.version_id").
- Join("INNER", "package", "package.id = package_version.package_id").
- Where(opts.toConds())
-
- if limit > 0 {
- sess = sess.Limit(limit)
- }
-
- if err := sess.Find(&pfs); err != nil {
- return nil, err
- }
-
- return packages.GetPackageFileDescriptors(ctx, pfs)
- }
-
- // GetManifestVersions gets all package versions representing the matching manifest
- func GetManifestVersions(ctx context.Context, opts *BlobSearchOptions) ([]*packages.PackageVersion, error) {
- cond := opts.toConds().And(builder.Eq{"package_version.is_internal": false})
-
- pvs := make([]*packages.PackageVersion, 0, 10)
- return pvs, db.GetEngine(ctx).
- Join("INNER", "package", "package.id = package_version.package_id").
- Join("INNER", "package_file", "package_file.version_id = package_version.id").
- Where(cond).
- Find(&pvs)
- }
-
- // GetImageTags gets a sorted list of the tags of an image
- // The result is suitable for the api call.
- func GetImageTags(ctx context.Context, ownerID int64, image string, n int, last string) ([]string, error) {
- // Short circuit: n == 0 should return an empty list
- if n == 0 {
- return []string{}, nil
- }
-
- var cond builder.Cond = builder.Eq{
- "package.type": packages.TypeContainer,
- "package.owner_id": ownerID,
- "package.lower_name": strings.ToLower(image),
- "package_version.is_internal": false,
- }
-
- var propsCond builder.Cond = builder.Eq{
- "package_property.ref_type": packages.PropertyTypeVersion,
- "package_property.name": container_module.PropertyManifestTagged,
- }
-
- cond = cond.And(builder.In("package_version.id", builder.Select("package_property.ref_id").Where(propsCond).From("package_property")))
-
- if last != "" {
- cond = cond.And(builder.Gt{"package_version.lower_version": strings.ToLower(last)})
- }
-
- sess := db.GetEngine(ctx).
- Table("package_version").
- Select("package_version.lower_version").
- Join("INNER", "package", "package.id = package_version.package_id").
- Where(cond).
- Asc("package_version.lower_version")
-
- var tags []string
- if n > 0 {
- sess = sess.Limit(n)
-
- tags = make([]string, 0, n)
- } else {
- tags = make([]string, 0, 10)
- }
-
- return tags, sess.Find(&tags)
- }
-
- type ImageTagsSearchOptions struct {
- PackageID int64
- Query string
- IsTagged bool
- Sort packages.VersionSort
- db.Paginator
- }
-
- func (opts *ImageTagsSearchOptions) toConds() builder.Cond {
- var cond builder.Cond = builder.Eq{
- "package.type": packages.TypeContainer,
- "package.id": opts.PackageID,
- "package_version.is_internal": false,
- }
-
- if opts.Query != "" {
- cond = cond.And(builder.Like{"package_version.lower_version", strings.ToLower(opts.Query)})
- }
-
- var propsCond builder.Cond = builder.Eq{
- "package_property.ref_type": packages.PropertyTypeVersion,
- "package_property.name": container_module.PropertyManifestTagged,
- }
-
- in := builder.In("package_version.id", builder.Select("package_property.ref_id").Where(propsCond).From("package_property"))
-
- if opts.IsTagged {
- cond = cond.And(in)
- } else {
- cond = cond.And(builder.Not{in})
- }
-
- return cond
- }
-
- func (opts *ImageTagsSearchOptions) configureOrderBy(e db.Engine) {
- switch opts.Sort {
- case packages.SortVersionDesc:
- e.Desc("package_version.version")
- case packages.SortVersionAsc:
- e.Asc("package_version.version")
- case packages.SortCreatedAsc:
- e.Asc("package_version.created_unix")
- default:
- e.Desc("package_version.created_unix")
- }
- }
-
- // SearchImageTags gets a sorted list of the tags of an image
- func SearchImageTags(ctx context.Context, opts *ImageTagsSearchOptions) ([]*packages.PackageVersion, int64, error) {
- sess := db.GetEngine(ctx).
- Join("INNER", "package", "package.id = package_version.package_id").
- Where(opts.toConds())
-
- opts.configureOrderBy(sess)
-
- if opts.Paginator != nil {
- sess = db.SetSessionPagination(sess, opts)
- }
-
- pvs := make([]*packages.PackageVersion, 0, 10)
- count, err := sess.FindAndCount(&pvs)
- return pvs, count, err
- }
-
- // SearchExpiredUploadedBlobs gets all uploaded blobs which are older than specified
- func SearchExpiredUploadedBlobs(ctx context.Context, olderThan time.Duration) ([]*packages.PackageFile, error) {
- var cond builder.Cond = builder.Eq{
- "package_version.is_internal": true,
- "package_version.lower_version": UploadVersion,
- "package.type": packages.TypeContainer,
- }
- cond = cond.And(builder.Lt{"package_file.created_unix": time.Now().Add(-olderThan).Unix()})
-
- var pfs []*packages.PackageFile
- return pfs, db.GetEngine(ctx).
- Join("INNER", "package_version", "package_version.id = package_file.version_id").
- Join("INNER", "package", "package.id = package_version.package_id").
- Where(cond).
- Find(&pfs)
- }
-
- // GetRepositories gets a sorted list of all repositories
- func GetRepositories(ctx context.Context, actor *user_model.User, n int, last string) ([]string, error) {
- var cond builder.Cond = builder.Eq{
- "package.type": packages.TypeContainer,
- "package_property.ref_type": packages.PropertyTypePackage,
- "package_property.name": container_module.PropertyRepository,
- }
-
- cond = cond.And(builder.Exists(
- builder.
- Select("package_version.id").
- Where(builder.Eq{"package_version.is_internal": false}.And(builder.Expr("package.id = package_version.package_id"))).
- From("package_version"),
- ))
-
- if last != "" {
- cond = cond.And(builder.Gt{"package_property.value": strings.ToLower(last)})
- }
-
- if actor.IsGhost() {
- actor = nil
- }
-
- cond = cond.And(user_model.BuildCanSeeUserCondition(actor))
-
- sess := db.GetEngine(ctx).
- Table("package").
- Select("package_property.value").
- Join("INNER", "user", "`user`.id = package.owner_id").
- Join("INNER", "package_property", "package_property.ref_id = package.id").
- Where(cond).
- Asc("package_property.value").
- Limit(n)
-
- repositories := make([]string, 0, n)
- return repositories, sess.Find(&repositories)
- }
|