karmen/generator/generators/department/department.go

package department

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

	"github.com/remogatto/cloud"

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

type Data struct {
	Departments []*orm.Department
}

type DepartmentGenerator struct {
	sync.Mutex
	Config *generator.Config

	JobID uint

	Data *Data

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

func (generator *DepartmentGenerator) generate(outputPath string, cloudPath string, funcMap template.FuncMap) ([]*orm.File, error) {
	var files []*orm.File

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

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

	refFilename := filepath.Join(cloudPath, "reference.odt")
	log.Printf("Downloading %s from the cloud...", refFilename)
	refContent, err := generator.ncClient.Download(refFilename)
	if err != nil {
		return nil, err
	}
	log.Println("Download completed.")

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

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

	for _, department := range generator.Data.Departments {
		filename := filepath.Join(outputPath, fmt.Sprintf("Dipartimento di %s", department.Name))
		f, err := os.Create(filename + ".md")
		if err != nil {
			return nil, err
		}
		defer f.Close()

		err = tpl.Execute(f, department)
		if err != nil {
			return nil, 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 {
			return nil, err
		}

		files = append(files, &orm.File{Path: filepath.Base(filename) + ".pdf"})
	}

	return files, nil

}

func NewDepartmentGenerator(config *generator.Config) *DepartmentGenerator {
	return &DepartmentGenerator{
		Config: config,
		Data:   new(Data),
	}
}

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

func (generator *DepartmentGenerator) 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 = logger.NewJobLogger(job)
		log.SetOutput(generator.logger)

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

	generator.Data.Departments, err = generator.kaClient.GetDepartments()
	if err != nil {
		log.Println(err)
	}

	var (
		outputPath string
		cloudPath  string
	)

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

	log.Println("Output dir", outputPath)

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

	files, err := generator.generate(outputPath, cloudPath, funcmap.FuncMap)
	if err != nil {
		log.Println(err)
	}

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

		job.Files = append(job.Files, files...)

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

	var keepArtifacts bool

	if job != nil {
		keepArtifacts = job.Document.KeepArtifacts
	} else {
		keepArtifacts = generator.Config.KeepArtifacts
	}

	if !keepArtifacts {
		log.Println("Uploading resulting files to the cloud...")
		_, err = generator.ncClient.UploadDir(filepath.Join(outputPath, "*.pdf"), cloudPath)
		if err != nil {
			log.Println(err)
		}
	} else {
		log.Println("Uploading resulting files and artifacts to the cloud...")
		_, err = generator.ncClient.UploadDir(filepath.Join(outputPath, "*"), cloudPath)
		if err != nil {
			log.Println(err)
		}
	}
	log.Println("Files were uploaded to the cloud...")
}