file.go

package registry

import (
	"github.com/doozer-de/restgen/pbmap"
	"google.golang.org/protobuf/proto"
	"google.golang.org/protobuf/types/descriptorpb"
)

const OptionDartPackage = "dart_package"

type Files []*File

func (fs *Files) Add(f *File) {
	for _, c := range *fs {
		if c.Name == f.Name {
			return
		}
	}
	*fs = append(*fs, f)
}

func (fs *Files) Get(key string) (*File, bool) {
	for _, c := range *fs {
		if c.Name == key {
			return c, true
		}
	}
	return nil, false
}

type File struct {
	Type     *descriptorpb.FileDescriptorProto
	Name     string
	Messages []*Message
	Enums    []*Enum
	Package  string
	Registry *Registry
	// Options holds additional options. For example our dart_package options
	Options map[string]string
}

func NewFile(f *descriptorpb.FileDescriptorProto, r *Registry) *File {
	file := &File{
		Type:     f,
		Name:     *f.Name,
		Package:  f.GetPackage(),
		Registry: r,
		Options:  getAdditionalOptions(f),
	}

	for j, m := range f.MessageType {
		file.Messages = append(file.Messages, NewMessage(m, file, j))
	}

	for j, e := range f.EnumType {
		file.Enums = append(file.Enums, NewEnum(e, file, j))
	}

	return file
}

func getAdditionalOptions(f *descriptorpb.FileDescriptorProto) map[string]string {
	m := make(map[string]string)

	if dp, ok := dartPackageOption(f); ok {
		m[OptionDartPackage] = dp
	}

	return m
}

func dartPackageOption(f *descriptorpb.FileDescriptorProto) (string, bool) {
	opt := f.GetOptions()
	if opt == nil {
		return "", false
	}

	if !proto.HasExtension(opt, pbmap.E_DartPackage) {
		return "", false
	}
	ext := proto.GetExtension(opt, pbmap.E_DartPackage)
	sm, ok := ext.(*string)
	if !ok {
		return "", false
	}

	return *sm, true
}