//獲取用戶名
const char* GetUserName()
{
	const char* name = getenv("USER");
	return name == NULL ? "None" : name;
}

 //獲取主機(jī)名
const char* GetHostName()
{
	const char* hostname = getenv("HOSTNAME");
	return hostname == NULL ? "None" : hostname;
}

 //獲取當(dāng)前路徑
const char* GetPwd()
{
	const char* pwd = getenv("PWD");
	return pwd == NULL ? "None" : pwd;
}

//制作命令行提示符
void MakeCommandline(char com_prompt[], int size)
{
	snprintf(com_prompt, size, FORMAT, GetUserName(), GetHostName(), GetPwd());//我們想讓"[%s@%s %s]# "以后能隨便調(diào)整，所以我們define一下
}

//打印命令行提示符
void PrintCommandline()
{
	char prompt[COMMAND_SIZE];
	MakeCommandline(prompt, sizeof(prompt));//先制作
	printf("%s", prompt);//再打印
	fflush(stdout);
}

 //獲取用戶輸入
bool GetCommandline(char* out, int size)
{
	//"ls -a -l "=>"ls -a -l \n"
	const char* c = fgets(out, size, stdin);//從標(biāo)準(zhǔn)輸入里獲取，放到out當(dāng)中
	if (c == NULL) return 1;
	out[strlen(out) - 1] = 0;//清理\n 
	if (strlen(out) == 0) return false; //對(duì)于我們用戶來(lái)說(shuō)，有可能獲取到的字符串的長(zhǎng)度為0，為0直接return false 
	return true; //否則return true
}

int main()
{
	//printf("[%s@%s %s]# ",GetUserName(),GetHostName(),GetPwd());
    //1. 打印命令行提示符
	PrintCommandline();

	//2.獲取用戶輸入
	char commandline[COMMAND_SIZE];//定義一個(gè)數(shù)組
	if (GetCommandline(commandline, sizeof(commandline)))//如果獲取成功
	{
		printf("echo %s\n", commandline);//回顯一下我們輸入的內(nèi)容，用作測(cè)試
	}

	return 0;
}

//命令行分析
bool CommandParse(char* commandline)
{
#define SEP " "
	g_argc = 0; //每次進(jìn)來(lái)初始化為0
	//"ls -a -l" => "ls" "-a" "-l"
	g_argv[g_argc++] = strtok(commandline, SEP);

	while (g_argv[g_argc++] = strtok(nullptr, SEP));//再次想切的話傳commandline就不對(duì)了，再要切歷史字符串就得把它設(shè)為nullptr,再次切的話分隔符依舊是SEP,如果切成了返回的就是下一個(gè)字符串的起始地址
	//為什么可以這樣切呢？因?yàn)樵俅吻凶执畷r(shí)，它一直切一直切最后就會(huì)會(huì)變成NULL,切成NULL首先會(huì)把g_argv數(shù)組置為NULL，符合命令行參數(shù)表的設(shè)定，NULL也會(huì)作為while的條件判斷，最后就直接結(jié)束了
		 //并且g_argc也會(huì)統(tǒng)計(jì)出命令行參數(shù)有多少個(gè)
	g_argc--;//因?yàn)镹ULL也被統(tǒng)計(jì)到了里面
	return true;
}

 //測(cè)試形成的表結(jié)構(gòu)
void PrintArgv()
{
	for (int i = 0; i < g_argc; i++)
	{
		printf("argv[%d]->%s\n", i, g_argv[i]);
	}
	printf("argc:%d", g_argc);
}

//執(zhí)行命令
int Execute()
{
	pid_t id = fork();
	if (id == 0)
	{
		//子進(jìn)程
		execvp(g_argv[0], g_argv);
		exit(1);
	}
	//父進(jìn)程
	pid_t rid = waitpid(id, nullptr, 0);
	return 0;
}

//路徑切割
std::string DirName(const char* pwd)
{
#define SLASH "/"
	std::string dir = pwd;
	if (dir == SLASH) return SLASH;
	auto pos = dir.rfind(SLASH);
	if (pos == std::string::npos) return "BUG";//這里表示沒(méi)有找到/
	return dir.substr(pos + 1);
}

 //處理cd命令
