はじめに

前回は、lsコマンドの一部実装に取り組んでみました。今回は、grepコマンドの一部実装に取り組んでみます。
目標としては、

検索文字列が出現する行を出力し、その検索文字列は区別できるように色づけてあげること
-nオプションで出現位置の行数も出力するようにすること

の2つを設定したいと思います。

前回の記事はこちらから

実装

今回のリポジトリですmm

まず、1つ目の「 my-grep <target> sample.txtで検索文字列が出現する行を出力すること」を実装してみます。ポイントは、Scannerを用いて、一行ずつテキストを読み込み、対象の文字列が含まれているか走査している点です。また、対象文字列の色付けにはfatih/colorというライブラリを拝借させていただきました。

main.go

package main

import (
	"bufio"
	"fmt"
	"os"
	"strings"

	"github.com/fatih/color"
)

type Line struct {
	Content  string
}

func main() {
	targetString := os.Args[1]
	targetFile := os.Args[2]

	file, err := os.Open(targetFile)
	if err != nil {
		panic("Error opening file")
	}
	defer file.Close()

	scanner := bufio.NewScanner(file)

	matchedLines := []Line{}

	for scanner.Scan() {
		if strings.Contains(scanner.Text(), targetString) {
			matchedLines = append(matchedLines, Line{Content: scanner.Text()})
		}
	}

	listLines(matchedLines, targetString)
}

func listLines(lines []Line, target string) {
	for _, line := range lines {
		fmt.Println(reconstructLine(line.Content, target))
	}
}

func reconstructLine(line string, target string) string {
	green := color.New(color.FgGreen).SprintFunc()
	return strings.ReplaceAll(line, target, green(target))
}

検証用のテキストファイルの内容を示しておきます。テキストはジャレド・ダイアモンドの「銃・病原菌・鉄」からの引用です。

sample.txt

"History followed different courses for
different peoples because of differences among peoples' environments,
not because of biological differences among peoples themselves"
― Jared Diamond, Guns, Germs and Steel: The Fates of Human Societies

では、実行してみます。

peoplesという文字列の出現した行がきちんと出力されています！

では、目標の2つ目の-nオプションも実装してみます。内容としては、flagライブラリを利用して、コマンドライン引数を受け取るようにし、あとは単純にループで行数のインクリメントを行っています。

main.go

package main

import (
	"bufio"
	"flag"
	"fmt"
	"os"
	"strings"

	"github.com/fatih/color"
)

type Line struct {
	Content  string
+   Position int
}

func main() {
+	isLineNumberCountingEnabled := flag.Bool("n", false, "Count lines")
+	flag.Parse()

+	targetString := flag.Arg(0)
+	targetFile := flag.Arg(1)
-   targetString := os.Args[1]
-	targetFile := os.Args[2]	

	file, err := os.Open(targetFile)
	if err != nil {
		panic("Error opening file")
	}
	defer file.Close()

	scanner := bufio.NewScanner(file)

	matchedLines := []Line{}
+	currentLineNumber := 1

	for scanner.Scan() {
		if strings.Contains(scanner.Text(), targetString) {
-  			matchedLines = append(matchedLines, Line{Content: scanner.Text()})
+			matchedLines = append(matchedLines, Line{Content: scanner.Text(), Position: currentLineNumber})
		}
		currentLineNumber++
	}

+	if *isLineNumberCountingEnabled {
+		listLinesWithLineNumber(matchedLines, targetString)
+		return
+	}

	listLines(matchedLines, targetString)
}

func listLines(lines []Line, target string) {
	for _, line := range lines {
		fmt.Println(reconstructLine(line.Content, target))
	}
}

+func listLinesWithLineNumber(lines []Line, target string) {
+	for _, line := range lines {
+		fmt.Printf("%d: %s\n", line.Position, reconstructLine(line.Content, target))
+	}
+}

func reconstructLine(line string, target string) string {
	green := color.New(color.FgGreen).SprintFunc()
	return strings.ReplaceAll(line, target, green(target))
}

では、こちらも実行してみます。

無事、amongという単語が行数と共に出力されていることが分かります！

おわりに

今回は、grepコマンドの一部実装をしてみました。次回はlinuxの標準コマンドではありませんが、treeコマンドの実装にチャレンジしてみます！