forked from macdylan/sm2uploader
-
Notifications
You must be signed in to change notification settings - Fork 0
/
utils.go
94 lines (80 loc) · 1.79 KB
/
utils.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
package main
import (
"bufio"
"bytes"
"fmt"
"io"
"path/filepath"
"regexp"
"strings"
"github.com/macdylan/SMFix/fix"
)
type empty struct{}
func humanReadableSize(size int64) string {
const unit = 1024
if size < unit {
return fmt.Sprintf("%d B", size)
}
div, exp := int64(unit), 0
for n := size / unit; n >= unit; n /= unit {
div *= unit
exp++
}
return fmt.Sprintf("%.1f %cB", float64(size)/float64(div), "KMGTPE"[exp])
}
var reFilename = regexp.MustCompile(`^[\.\/\\~]+`)
func normalizedFilename(filename string) string {
return reFilename.ReplaceAllString(filename, "")
}
/*
func postProcessFile(file_path string) (out []byte, err error) {
var r *os.File
if r, err = os.Open(file_path); err != nil {
return
}
defer r.Close()
return postProcess(r)
}
*/
func postProcess(r io.Reader) (out []byte, err error) {
header, errfix := fix.ExtractHeader(r)
if h, ok := r.(io.ReadSeeker); ok {
h.Seek(0, 0)
}
gcodes := make([]string, 0, fix.Params.TotalLines+len(header)+128)
sc := bufio.NewScanner(r)
for sc.Scan() {
gcodes = append(gcodes, sc.Text())
}
if err = sc.Err(); err != nil {
return nil, err
}
if errfix == nil {
funcs := make([]func([]string) []string, 0, 4)
if !noTrim {
funcs = append(funcs, fix.GcodeTrimLines)
}
if !noShutoff {
funcs = append(funcs, fix.GcodeFixShutoff)
}
if !noPreheat {
funcs = append(funcs, fix.GcodeFixPreheat)
}
if !noReinforceTower {
funcs = append(funcs, fix.GcodeReinforceTower)
}
for _, fn := range funcs {
gcodes = fn(gcodes)
}
}
out = []byte(strings.Join(gcodes, "\n"))
if len(header) > 25 {
return append(bytes.Join(header, []byte("\n")), out...), nil
} else {
return out, errfix
}
}
func shouldBeFix(fpath string) bool {
ext := strings.ToLower(filepath.Ext(fpath))
return SmFixExtensions[ext]
}