Add modes
This commit is contained in:
parent
ca5dee2467
commit
ccdc004365
91
main.go
91
main.go
|
@ -10,6 +10,7 @@ import (
|
||||||
"path"
|
"path"
|
||||||
"path/filepath"
|
"path/filepath"
|
||||||
"runtime"
|
"runtime"
|
||||||
|
"strconv"
|
||||||
"strings"
|
"strings"
|
||||||
)
|
)
|
||||||
|
|
||||||
|
@ -26,6 +27,47 @@ func readText() (text string, err error) {
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// One QR Code for each line of a file
|
||||||
|
const modeLine int = 1
|
||||||
|
|
||||||
|
// One QR Code for each text file
|
||||||
|
const modeFile int = 2
|
||||||
|
|
||||||
|
func getMode() int {
|
||||||
|
fmt.Println("Availabl Modes:")
|
||||||
|
fmt.Println("1: Generate one QR-Code for each line of text in each text file")
|
||||||
|
fmt.Println("2: Generate one QR-Code for each text file")
|
||||||
|
fmt.Println("What mode should the tool use? [1]")
|
||||||
|
modeStr, err := readText()
|
||||||
|
if err != nil {
|
||||||
|
handleErr("Could not get mode: %s", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
modeStr = strings.Trim(modeStr, "\n")
|
||||||
|
modeStr = strings.Trim(modeStr, "\r")
|
||||||
|
|
||||||
|
switch modeStr {
|
||||||
|
case "":
|
||||||
|
fallthrough
|
||||||
|
case "1":
|
||||||
|
return modeLine
|
||||||
|
case "2":
|
||||||
|
return modeFile
|
||||||
|
default:
|
||||||
|
handleErr("Unknown mode '%s'", modeStr)
|
||||||
|
}
|
||||||
|
|
||||||
|
return 0
|
||||||
|
}
|
||||||
|
|
||||||
|
func createQrCode(filenameWithoutSuffix, outPath, content string) {
|
||||||
|
outfile := path.Join(outPath, filenameWithoutSuffix+".png")
|
||||||
|
err := qrcode.WriteFile(content, qrcode.Low, 2048, outfile)
|
||||||
|
if err != nil {
|
||||||
|
handleErr("Could not create qr code for file '%s.txt': %s", filenameWithoutSuffix, err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
func main() {
|
func main() {
|
||||||
|
|
||||||
fmt.Printf("Bulk QR Code generate tool, Version %s\n\r", VersionNum)
|
fmt.Printf("Bulk QR Code generate tool, Version %s\n\r", VersionNum)
|
||||||
|
@ -44,8 +86,12 @@ func main() {
|
||||||
workingDir = cwd
|
workingDir = cwd
|
||||||
}
|
}
|
||||||
|
|
||||||
outPath := path.Join(workingDir, "qrcodes")
|
// 1.1 Operating Mode
|
||||||
if err := os.MkdirAll(outPath, os.ModePerm); err != nil {
|
mode := getMode()
|
||||||
|
|
||||||
|
// Out directory
|
||||||
|
outBasePath := path.Join(workingDir, "qrcodes")
|
||||||
|
if err := os.MkdirAll(outBasePath, os.ModePerm); err != nil {
|
||||||
handleErr("Could not create output folder: %s", err)
|
handleErr("Could not create output folder: %s", err)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -55,7 +101,7 @@ func main() {
|
||||||
handleErr("Could not get files: %s", err)
|
handleErr("Could not get files: %s", err)
|
||||||
}
|
}
|
||||||
|
|
||||||
fmt.Printf("Generating %d QR Codes from files in %s...\n\r", len(files), workingDir)
|
fmt.Printf("Generating QR Codes from %d files in %s...\n\r", len(files), workingDir)
|
||||||
bar := progressbar.Default(int64(len(files)))
|
bar := progressbar.Default(int64(len(files)))
|
||||||
|
|
||||||
for _, file := range files {
|
for _, file := range files {
|
||||||
|
@ -66,16 +112,41 @@ func main() {
|
||||||
if !strings.HasSuffix(f.Name(), ".txt") {
|
if !strings.HasSuffix(f.Name(), ".txt") {
|
||||||
continue
|
continue
|
||||||
}
|
}
|
||||||
content, err := ioutil.ReadFile(file)
|
|
||||||
if err != nil {
|
fNameNoSuffix := strings.TrimSuffix(path.Base(filepath.ToSlash(f.Name())), path.Ext(file))
|
||||||
handleErr("Could not open file '%s': %s", f.Name(), err)
|
|
||||||
|
var outPath = outBasePath
|
||||||
|
|
||||||
|
if mode == modeLine {
|
||||||
|
outPath = path.Join(outBasePath, fNameNoSuffix)
|
||||||
|
if err := os.MkdirAll(outPath, os.ModePerm); err != nil {
|
||||||
|
handleErr("Could not create output folder: %s", err)
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
// 3. Generate QR-Codes and save
|
// 3. Generate QR-Codes and save
|
||||||
outfile := path.Join(outPath, strings.TrimSuffix(path.Base(filepath.ToSlash(f.Name())), path.Ext(file))) + ".png"
|
if mode == modeFile {
|
||||||
err = qrcode.WriteFile(string(content), qrcode.Low, 2048, outfile)
|
content, err := ioutil.ReadFile(file)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
handleErr("Could not create qr code for file '%s': %s", f.Name(), err)
|
handleErr("Could not open file '%s': %s", f.Name(), err)
|
||||||
|
}
|
||||||
|
|
||||||
|
createQrCode(fNameNoSuffix, outPath, string(content))
|
||||||
}
|
}
|
||||||
|
|
||||||
|
if mode == modeLine {
|
||||||
|
scanner := bufio.NewScanner(f)
|
||||||
|
i := 0
|
||||||
|
for scanner.Scan() {
|
||||||
|
createQrCode(fNameNoSuffix+"_"+strconv.Itoa(i), outPath, scanner.Text())
|
||||||
|
i++
|
||||||
|
}
|
||||||
|
|
||||||
|
if err := scanner.Err(); err != nil {
|
||||||
|
handleErr("Could not read file '%s': %s", f.Name(), err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
_ = bar.Add(1)
|
_ = bar.Add(1)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
Loading…
Reference in New Issue