-
Notifications
You must be signed in to change notification settings - Fork 0
/
main.go
96 lines (83 loc) · 1.42 KB
/
main.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
package main
import (
"bufio"
"flag"
"fmt"
"io"
"os"
"strconv"
)
func usage() {
fmt.Println("usage:")
fmt.Println("to indent with tabs: indent +")
fmt.Println("to unindent with 4 spaces: indent - 4")
}
func main() {
flag.Parse()
args := flag.Args()
if len(args) < 1 {
usage()
os.Exit(0)
}
op := args[0]
var spaces bool
var amount int
if len(args) == 2 {
spaces = true
thisAmount, err := strconv.Atoi(args[1])
amount = thisAmount
if err != nil {
fmt.Printf("err: %s\n", err.Error())
os.Exit(1)
}
} else {
spaces = false
amount = 1
}
if op != "+" && op != "-" {
fmt.Println("operator must be '+' or '-'!")
usage()
os.Exit(1)
}
reader := bufio.NewReader(os.Stdin)
for {
line, err := reader.ReadString('\n')
dieAfterwards := false
if err == io.EOF {
dieAfterwards = true
}
if err != nil && err != io.EOF {
fmt.Printf("err: %s\n", err.Error())
os.Exit(1)
}
if op == "+" {
var char string
if spaces {
char = " "
} else {
char = "\t"
}
for i := 0; i < amount; i++ {
fmt.Printf(char)
}
fmt.Printf("%s", line)
}
if op == "-" {
canUnindent := true
for i := 0; i < len(line) && i < amount; i++ {
if line[i] != ' ' && line[i] != '\t' {
canUnindent = false
break
}
}
if canUnindent {
fmt.Printf("%s", line[amount:])
} else {
fmt.Printf("%s", line)
}
}
if dieAfterwards {
break
}
}
}