如何用波浪号替换主目录？ Linux

Question

我目前正在用 C++ 制作一个 shell，我已经能够使用 getcwd() 将绝对目录作为提示输出，但是当用户到达主目录时，我想输出一个波浪号而不是主目录目录。

所以例如代替

[bb@bbmachine bb]

我想输出

[bb@bbmachine ~]

我该怎么做？

Answer 1

首先阅读globbing，尤其是glob(7)。对于以波浪号~ 开头的路径的扩展，请考虑wordexp(3) 和glob(3)。

要显示以主目录开头的路径，您需要获取该主目录。您可以在getuid(2) 的结果上使用getenv(3)，如getenv("HOME")（在极少数病理情况下可能会失败）或getpwuid(3)（甚至getpw(3)）。主目录（当getpwuid 成功时）由pw_dir 字段给出。另见credentials(7)（和execve(2)...）

getenv("HOME") 和 getpwuid(getuid()) 失败或给出不一致的结果（关于用户的主目录）的情况很少见。我相信你应该发现失败，但你可能不关心不一致。另见environ(7)。您甚至可以考虑 GNU 特定的 secure_getenv(3)。

顺便说一句，您可以缓存主目录，即在您的 shell 中计算一次。不太可能经常更改。

您还可以决定使用realpath(3) 规范化路径。

您肯定有极端情况：假设您的进程在@987654347 下运行，您是否要将/home/john/../mary/ 显示为~/../mary、/home/mary/、~mary，（或者甚至将其保留为/home/john/../mary） @uid？ （john 和 mary 都作为用户名存在于 passwd(5) 中，主目录为 /home/john/ 和 /home/mary/）。那么/home/john/../../proc/$$/maps（见proc(5)）呢？

您需要决定如何处理这些极端情况或病态情况，并记录您的 shell 的行为。

^{（提到的函数有一个 C API，你可能想在你的 C++ 代码中使用extern "C"）}

Answer 2

正是@TheDude 所说的。这是一些伪代码。

#include "pwd.h"
#include <string>
string home=getenv("HOME");
int location=yourPathString.find(home);
length=home.length();
originalLength=yourPathString.length();
string newString=yourPathString.substr(0,location);
newString+= "~";
newString+= yourPathString.substr(location+length,originalLength);

这应该会给你正确的想法，但请记住，文档和 Google 是你的朋友。String Reference

Answer 3

你可以像这样使用wordexp：

#include <wordexp.h>

// put all the horrible bits into a function
std::string wordexp(std::string var, int flags = 0)
{
    wordexp_t p;
    if(!wordexp(var.c_str(), &p, flags))
    {
        if(p.we_wordc && p.we_wordv[0])
            var = p.we_wordv[0];
        wordfree(&p);
    }
    return var;
}

int main()
{
    auto const HOME = wordexp("~");

    std::string path = "/home/galik/wibble";

    if(path.find(HOME) == 0)
        path.replace(0, HOME.size(), "~");

    std::cout << path << '\n';
}

输出：

~/wibble