forked from tgulacsi/agostle
/
mail_actions.go
140 lines (128 loc) · 3.69 KB
/
mail_actions.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
// Copyright 2013 The Agostle Authors. All rights reserved.
// Use of this source code is governed by an Apache 2.0
// license that can be found in the LICENSE file.
package main
import (
"io"
"os"
"strings"
"github.com/spf13/cobra"
"github.com/tgulacsi/agostle/converter"
)
func mailToPdfZip(outfn, inpfn string, splitted bool, outimg string, imgsize string) error {
input, err := openIn(inpfn)
if err != nil {
return err
}
defer func() { _ = input.Close() }()
if !splitted && outimg == "" {
return converter.MailToPdfZip(outfn, input, "message/rfc822", nil)
}
return converter.MailToSplittedPdfZip(outfn, input, "message/rfc822",
splitted, outimg, imgsize, nil)
}
func mailToTree(outdir, inpfn string) error {
input, err := openIn(inpfn)
if err != nil {
return err
}
if strings.HasSuffix(outdir, ".zip") && !isDir(outdir) {
return converter.MailToZip(outdir, input, "message/rfc822", nil)
}
return converter.MailToTree(outdir, input)
}
func outlookToEmail(outfn, inpfn string) error {
inp, err := openIn(inpfn)
if err != nil {
return err
}
out, err := openOut(outfn)
if err != nil {
return err
}
defer func() {
if closeErr := out.Close(); closeErr != nil {
logger.Error("msg", "Close", "error", err)
if err == nil {
err = closeErr
}
}
}()
var r io.ReadCloser
r, err = converter.NewOLEStorageReader(inp)
if err != nil {
return err
}
_, err = io.Copy(out, r)
_ = r.Close()
return err
}
func init() {
var out string
{
var (
split bool
outimg string
imgsize = "640x640"
)
mailToPdfZipCmd := &cobra.Command{
Use: "mail",
Short: "convert mail to zip of PDFs",
Long: `reads a message/rfc822 email, converts all of it to PDF files
(including attachments), and outputs a zip file containing these pdfs,
optionally splits the PDFs to separate pages, and converts these pages to images.
Usage:
mail2pdfzip [-split] [-outimg=image/gif] [-imgsize=640x640] mailfile.eml
Examples:
mail2pdfzip -split --outimg=image/gif --imgsize=800x800 -o=/tmp/email.pdf.zip email.eml
`,
Aliases: []string{"mail2pdfzip", "mail", "mailToPdfZip"},
Run: func(cmd *cobra.Command, args []string) {
fn := inpFromArgs(args)
if err := mailToPdfZip(out, fn, split, outimg, imgsize); err != nil {
logger.Error("msg", "mailToPdfZip", "out", out, "error", err)
os.Exit(1)
}
},
}
f := mailToPdfZipCmd.Flags()
f.StringVarP(&out, "out", "o", "", "output file")
f.BoolVar(&split, "split", false, "split PDF to pages")
f.StringVar(&outimg, "outimg", "", "output image format")
f.StringVar(&imgsize, "imgsize", "640x640", "image size")
agostleCmd.AddCommand(mailToPdfZipCmd)
}
mailToTreeCmd := &cobra.Command{
Use: "mail2tree",
Short: "extract mail tree to a directory",
Run: func(cmd *cobra.Command, args []string) {
fn := inpFromArgs(args)
if err := mailToTree(out, fn); err != nil {
logger.Error("msg", "mailToTree", "out", out, "inp", fn, "error", err)
os.Exit(1)
}
},
}
mailToTreeCmd.Flags().StringVar(&out, "o", "", "output file")
agostleCmd.AddCommand(mailToTreeCmd)
outlookToEmailCmd := &cobra.Command{
Use: "outlook2email",
Short: "convert outlook .msg to standard .eml",
Long: "uses libemail-outlook-message-perl if installed, or docker to install && run that script",
Run: func(cmd *cobra.Command, args []string) {
fn := inpFromArgs(args)
if err := outlookToEmail(out, fn); err != nil {
logger.Error("msg", "outlookToEmail", "out", out, "inp", fn, "error", err)
os.Exit(1)
}
},
}
outlookToEmailCmd.Flags().StringVarP(&out, "out", "o", "", "output file")
agostleCmd.AddCommand(outlookToEmailCmd)
}
func inpFromArgs(args []string) string {
if len(args) > 0 {
return args[0]
}
return "-"
}