实现一个类似grep的文本搜索工具

大多数操作系统都会提供本地的搜索引擎。用户可以使用一些快捷键，对本地文件进行查找。

这种功能出现之前，命令行用户会通过grep或awk工具对文件进行查找。用户可以简单的输入grep -r foobar .，然后工具将会基于当前目录，进行递归的的查找，并显示包含有"foobar"名字的文件。

本节中，将实现这样一种应用。我们的grep使用命令行方式使用，并基于给定文件夹递归的对文件进行查找。然后，将找到的文件名打印出来。我们将使用线性的模式匹配方式，将匹配文件中的对应行号进行打印。

How to do it...

我们将实现小工具，用于查找与用户提供的文本段匹配的文件。这工具与UNIX中的grep工具类似，不过为了简单起见，其功能没有那么强大：

包含必要的头文件，并声明所使用的命名空间：

#include <iostream>
#include <fstream>
#include <regex>
#include <vector>
#include <string>
#include <filesystem>

using namespace std;
using namespace filesystem;

先来实现一个辅助函数，这个函数能接受一个文件地址和一个正则表达式对象，正则表达式对象用来描述我们要查找的文本段。然后，实例化一个vector，用于保存匹配的文件行和其对应的内容。然后，实例化一个输入文件流对象，读取文件，并进行逐行的文本匹配。
```
static vector<pair<size_t, string>>
matches(const path &p, const regex &re)
{
    vector<pair<size_t, string>> d;
    ifstream is {p.c_str()};
```

通过getline函数对文件进行逐行读取，当字符串中包含有我们提供文本段，则regex_search返回true，如果匹配会将字符串和对应的行号保存在vector中。最后，我们将返回所有匹配的结果：

    string s;
    for (size_t line {1}; getline(is, s); ++line) {
        if (regex_search(begin(s), end(s), re)) {
            d.emplace_back(line, move(s));
        }
    }

    return d;
}

主函数会先对用户提供的文本段进行检查，如果这个文本段不能用，则返回错误信息：

int main(int argc, char *argv[])
{
    if (argc != 2) {
        cout << "Usage: " << argv[0] << " <pattern>\n";
        return 1;
    }

接下来，会通过输入文本创建一个正则表达式对象。如果表达式是一个非法的正则表达式，这将会导致一个异常抛出。如果触发了异常，我们将对异常进行捕获并处理：
```
    regex pattern;

    try { pattern = regex{argv[1]}; }
    catch (const regex_error &e) {
        cout << "Invalid regular expression provided.n";
        return 1;
    }
```
现在，可以对文件系统进行迭代，然后对我们提供的文本段进行匹配。使用recursive_directory_iterator对工作目录下的所有文件进行迭代。原理和之前章节的directory_iterator类似，不过会对子目录进行递归迭代。对于每个匹配的文件，我们都会调用辅助函数matches：
```
    for (const auto &entry :
            recursive_directory_iterator{current_path()}) {
        auto ms (matches(entry.path(), pattern));
```

如果有匹配的结果，我们将会对文件地址，对应文本行数和匹配行的内容进行打印：

    for (const auto &[number, content] : ms) {
        cout << entry.path().c_str() << ":" << number
             << " - " << content << '\n';
        }
    }
}

现在，准备一个文件foobar.txt，其中包含一些测试行：
```
foo
bar
baz
```
编译并运行程序，就会得到如下输出。我们在/Users/tfc/testdir文件夹下运行这个程序，我们先来对bar进行查找。在这个文件夹下，其会在foobar.txt的第二行和testdir/dir1文件夹下的另外一个文件text1.txt中匹配到：
```
$ ./grepper bar
/Users/tfc/testdir/dir1/text1.txt:1 - foo bar bla blubb
/Users/tfc/testdir/foobar.txt:2 - bar
```
再次运行程序，这次我们对baz进行查找，其会在第三行找到对应内容：
```
$ ./grepper baz
/Users/tfc/testdir/foobar.txt:3 - baz
```

How it works...

本节的主要任务是使用正则表达式对文件的内容进行查找。不过，让我们关注一下recursive_directory_iterator，因为我们会使用这个迭代器来进行本节的子文件夹的递归迭代。

与directory_iterator和recursive_directory_iterator迭代类似，其可以用来对子文件夹进行递归，就如其名字一样。当进入文件系统中的一个文件夹时，将会产生一个directory_entry实例。当递归到子文件夹时，也会产生对应的directory_entry实例。

recursive_directory_iterator具有一些有趣的成员函数：

depth()代表我们需要迭代多少层子文件夹。
recursion_pending()代表在进行当前迭代器后，是否会在进行对子文件夹进行迭代。
disable_recursion_pending()当迭代器需要再对子文件夹进行迭代时，提前调用这个函数，则会让递归停止。
pop()将会终止当前级别的迭代，并返回上一级目录。

There's more...

我们需要了解的另一个就是directory_options枚举类。recursive_directory_iterator能将directory_options的实例作为其构造函数的第二个参数，通常将directory_options::none作为默认值传入。其他值为：

follow_directory_symlink能允许对符号链接的文件夹进行递归迭代。
skip_permission_denied这会告诉迭代器，是否跳过由于权限错误而无法访问的目录。

这两个选项可以通过|进行组合。

Previous列出目录下的所有文件 Next实现一个自动文件重命名器

Last updated 3 years ago