bool cd()
{
	if (g_argc == 1) //表明只是cd,沒(méi)有帶任何參數(shù)
	{
		std::string home = GetHome();//將home的路徑拿過(guò)來(lái)
		if (home.empty())  return true;//如果是空就相當(dāng)于環(huán)境變量獲取失敗了
		chdir(home.c_str());//走到這里不為空，就把當(dāng)前路徑切換成家路徑了
	}
	else
	{
		std::string where = g_argv[1];
		//"cd -" / "cd ~"
		if (where == "-")
		{

		}
		else if (where == "~")
		{

		}
		else
		{
			chdir(where.c_str());
		}
	}
}
 //檢測(cè)并處理內(nèi)建命令
bool CheckAndExecBuiltin()
{
	std::string cmd = g_argv[0];
	if (cmd == "cd")//如果是內(nèi)建命令
	{
		cd();
		return true;//是內(nèi)建命令
	}
	return false;//否則不是內(nèi)建命令
}

 //獲取當(dāng)前路徑
const char* GetPwd()
{
	//const char* pwd = getenv("PWD");
	const char* pwd = getcwd(cwd, sizeof(cwd));
	if (pwd != NULL)
	{
		snprintf(cwdenv, sizeof(cwdenv), "PWD=%s", cwd);//獲得環(huán)境變量
		putenv(cwdenv);//將環(huán)境變量導(dǎo)給當(dāng)前進(jìn)程
	}
	return pwd == NULL ? "None" : pwd;
}

//處理echo命令
void Echo()
{
	if (g_argc == 2)//意思是echo后面必須得跟東西
	{
		//echo "heool world"
		   //echo $?
			//echo $PATH
		std::string opt = g_argv[1];
		if (opt == "$?")//輸出上一個(gè)程序退出的退出碼
		{
			std::cout << lastcode << std::endl;
			lastcode = 0;//lastcode清零
		}
		else if (opt[0] == '$')//如果第一個(gè)字符時(shí)是$，那么說(shuō)明查環(huán)境變量的內(nèi)容了
		{
			std::string env_name = opt.substr(1);//去掉$后的內(nèi)容就是環(huán)境變量的名字
			const char* env_value = getenv(env_name.c_str());
			if (env_value)
				std::cout << env_value << std::endl;
		}
		else
		{
			std::cout << opt << std::endl;
		}
	}
}

 //初始化環(huán)境變量表
void InitEnv()
{
	extern char** environ;//聲明一個(gè)環(huán)境變量所對(duì)應(yīng)的信息
	memset(g_env, 0, sizeof(g_env));//將表中的信息全部置為0
	g_envs = 0;

	//本來(lái)要從配合文件中來(lái)
	//今天從父shell中來(lái)
	//1. 獲取環(huán)境變量
	for (int i = 0; environ[i]; i++)
	{
		g_env[i] = (char*)malloc(strlen(environ[i]) + 1);
		//1.2拷貝
		strcpy(g_env[i], environ[i]);//把父進(jìn)程環(huán)境變量里的值拷貝給g_env
		g_envs++;
	}
	g_env[g_envs] = NULL;
	//2.導(dǎo)入環(huán)境變量
	for (int i = 0; g_env[i]; i++)
	{
		putenv(g_env[i]);
	}
	environ = g_env;
	//3.clean清理
}

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<unistd.h>
#include<sys/types.h>
#include<sys/wait.h>
#include<unordered_map>

#define COMMAND_SIZE 1024 
#define FORMAT "[%s@%s %s]# "
#define MAXARGC 128

//命令行參數(shù)表
char* g_argv[MAXARGC];//全局的命令行參數(shù)表
int g_argc = 0;

//環(huán)境變量表
#define MAX_ENVS 100
char* g_env[MAX_ENVS];//正常情況下shell啟動(dòng)時(shí)應(yīng)該從配置文件中讀取環(huán)境變量來(lái)填充這張環(huán)境變量表，也就是說(shuō)當(dāng)shell啟動(dòng)時(shí)就應(yīng)該拿環(huán)境變量表來(lái)初始化它
int g_envs = 0;//環(huán)境變量的個(gè)數(shù)

//別名映射表
std::unordered_map<std::string,std::string> alias_list;


//定義一個(gè)cwd
char cwd[1024];
char cwdenv[1024];//當(dāng)前工作路徑的env

//最新程序的退出碼 last exit cide
int lastcode = 0;



//獲取用戶名
const char* GetUserName()
{
    const char* name = getenv("USER");
    return name == NULL ? "None" : name;
}

//獲取主機(jī)名
const char* GetHostName()
{
    const char* hostname = getenv("HOSTNAME");
    return hostname == NULL ? "None" : hostname;
}

