individual_dates.go

package html

import (
	"github.com/elliotchance/gedcom"
	"github.com/elliotchance/gedcom/html/core"
	"io"
)

type IndividualDates struct {
	individual *gedcom.IndividualNode
	visibility LivingVisibility
}

func NewIndividualDates(individual *gedcom.IndividualNode, visibility LivingVisibility) *IndividualDates {
	return &IndividualDates{
		individual: individual,
		visibility: visibility,
	}
}

func (c *IndividualDates) IsBlank() bool {
	eventDates := c.EventDates()

	for _, eventDate := range eventDates {
		if !eventDate.IsBlank() {
			return false
		}
	}

	return true
}

func (c *IndividualDates) WriteHTMLTo(w io.Writer) (int64, error) {
	isLiving := c.individual != nil && c.individual.IsLiving()

	if isLiving && c.visibility == LivingVisibilityHide {
		return writeNothing()
	}

	if isLiving && c.visibility == LivingVisibilityPlaceholder {
		return core.NewText("living").WriteHTMLTo(w)
	}

	eventDates := c.EventDates()

	return NewEventDates(eventDates).WriteHTMLTo(w)
}

func (c *IndividualDates) EventDates() []*EventDate {
	eventDates := []*EventDate{}

	// Use birth or fallback to baptism.
	births := c.individual.Births()
	baptisms := c.individual.Baptisms()
	switch {
	case len(births) > 0:
		eventDate := NewEventDate("b.", births[0].Dates())
		eventDates = append(eventDates, eventDate)

	case len(baptisms) > 0:
		eventDate := NewEventDate("bap.", baptisms[0].Dates())
		eventDates = append(eventDates, eventDate)
	}

	// Use death or fallback to burial.
	deaths := c.individual.Deaths()
	burials := c.individual.Burials()
	switch {
	case len(deaths) > 0:
		eventDate := NewEventDate("d.", deaths[0].Dates())
		eventDates = append(eventDates, eventDate)

	case len(burials) > 0:
		eventDate := NewEventDate("bur.", burials[0].Dates())
		eventDates = append(eventDates, eventDate)
	}

	return eventDates
}