photoprism/internal/entity/country.go

package entity

import (
	"github.com/gosimple/slug"
	"github.com/jinzhu/gorm"
	"github.com/photoprism/photoprism/internal/maps"
)

var altCountryNames = map[string]string{
	"United States of America": "USA",
	"United States":            "USA",
	"":                         "Unknown",
}

type Country struct {
	ID                 string `gorm:"primary_key"`
	CountrySlug        string
	CountryName        string
	CountryDescription string `gorm:"type:text;"`
	CountryNotes       string `gorm:"type:text;"`
	CountryPhoto       *Photo
	CountryPhotoID     uint
	New                bool `gorm:"-"`
}

var UnknownCountry = NewCountry("zz", maps.CountryNames["zz"])

func CreateUnknownCountry(db *gorm.DB) {
	UnknownCountry.FirstOrCreate(db)
}

// Create a new country
func NewCountry(countryCode string, countryName string) *Country {
	if countryCode == "" {
		countryCode = "zz"
	}

	if altName, ok := altCountryNames[countryName]; ok {
		countryName = altName
	}

	countrySlug := slug.MakeLang(countryName, "en")

	result := &Country{
		ID:          countryCode,
		CountryName: countryName,
		CountrySlug: countrySlug,
	}

	return result
}

func (m *Country) FirstOrCreate(db *gorm.DB) *Country {
	if err := db.FirstOrCreate(m, "id = ?", m.ID).Error; err != nil {
		log.Errorf("country: %s", err)
	}

	return m
}

func (m *Country) AfterCreate(scope *gorm.Scope) error {
	return scope.SetColumn("New", true)
}

func (m *Country) Code() string {
	return m.ID
}

func (m *Country) Name() string {
	return m.CountryName
}