//獲取當(dāng)前路徑
const char* GetPwd()
{
    //const char* pwd = getenv("PWD");
    const char* pwd = getcwd(cwd,sizeof(cwd));
    if(pwd != NULL)
    {
        snprintf(cwdenv, sizeof(cwdenv), "PWD=%s", cwd);//獲得環(huán)境變量
        putenv(cwdenv);//將環(huán)境變量導(dǎo)給當(dāng)前進(jìn)程
    }
    return pwd == NULL ? "None" : pwd;
}

//獲取家目錄
const char* GetHome()
{
    const char* home = getenv("HOME");
    return home == NULL ? "" : home;
}

//初始化環(huán)境變量表
void InitEnv()
{
    extern char** environ;//聲明一個(gè)環(huán)境變量所對(duì)應(yīng)的信息
    memset(g_env, 0, sizeof(g_env));//將表中的信息全部置為0
    g_envs = 0;

    //本來(lái)要從配合文件中來(lái)
    //今天從父shell中來(lái)
    //1. 獲取環(huán)境變量
    for(int i = 0;environ[i];i++)
    {
        //1.1申請(qǐng)空間
        g_env[i] = (char*)malloc(strlen(environ[i])+1);
        //1.2拷貝
        strcpy(g_env[i],environ[i]);//把父進(jìn)程環(huán)境變量里的值拷貝給g_env
        g_envs++;
    }
    g_env[g_envs] = NULL;
    //2.導(dǎo)入環(huán)境變量
    for(int i = 0;g_env[i];i++)
    {
        putenv(g_env[i]);
    }
    environ = g_env;
    //3.clean清理
}

//處理cd命令
bool cd()
{
    if(g_argc == 1) //表明只是cd,沒(méi)有帶任何參數(shù)
    {
        std::string home = GetHome();//將home的路徑拿過(guò)來(lái)
        if(home.empty())  return true;//如果是空就相當(dāng)于環(huán)境變量獲取失敗了
        chdir(home.c_str());//走到這里不為空，就把當(dāng)前路徑切換成家路徑了
    }
    else
    {
        std::string where = g_argv[1];
        //"cd -" / "cd ~"
        if(where == "-")
        {
        
        }
        else if(where == "~")
        {
        
        }
        else
        {
            chdir(where.c_str());
        }
    }
    return true;
}

//處理echo命令
void Echo()
{
    if(g_argc == 2)//意思是echo后面必須得跟東西
    {
        //echo "heool world"
        //echo $?
        //echo $PATH
        std::string opt = g_argv[1];
        if(opt == "$?")//輸出上一個(gè)程序退出的退出碼
        {
            std::cout << lastcode <<std::endl;
            lastcode = 0;//lastcode清零
        }
        else if(opt[0] == '$')//如果第一個(gè)字符時(shí)是$，那么說(shuō)明查環(huán)境變量的內(nèi)容了
        {
            std::string env_name = opt.substr(1);//去掉$后的內(nèi)容就是環(huán)境變量的名字
            const char* env_value = getenv(env_name.c_str());
            if(env_value)
                std::cout << env_value << std::endl;
        }
        else
        {
            std::cout << opt << std::endl;
        }
    }   

}

//路徑切割
std::string DirName(const char* pwd)
{
#define SLASH "/"
    std::string dir = pwd;
    if(dir == SLASH) return SLASH;
    auto pos = dir.rfind(SLASH);
    if(pos == std::string::npos) return "BUG";//這里表示沒(méi)有找到/
    return dir.substr(pos+1);
}

//制作命令行提示符
void MakeCommandline(char com_prompt[],int size)
{
    //snprintf(com_prompt, size, FORMAT,GetUserName(),GetHostName(),GetPwd());//我們想讓"[%s@%s %s]# "以后能隨便調(diào)整，所以我們define一下
    snprintf(com_prompt, size, FORMAT,GetUserName(),GetHostName(),DirName(GetPwd()).c_str());//我們想讓"[%s@%s %s]# "以后能隨便調(diào)整，所以我們define一下
}

//打印命令行提示符
void PrintCommandline()
{
    char prompt[COMMAND_SIZE];
    MakeCommandline(prompt,sizeof(prompt));//先制作
    printf("%s",prompt);//再打印
    fflush(stdout);
}

