database.go

package ortfodb

import (
	"errors"

	jsoniter "github.com/json-iterator/go"
)

func LoadDatabase(at string, skipValidation bool) (database Database, err error) {
	json := jsoniter.ConfigFastest
	content, err := readFileBytes(at)
	if err != nil {
		return
	}
	if !skipValidation {
		validated, validationErrors, err := validateWithJSONSchema(string(content), DatabaseJSONSchema())
		if err != nil {
			return database, err
		}
		if !validated {
			DisplayValidationErrors(validationErrors, "database JSON")
			err = errors.New("database JSON is invalid")
			return database, err
		}
	}
	err = json.Unmarshal(content, &database)
	return
}

func (db Database) FindMedia(mediaEmbed Media, workID string) (found bool, media Media) {
	w, ok := db.FindWork(workID)

	if !ok {
		return false, Media{}
	}

	for _, wsl := range w.Content {
		for _, b := range wsl.Blocks {
			if b.Type == "media" && b.RelativeSource == mediaEmbed.RelativeSource {
				return true, b.Media
			}
		}
	}

	return false, Media{}
}

func (db Database) FindWork(idOrAlias string) (work Work, found bool) {
	work, found = db[idOrAlias]
	if !found {
		for _, w := range db {
			for _, alias := range w.Metadata.Aliases {
				if alias == idOrAlias {
					return w, true
				}
			}
		}
	}
	return
}

// FirstParagraph returns the first paragraph content block of the given work in the given language
func (work Work) FirstParagraph(lang string) (found bool, paragraph ContentBlock) {
	for _, block := range work.Content[lang].Blocks {
		if block.Type == "paragraph" {
			return true, block
		}
	}
	return
}