list.go 4.5 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203
  1. package list
  2. import (
  3. "fmt"
  4. "io/ioutil"
  5. "log"
  6. "os"
  7. "path/filepath"
  8. "sync"
  9. "text/template"
  10. "time"
  11. "github.com/remogatto/cloud"
  12. karmen_client "gogs.carducci-dante.gov.it/karmen/client"
  13. "gogs.carducci-dante.gov.it/karmen/core/config"
  14. "gogs.carducci-dante.gov.it/karmen/core/generator/funcmap"
  15. "gogs.carducci-dante.gov.it/karmen/core/orm"
  16. "gogs.carducci-dante.gov.it/karmen/util/fileutil"
  17. "gogs.carducci-dante.gov.it/karmen/util/libreoffice"
  18. "gogs.carducci-dante.gov.it/karmen/util/pandoc"
  19. tpl_util "gogs.carducci-dante.gov.it/karmen/util/template"
  20. )
  21. type ListGenerator struct {
  22. sync.Mutex
  23. Config *config.ConfigT
  24. CloudPath string
  25. JobID uint
  26. ncClient *cloud.Client
  27. kaClient *karmen_client.Client
  28. logger *JobLogger
  29. }
  30. type JobLogger struct {
  31. job *orm.Job
  32. }
  33. func NewJobLogger(job *orm.Job) *JobLogger {
  34. return &JobLogger{job}
  35. }
  36. func (l *JobLogger) Write(p []byte) (n int, err error) {
  37. os.Stdout.Write(p)
  38. origLen := len(p)
  39. if len(p) > 0 && p[len(p)-1] == '\n' {
  40. p = p[:len(p)-1] // Cut terminating newline
  41. }
  42. l.job.Logs = append(l.job.Logs, &orm.Log{Content: string(p)})
  43. return origLen, nil
  44. }
  45. func (generator *ListGenerator) generate(outputPath string, cloudPath string, teachers []*orm.Teacher, funcMap template.FuncMap) {
  46. filename := filepath.Join(outputPath, "elenco_docenti")
  47. tplFilename := filepath.Join(cloudPath, "list.tpl.md")
  48. log.Printf("Downloading %s from the cloud...", tplFilename)
  49. tplContent, err := generator.ncClient.Download(tplFilename)
  50. if err != nil {
  51. log.Println(err)
  52. }
  53. log.Println("Download completed.")
  54. refContent, err := generator.ncClient.Download(filepath.Join(cloudPath, "reference.odt"))
  55. if err != nil {
  56. log.Println(err)
  57. }
  58. err = ioutil.WriteFile(filepath.Join(outputPath, "reference.odt"), refContent, 0777)
  59. if err != nil {
  60. log.Println(err)
  61. }
  62. tpl, err := tpl_util.LoadTextTemplateFromString(string(tplContent), funcMap)
  63. if err != nil {
  64. log.Println(err)
  65. }
  66. f, err := os.Create(filename + ".md")
  67. if err != nil {
  68. log.Println(err)
  69. }
  70. defer f.Close()
  71. err = tpl.Execute(f, teachers)
  72. if err != nil {
  73. log.Println(err)
  74. }
  75. odtFilename := fileutil.ReplaceExt(filename, "odt")
  76. log.Println("Generate", odtFilename)
  77. if err := pandoc.Convert(filename+".md", odtFilename, "--data-dir", outputPath); err != nil {
  78. log.Println(outputPath, err)
  79. }
  80. if err := libreoffice.Convert(filename+".odt", "pdf", "--outdir", outputPath); err != nil {
  81. log.Println(err)
  82. }
  83. }
  84. func NewListGenerator(config *config.ConfigT) *ListGenerator {
  85. return &ListGenerator{Config: config}
  86. }
  87. func (generator *ListGenerator) SetJobId(id uint) {
  88. generator.Lock()
  89. generator.JobID = id
  90. generator.Unlock()
  91. }
  92. func (generator *ListGenerator) Run() {
  93. var err error
  94. generator.ncClient, err = cloud.Dial(
  95. generator.Config.Cloud.Url,
  96. generator.Config.Cloud.Username,
  97. generator.Config.Cloud.Password,
  98. )
  99. if err != nil {
  100. log.Println(err)
  101. }
  102. log.Printf("Connecting to karmen...")
  103. generator.kaClient, err = karmen_client.Dial(
  104. generator.Config.Url,
  105. generator.Config.Admin.Username,
  106. generator.Config.Admin.Password,
  107. )
  108. if err != nil {
  109. log.Println(err)
  110. }
  111. var job *orm.Job
  112. if generator.JobID > 0 {
  113. job, err = generator.kaClient.GetJob(generator.JobID)
  114. if err != nil {
  115. log.Println(err)
  116. }
  117. generator.logger = NewJobLogger(job)
  118. log.SetOutput(generator.logger)
  119. job.Start = time.Now()
  120. err = generator.kaClient.UpdateJob(job)
  121. if err != nil {
  122. log.Println(err)
  123. }
  124. }
  125. teachers, err := generator.kaClient.GetTeachers()
  126. if err != nil {
  127. log.Println(err)
  128. }
  129. var (
  130. outputPath string
  131. cloudPath string
  132. )
  133. if generator.JobID > 0 {
  134. outputPath = filepath.Join(generator.Config.Documents.OutputPath, fmt.Sprintf("%d/%d", job.DocumentID, job.ID))
  135. cloudPath = job.Document.CloudPath
  136. } else {
  137. outputPath = "output"
  138. cloudPath = "Documents"
  139. }
  140. if err := os.MkdirAll(outputPath, 0777); err != nil {
  141. log.Println(err)
  142. }
  143. generator.generate(outputPath, cloudPath, teachers, funcmap.FuncMap)
  144. if generator.JobID > 0 {
  145. job.End = time.Now()
  146. job.Files = append(job.Files, &orm.File{Path: "elenco_docenti.pdf"})
  147. err = generator.kaClient.UpdateJob(job)
  148. if err != nil {
  149. log.Println(err)
  150. }
  151. }
  152. log.Println("Uploading files to the cloud...")
  153. src, err := ioutil.ReadFile(filepath.Join(outputPath, "elenco_docenti.pdf"))
  154. if err != nil {
  155. log.Println(err)
  156. }
  157. err = generator.ncClient.Upload(src, filepath.Join(cloudPath, "elenco_docenti.pdf"))
  158. if err != nil {
  159. log.Println(err)
  160. }
  161. log.Println("Files were uploaded to the cloud...")
  162. }