Skip to content

Commit d484a9b

Browse files
committed
refactor(main.go): rename startLine and endLine variables to start and end for brevity
feat(main.go): use original file permissions when writing to file for consistency
1 parent c652157 commit d484a9b

File tree

1 file changed

+9
-5
lines changed

1 file changed

+9
-5
lines changed

main.go

+9-5
Original file line numberDiff line numberDiff line change
@@ -34,12 +34,12 @@ func main() {
3434
log.Fatalf("Failed to parse file: %v", err)
3535
}
3636

37-
var startLine, endLine int
37+
var start, end int
3838
ast.Inspect(node, func(n ast.Node) bool {
3939
fn, ok := n.(*ast.FuncDecl)
4040
if ok && fn.Name.Name == functionName {
41-
startLine = fset.Position(fn.Pos()).Line
42-
endLine = fset.Position(fn.End()).Line
41+
start = fset.Position(fn.Pos()).Line
42+
end = fset.Position(fn.End()).Line
4343
return false
4444
}
4545
return true
@@ -54,7 +54,7 @@ func main() {
5454
buf := new(bytes.Buffer)
5555
scanner := bufio.NewScanner(file)
5656
for l := 1; scanner.Scan(); {
57-
if l < startLine || l > endLine {
57+
if l < start || l > end {
5858
if _, err := buf.Write(append(scanner.Bytes(), '\n')); err != nil {
5959
log.Fatalf("Failed to write to buffer: %v", err)
6060
}
@@ -71,7 +71,11 @@ func main() {
7171
log.Fatalf("Failed to format source: %v", err)
7272
}
7373
// write to file
74-
if err := os.WriteFile(fileName, b, 0644); err != nil {
74+
info, err := file.Stat()
75+
if err != nil {
76+
log.Fatalf("Failed to get file info: %v", err)
77+
}
78+
if err := os.WriteFile(fileName, b, info.Mode()); err != nil {
7579
log.Fatalf("Failed to write to file: %v", err)
7680
}
7781
}

0 commit comments

Comments
 (0)