//獲取用戶輸入
bool GetCommandline(char* out,int size)
{
    //"ls -a -l "=>"ls -a -l \n"
    const char *c = fgets(out,size,stdin);//從標(biāo)準(zhǔn)輸入里獲取，放到out當(dāng)中
    if(c == NULL) return 1;
    out[strlen(out)-1] = 0;//清理\n 
    if(strlen(out) == 0) return false; //對(duì)于我們用戶來(lái)說(shuō)，有可能獲取到的字符串的長(zhǎng)度為0，為0直接return false 
    return true; //否則return true
}

//命令行分析
 bool CommandParse(char* commandline)
{
#define SEP " "
    g_argc = 0; //每次進(jìn)來(lái)初始化為0
    //"ls -a -l" => "ls" "-a" "-l"
    g_argv[g_argc++] = strtok(commandline,SEP);
    
   while((bool)(g_argv[g_argc++] = strtok(nullptr,SEP)));//再次想切的話傳commandline就不對(duì)了，再要切歷史字符串就得把它設(shè)為nullptr,再次切的話分隔符依舊是SEP,如果切成了返回的就是下一個(gè)字符串的起始地址
   //為什么可以這樣切呢？因?yàn)樵俅吻凶执畷r(shí)，它一直切一直切最后就會(huì)會(huì)變成NULL,切成NULL首先會(huì)把g_argv數(shù)組置為NULL，符合命令行參數(shù)表的設(shè)定，NULL也會(huì)作為while的條件判斷，最后就直接結(jié)束了
   //并且g_argc也會(huì)統(tǒng)計(jì)出命令行參數(shù)有多少個(gè)
   g_argc--;//因?yàn)镹ULL也被統(tǒng)計(jì)到了里面
   return g_argc > 0 ? true : false;
}

//測(cè)試形成的表結(jié)構(gòu)
void PrintArgv()
{
    for(int i = 0;i < g_argc; i++)
    {
        printf("argv[%d]->%s\n",i,g_argv[i]);
    }
    printf("argc:%d\n",g_argc);
}

//檢測(cè)并處理內(nèi)建命令
bool CheckAndExecBuiltin()
{
    std::string cmd = g_argv[0];
    if(cmd == "cd")//如果是內(nèi)建命令
    {
        cd(); 
        return true;//是內(nèi)建命令
    }
    else if(cmd == "echo")
    {
        Echo();
        return true;//內(nèi)建命令執(zhí)行完后return true
    }
    else if(cmd == "export")
    {
        
    }
    else if(cmd == "alias")//說(shuō)明是一個(gè)別名
    {
        //std::string nickname = g_argv[1];
       // alisa_list.insert(k,v);
    }
    return false;//否則不是內(nèi)建命令
}

//執(zhí)行命令
int Execute()
{
    pid_t id = fork();
    if(id == 0)
    {
        //子進(jìn)程
        execvp(g_argv[0],g_argv);
        exit(1);
    }
    int status = 0;//退出信息
    //父進(jìn)程
    pid_t rid = waitpid(id, &status, 0);
    if(rid > 0) //說(shuō)明等成功了
    {
        lastcode = WEXITSTATUS(status);//最后一個(gè)進(jìn)程退出的退出碼
    }
    return 0;
}
int main()
{
    //shell啟動(dòng)時(shí)需要從系統(tǒng)中獲取環(huán)境變量
    //我們的環(huán)境變量信息應(yīng)該從父shell中統(tǒng)一來(lái)
    InitEnv();


    while(true)
    {
         //printf("[%s@%s %s]# ",GetUserName(),GetHostName(),GetPwd());
         //1. 打印命令行提示符
         PrintCommandline();

         //2.獲取用戶輸入
         char commandline[COMMAND_SIZE];//定義一個(gè)數(shù)組
         if(!GetCommandline(commandline,sizeof(commandline)))//如果獲取失敗continue重新獲取
            continue;    
        
         // printf("echo %s\n",commandline);//回顯一下我們輸入的內(nèi)容，用作測(cè)試

         //3."ls -a -l" -> "ls" "-a" "-l"
         //命令行分析
         if(!CommandParse(commandline))//分析commandline中的命令行
            continue;//只有解析成功后才往下走，否則就繼續(xù)
         //PrintArgv();

         // 檢查別名 直接將commandline替換成別名

         //4.檢測(cè)并處理內(nèi)建命令
         if(CheckAndExecBuiltin())
            continue;//若為內(nèi)建命令就不用創(chuàng)建子進(jìn)程了，由bash親自執(zhí)行
         //5. 執(zhí)行命令
         Execute();

   }
    return 0;
}