8d5be2d657
* Document extractor service * Fixing vendor modules * Addressing PR Review comments * Some small simplifications * Fixing a linter complain * simplifying a bit the code using package variables Co-authored-by: Mattermod <mattermod@users.noreply.github.com>
70 строки
1.6 KiB
Go
70 строки
1.6 KiB
Go
package docconv
|
|
|
|
import (
|
|
"archive/zip"
|
|
"bytes"
|
|
"fmt"
|
|
"io"
|
|
"io/ioutil"
|
|
"time"
|
|
)
|
|
|
|
// ConvertODT converts a ODT file to text
|
|
func ConvertODT(r io.Reader) (string, map[string]string, error) {
|
|
meta := make(map[string]string)
|
|
var textBody string
|
|
|
|
b, err := ioutil.ReadAll(r)
|
|
if err != nil {
|
|
return "", nil, err
|
|
}
|
|
zr, err := zip.NewReader(bytes.NewReader(b), int64(len(b)))
|
|
if err != nil {
|
|
return "", nil, fmt.Errorf("error unzipping data: %v", err)
|
|
}
|
|
|
|
for _, f := range zr.File {
|
|
switch f.Name {
|
|
case "meta.xml":
|
|
rc, err := f.Open()
|
|
if err != nil {
|
|
return "", nil, fmt.Errorf("error extracting '%v' from archive: %v", f.Name, err)
|
|
}
|
|
defer rc.Close()
|
|
|
|
info, err := XMLToMap(rc)
|
|
if err != nil {
|
|
return "", nil, fmt.Errorf("error parsing '%v': %v", f.Name, err)
|
|
}
|
|
|
|
if tmp, ok := info["creator"]; ok {
|
|
meta["Author"] = tmp
|
|
}
|
|
if tmp, ok := info["date"]; ok {
|
|
if t, err := time.Parse("2006-01-02T15:04:05", tmp); err == nil {
|
|
meta["ModifiedDate"] = fmt.Sprintf("%d", t.Unix())
|
|
}
|
|
}
|
|
if tmp, ok := info["creation-date"]; ok {
|
|
if t, err := time.Parse("2006-01-02T15:04:05", tmp); err == nil {
|
|
meta["CreatedDate"] = fmt.Sprintf("%d", t.Unix())
|
|
}
|
|
}
|
|
|
|
case "content.xml":
|
|
rc, err := f.Open()
|
|
if err != nil {
|
|
return "", nil, fmt.Errorf("error extracting '%v' from archive: %v", f.Name, err)
|
|
}
|
|
defer rc.Close()
|
|
|
|
textBody, err = XMLToText(rc, []string{"br", "p", "tab"}, []string{}, true)
|
|
if err != nil {
|
|
return "", nil, fmt.Errorf("error parsing '%v': %v", f.Name, err)
|
|
}
|
|
}
|
|
}
|
|
|
|
return textBody, meta, nil
|
|
}
|