76d492f9ab
* MM-37186: Update dependencies The split client libraries were excluded from being upgraded. See: https://github.com/splitio/go-split-commons/issues/56 https://mattermost.atlassian.net/browse/MM-37186 ```release-note NONE ``` * Ignore staticcheck deprecation warnings ```release-note NONE ```
112 строки
2.3 KiB
Go
112 строки
2.3 KiB
Go
package docconv
|
|
|
|
import (
|
|
"bytes"
|
|
"fmt"
|
|
"io"
|
|
"io/ioutil"
|
|
"log"
|
|
"os"
|
|
"os/exec"
|
|
"time"
|
|
|
|
"github.com/richardlehane/mscfb"
|
|
"github.com/richardlehane/msoleps"
|
|
)
|
|
|
|
// ConvertDoc converts an MS Word .doc to text.
|
|
func ConvertDoc(r io.Reader) (string, map[string]string, error) {
|
|
f, err := NewLocalFile(r)
|
|
if err != nil {
|
|
return "", nil, fmt.Errorf("error creating local file: %v", err)
|
|
}
|
|
defer f.Done()
|
|
|
|
// Meta data
|
|
mc := make(chan map[string]string, 1)
|
|
go func() {
|
|
defer func() {
|
|
if e := recover(); e != nil {
|
|
log.Printf("panic when reading doc format: %v", e)
|
|
}
|
|
}()
|
|
|
|
meta := make(map[string]string)
|
|
|
|
doc, err := mscfb.New(f)
|
|
if err != nil {
|
|
log.Printf("ConvertDoc: could not read doc: %v", err)
|
|
}
|
|
|
|
props := msoleps.New()
|
|
for entry, err := doc.Next(); err == nil; entry, err = doc.Next() {
|
|
if msoleps.IsMSOLEPS(entry.Initial) {
|
|
if oerr := props.Reset(doc); oerr != nil {
|
|
log.Printf("ConvertDoc: could not reset props: %v", oerr)
|
|
break
|
|
}
|
|
|
|
for _, prop := range props.Property {
|
|
meta[prop.Name] = prop.String()
|
|
}
|
|
}
|
|
}
|
|
|
|
const defaultTimeFormat = "2006-01-02 15:04:05.999999999 -0700 MST"
|
|
|
|
// Convert parsed meta
|
|
if tmp, ok := meta["LastSaveTime"]; ok {
|
|
if t, err := time.Parse(defaultTimeFormat, tmp); err == nil {
|
|
meta["ModifiedDate"] = fmt.Sprintf("%d", t.Unix())
|
|
}
|
|
}
|
|
if tmp, ok := meta["CreateTime"]; ok {
|
|
if t, err := time.Parse(defaultTimeFormat, tmp); err == nil {
|
|
meta["CreatedDate"] = fmt.Sprintf("%d", t.Unix())
|
|
}
|
|
}
|
|
|
|
mc <- meta
|
|
}()
|
|
|
|
// Document body
|
|
bc := make(chan string, 1)
|
|
go func() {
|
|
|
|
// Save output to a file
|
|
outputFile, err := ioutil.TempFile("/tmp", "sajari-convert-")
|
|
if err != nil {
|
|
// TODO: Remove this.
|
|
log.Println("TempFile Out:", err)
|
|
return
|
|
}
|
|
defer os.Remove(outputFile.Name())
|
|
|
|
err = exec.Command("wvText", f.Name(), outputFile.Name()).Run()
|
|
if err != nil {
|
|
// TODO: Remove this.
|
|
log.Println("wvText:", err)
|
|
}
|
|
|
|
var buf bytes.Buffer
|
|
_, err = buf.ReadFrom(outputFile)
|
|
if err != nil {
|
|
// TODO: Remove this.
|
|
log.Println("wvText:", err)
|
|
}
|
|
|
|
bc <- buf.String()
|
|
}()
|
|
|
|
// TODO: Should errors in either of the above Goroutines stop things from progressing?
|
|
body := <-bc
|
|
meta := <-mc
|
|
|
|
// TODO: Check for errors instead of len(body) == 0?
|
|
if len(body) == 0 {
|
|
f.Seek(0, 0)
|
|
return ConvertDocx(f)
|
|
}
|
|
return body, meta, nil
|
|
}
|