pkg/bufman/bufpkg/bufimage/image_file.go

// Licensed to the Apache Software Foundation (ASF) under one or more
// contributor license agreements.  See the NOTICE file distributed with
// this work for additional information regarding copyright ownership.
// The ASF licenses this file to You 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 bufimage

import (
	"google.golang.org/protobuf/types/descriptorpb"
)

import (
	"github.com/apache/dubbo-kubernetes/pkg/bufman/bufpkg/bufmodule/bufmoduleref"
	"github.com/apache/dubbo-kubernetes/pkg/bufman/pkg/protodescriptor"
)

var _ ImageFile = &imageFile{}

type imageFile struct {
	bufmoduleref.FileInfo

	fileDescriptorProto *descriptorpb.FileDescriptorProto

	isSyntaxUnspecified           bool
	storedUnusedDependencyIndexes []int32
}

func newImageFile(
	fileDescriptor protodescriptor.FileDescriptor,
	moduleIdentity bufmoduleref.ModuleIdentity,
	commit string,
	externalPath string,
	isImport bool,
	isSyntaxUnspecified bool,
	unusedDependencyIndexes []int32,
) (*imageFile, error) {
	if err := protodescriptor.ValidateFileDescriptor(fileDescriptor); err != nil {
		return nil, err
	}
	fileInfo, err := bufmoduleref.NewFileInfo(
		fileDescriptor.GetName(),
		externalPath,
		isImport,
		moduleIdentity,
		commit,
	)
	if err != nil {
		return nil, err
	}
	// just to normalize in other places between empty and unset
	if len(unusedDependencyIndexes) == 0 {
		unusedDependencyIndexes = nil
	}
	return &imageFile{
		FileInfo: fileInfo,
		// protodescriptor.FileDescriptorProtoForFileDescriptor is a no-op if fileDescriptor
		// is already a *descriptorpb.FileDescriptorProto
		fileDescriptorProto:           protodescriptor.FileDescriptorProtoForFileDescriptor(fileDescriptor),
		isSyntaxUnspecified:           isSyntaxUnspecified,
		storedUnusedDependencyIndexes: unusedDependencyIndexes,
	}, nil
}

func (f *imageFile) Proto() *descriptorpb.FileDescriptorProto {
	return f.fileDescriptorProto
}

func (f *imageFile) FileDescriptor() protodescriptor.FileDescriptor {
	return f.fileDescriptorProto
}

func (f *imageFile) IsSyntaxUnspecified() bool {
	return f.isSyntaxUnspecified
}

func (f *imageFile) UnusedDependencyIndexes() []int32 {
	return f.storedUnusedDependencyIndexes
}

func (f *imageFile) withIsImport(isImport bool) ImageFile {
	if f.IsImport() == isImport {
		return f
	}
	return &imageFile{
		FileInfo:                      f.FileInfo.WithIsImport(isImport),
		fileDescriptorProto:           f.fileDescriptorProto,
		isSyntaxUnspecified:           f.isSyntaxUnspecified,
		storedUnusedDependencyIndexes: f.storedUnusedDependencyIndexes,
	}
}

func (*imageFile) isImageFile() {}