core/documents/generators/list/list.go

package list

import (
	"fmt"
	"log"
	"os"
	"strings"
	"text/template"

	karmen_client "gogs.carducci-dante.gov.it/karmen/client"
	"gogs.carducci-dante.gov.it/karmen/core/orm"
	"gogs.carducci-dante.gov.it/karmen/util/fileutil"
	"gogs.carducci-dante.gov.it/karmen/util/libreoffice"
	"gogs.carducci-dante.gov.it/karmen/util/pandoc"
	tpl_util "gogs.carducci-dante.gov.it/karmen/util/template"
)

type ListGenerator struct {
}

func NewListGenerator() *ListGenerator {
	return &ListGenerator{}
}

var funcMap template.FuncMap = template.FuncMap{
	"comma":        comma,
	"groupClasses": groupClasses,
	"abbrev":       abbrev,
	"nbsp":         nbsp,
}

func generate(teachers []*orm.Teacher, funcMap template.FuncMap) {
	filename := "output/elenco_docenti"
	tpl, err := tpl_util.LoadTextTemplate("./list.tpl.md", funcMap)
	if err != nil {
		panic(err)
	}

	f, err := os.Create(filename + ".md")
	if err != nil {
		panic(err)
	}
	defer f.Close()

	err = tpl.Execute(f, teachers)
	if err != nil {
		panic(err)
	}

	odtFilename := fileutil.ReplaceExt(filename, "odt")

	log.Println("Generate", odtFilename)

	if err := pandoc.Convert(filename+".md", odtFilename); err != nil {
		panic(err)
	}

	if err := libreoffice.Convert(filename+".odt", "pdf", "--outdir", "output"); err != nil {
		panic(err)
	}

}

func comma(classes []*orm.Class) string {
	var names []string
	for _, c := range classes {
		names = append(names, c.Name)
	}
	return strings.Join(names, ",")
}

func abbrev(name string) string {
	var result string
	splits := strings.Split(name, " ")
	for i := len(splits) - 1; i >= 0; i-- {
		if i == len(splits)-1 {
			result += splits[i] + " "
			continue
		}
		result += strings.ToUpper(string(splits[i][0]) + ".")
	}
	return result
}

func nbsp(num int, text string) string {
	return text + strings.Repeat(" ", num)
}

func groupClasses(classes []*orm.Class) string {
	var groups []string

	groupByAddresses := make(map[string][]string)
	addressesAbbrev := map[string]string{
		"Linguistico":       "LIN",
		"Classico":          "CLA",
		"Musicale":          "M",
		"Economico sociale": "ES",
		"Scienze umane":     "SU",
	}
	for _, c := range classes {
		groupByAddresses[c.Field] = append(groupByAddresses[c.Field], fmt.Sprintf("%d%s", c.Year, c.Section))
	}

	for address, classes := range groupByAddresses {
		var group string
		if len(classes) > 1 {
			group = fmt.Sprintf("[%s]%s", strings.Join(classes, ","), addressesAbbrev[address])
		} else {
			group = fmt.Sprintf("%s %s", strings.Join(classes, ","), addressesAbbrev[address])
		}
		groups = append(groups, group)
	}
	return strings.Join(groups, ",")
}

func (generator *ListGenerator) Run(done chan bool) {
	done <- true

	log.Printf("Connecting to karmen...")
	client, err := karmen_client.Dial(
		"https://karmen.carducci-dante.gov.it",
		"admin",
		"aolieVooju",
	)
	if err != nil {
		log.Println(err)
	}

	teachers, err := client.GetTeachers()
	if err != nil {
		panic(err)
	}

	outputDir := "output"

	if err := os.MkdirAll(outputDir, 0777); err != nil {
		panic(err)
	}

	generate(teachers, funcMap)
}