karmen/generator/generators/list/list.go

package list

import (
	"fmt"
	"io/ioutil"
	"log"
	"os"
	"path/filepath"
	"strings"
	"sync"
	"text/template"
	"time"

	"github.com/remogatto/cloud"

	karmen_client "gogs.carducci-dante.gov.it/karmen/client"
	"gogs.carducci-dante.gov.it/karmen/core/config"
	"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 {
	sync.Mutex
	Config *config.ConfigT

	CloudPath string
	JobID     uint

	ncClient *cloud.Client
	kaClient *karmen_client.Client
	logger   *JobLogger
}

type JobLogger struct {
	job *orm.Job
}

func NewJobLogger(job *orm.Job) *JobLogger {
	return &JobLogger{job}
}

func (l *JobLogger) Write(p []byte) (n int, err error) {
	os.Stdout.Write(p)
	origLen := len(p)
	if len(p) > 0 && p[len(p)-1] == '\n' {
		p = p[:len(p)-1] // Cut terminating newline
	}
	l.job.Logs = append(l.job.Logs, &orm.Log{Content: string(p)})

	return origLen, nil
}

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

func (generator *ListGenerator) generate(outputPath string, cloudPath string, teachers []*orm.Teacher, funcMap template.FuncMap) {
	filename := filepath.Join(outputPath, "elenco_docenti")

	tplFilename := filepath.Join(cloudPath, "list.tpl.md")
	log.Printf("Downloading %s from the cloud...", tplFilename)

	tplContent, err := generator.ncClient.Download(tplFilename)
	if err != nil {
		log.Println(err)
	}
	log.Println("Download completed.")

	refContent, err := generator.ncClient.Download(filepath.Join(cloudPath, "reference.odt"))
	if err != nil {
		log.Println(err)
	}

	err = ioutil.WriteFile(filepath.Join(outputPath, "reference.odt"), refContent, 0777)
	if err != nil {
		log.Println(err)
	}

	tpl, err := tpl_util.LoadTextTemplateFromString(string(tplContent), funcMap)
	if err != nil {
		log.Println(err)
	}

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

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

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

	log.Println("Generate", odtFilename)

	if err := pandoc.Convert(filename+".md", odtFilename, "--data-dir", outputPath); err != nil {
		log.Println(outputPath, err)
	}

	if err := libreoffice.Convert(filename+".odt", "pdf", "--outdir", outputPath); err != nil {
		log.Println(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 NewListGenerator(config *config.ConfigT) *ListGenerator {
	return &ListGenerator{Config: config}
}

func (generator *ListGenerator) SetJobId(id uint) {
	generator.Lock()
	generator.JobID = id
	generator.Unlock()
}

func (generator *ListGenerator) Run() {
	var err error

	log.Printf("Connecting to the cloud...")
	generator.ncClient, err = cloud.Dial(
		generator.Config.Cloud.Url,
		generator.Config.Cloud.Username,
		generator.Config.Cloud.Password,
	)
	if err != nil {
		log.Println(err)
	}

	log.Printf("Connecting to karmen...")
	generator.kaClient, err = karmen_client.Dial(
		generator.Config.Url,
		generator.Config.Admin.Username,
		generator.Config.Admin.Password,
	)
	if err != nil {
		log.Println(err)
	}

	var job *orm.Job

	if generator.JobID > 0 {
		job, err = generator.kaClient.GetJob(generator.JobID)
		if err != nil {
			log.Println(err)
		}

		generator.logger = NewJobLogger(job)
		log.SetOutput(generator.logger)

		job.Start = time.Now()
		err = generator.kaClient.UpdateJob(job)
		if err != nil {
			log.Println(err)
		}
	}

	teachers, err := generator.kaClient.GetTeachers()
	if err != nil {
		log.Println(err)
	}

	var (
		outputPath string
		cloudPath  string
	)

	if generator.JobID > 0 {
		outputPath = filepath.Join(generator.Config.Documents.OutputPath, fmt.Sprintf("%d/%d", job.DocumentID, job.ID))
		cloudPath = job.Document.CloudPath
	} else {
		outputPath = "output"
		cloudPath = generator.CloudPath
	}

	if err := os.MkdirAll(outputPath, 0777); err != nil {
		log.Println(err)
	}

	generator.generate(outputPath, cloudPath, teachers, funcMap)

	if generator.JobID > 0 {
		job.End = time.Now()

		job.Files = append(job.Files, &orm.File{Path: "elenco_docenti.pdf"})

		err = generator.kaClient.UpdateJob(job)
		if err != nil {
			log.Println(err)
		}
	}

	log.Println("Uploading files to the cloud...")
	src, err := ioutil.ReadFile(filepath.Join(outputPath, "elenco_docenti.pdf"))
	if err != nil {
		log.Println(err)
	}
	err = generator.ncClient.Upload(src, filepath.Join(cloudPath, "elenco_docenti.pdf"))
	if err != nil {
		log.Println(err)
	}
	log.Println("Files were uploaded to the cloud...")
}