mirror of
https://github.com/goharbor/harbor
synced 2025-04-12 17:21:32 +00:00
107 lines
3.1 KiB
Go
107 lines
3.1 KiB
Go
// Copyright Project Harbor Authors
|
|
//
|
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
// you may not use this file except in compliance with the License.
|
|
// You may obtain a copy of the License at
|
|
//
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
// See the License for the specific language governing permissions and
|
|
// limitations under the License.
|
|
|
|
package cnai
|
|
|
|
import (
|
|
"context"
|
|
"encoding/json"
|
|
|
|
modelspec "github.com/CloudNativeAI/model-spec/specs-go/v1"
|
|
ocispec "github.com/opencontainers/image-spec/specs-go/v1"
|
|
|
|
ps "github.com/goharbor/harbor/src/controller/artifact/processor"
|
|
"github.com/goharbor/harbor/src/controller/artifact/processor/base"
|
|
"github.com/goharbor/harbor/src/controller/artifact/processor/cnai/parser"
|
|
"github.com/goharbor/harbor/src/lib/errors"
|
|
"github.com/goharbor/harbor/src/lib/log"
|
|
"github.com/goharbor/harbor/src/pkg/artifact"
|
|
)
|
|
|
|
// const definitions
|
|
const (
|
|
// ArtifactTypeCNAI defines the artifact type for CNAI model.
|
|
ArtifactTypeCNAI = "CNAI"
|
|
|
|
// AdditionTypeReadme defines the addition type readme for API.
|
|
AdditionTypeReadme = "README.MD"
|
|
// AdditionTypeLicense defines the addition type license for API.
|
|
AdditionTypeLicense = "LICENSE"
|
|
// AdditionTypeFiles defines the addition type files for API.
|
|
AdditionTypeFiles = "FILES"
|
|
)
|
|
|
|
func init() {
|
|
pc := &processor{
|
|
ManifestProcessor: base.NewManifestProcessor(),
|
|
}
|
|
|
|
if err := ps.Register(pc, modelspec.ArtifactTypeModelManifest); err != nil {
|
|
log.Errorf("failed to register processor for artifact type %s: %v", modelspec.ArtifactTypeModelManifest, err)
|
|
return
|
|
}
|
|
}
|
|
|
|
type processor struct {
|
|
*base.ManifestProcessor
|
|
}
|
|
|
|
func (p *processor) AbstractAddition(ctx context.Context, artifact *artifact.Artifact, addition string) (*ps.Addition, error) {
|
|
var additionParser parser.Parser
|
|
switch addition {
|
|
case AdditionTypeReadme:
|
|
additionParser = parser.NewReadme(p.RegCli)
|
|
case AdditionTypeLicense:
|
|
additionParser = parser.NewLicense(p.RegCli)
|
|
case AdditionTypeFiles:
|
|
additionParser = parser.NewFiles(p.RegCli)
|
|
default:
|
|
return nil, errors.New(nil).WithCode(errors.BadRequestCode).
|
|
WithMessagef("addition %s isn't supported for %s", addition, ArtifactTypeCNAI)
|
|
}
|
|
|
|
mf, _, err := p.RegCli.PullManifest(artifact.RepositoryName, artifact.Digest)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
_, payload, err := mf.Payload()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
manifest := &ocispec.Manifest{}
|
|
if err := json.Unmarshal(payload, manifest); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
contentType, content, err := additionParser.Parse(ctx, artifact, manifest)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return &ps.Addition{
|
|
ContentType: contentType,
|
|
Content: content,
|
|
}, nil
|
|
}
|
|
|
|
func (p *processor) GetArtifactType(_ context.Context, _ *artifact.Artifact) string {
|
|
return ArtifactTypeCNAI
|
|
}
|
|
|
|
func (p *processor) ListAdditionTypes(_ context.Context, _ *artifact.Artifact) []string {
|
|
return []string{AdditionTypeReadme, AdditionTypeLicense, AdditionTypeFiles}
|
|
}
|