elliotchance/gedcom

package html

import (
    "io"
    "sort"

    "github.com/elliotchance/gedcom/v39"
    "github.com/elliotchance/gedcom/v39/html/core"
)

// PlaceListPage lists all places.
type PlaceListPage struct {
    document          *gedcom.Document
    googleAnalyticsID string
    options           *PublishShowOptions
    indexLetters      []rune
    placesMap         map[string]*place
}

func NewPlaceListPage(document *gedcom.Document, googleAnalyticsID string, options *PublishShowOptions, indexLetters []rune, placesMap map[string]*place) *PlaceListPage {
    return &PlaceListPage{
        document:          document,
        googleAnalyticsID: googleAnalyticsID,
        options:           options,
        indexLetters:      indexLetters,
        placesMap:         placesMap,
    }
}

func (c *PlaceListPage) WriteHTMLTo(w io.Writer) (int64, error) {
    table := []core.Component{}

    // Get all countries
    countries := gedcom.NewStringSet()
    for _, place := range c.placesMap {
        countries.Add(place.country)
    }

    sortedPlaces := []*place{}
    for _, placeName := range c.placesMap {
        sortedPlaces = append(sortedPlaces, placeName)
    }

    sort.SliceStable(sortedPlaces, func(i, j int) bool {
        if sortedPlaces[i].country != sortedPlaces[j].country {
            return sortedPlaces[i].country < sortedPlaces[j].country
        }

        return sortedPlaces[i].PrettyName < sortedPlaces[j].PrettyName
    })

    lastCountry := ""

    for _, place := range sortedPlaces {
        if lastCountry != place.country {
            anchor := core.NewAnchor(place.country)
            heading := core.NewHeading(2, "", core.NewText(place.country))
            components := core.NewComponents(anchor, heading)
            cell := core.NewTableCell(components)
            countryTitle := core.NewTableRow(cell)
            table = append(table, countryTitle)
        }

        placeEntry := NewPlaceInList(c.document, place, c.placesMap)
        table = append(table, placeEntry)

        lastCountry = place.country
    }

    // Render
    pills := []core.Component{}
    for _, country := range countries.Strings() {
        pills = append(pills, core.NewNavLink(country, "#"+country, false))
    }

    return core.NewPage("Places", core.NewComponents(
        NewPublishHeader(c.document, "", selectedPlacesTab, c.options,
            c.indexLetters, c.placesMap),
        core.NewNavPillsRow(pills),
        core.NewSpace(),
        core.NewRow(
            core.NewColumn(core.EntireRow, core.NewTable("", table...)),
        ),
    ), c.googleAnalyticsID).WriteHTMLTo(w)
}