/*
THE YUM TOOL, COPYRIGHT@2017~2019 BY HAPPY, VERSION 1.0
YUM.EXE
-----------------------------------------------------------------

_   _       _      _    |  The Batch Online Yum Tool,
 \ | |     | \    / |   |  Copyright@2017~2019 By Happy, yum.exe
\ \| |_   _| |\  /| |   |  Type \"help\" for help informaiton .\n\
 \_| | | | | | \/ | |   |
   | | |_| | |\  /| |   |  Version 1.0 (2017-04-30 20:32 UTC)
  _/ |\__'_|_| VV |_|   |  Officil http://www.bathome.net/thread-
 |__/                   |  43961-1-1.html

-----------------------------------------------------------------
                         .    .7     .,;;;::
                         v;   7u      ..:;UMBM7::,,,.
                         ;,   ;,    .       .BMsr7r7vL7r,
                         ,    ,    ;0         ER7uuLLJ1SKKU;
   .:xJx;.              .  ;      ;B.         .B7JSSu33FUPGRZL,
       :uDBGs:.       :   .  :.rP..            BRiJaS11UHUHKOOOP;
          :FMBRZJr.   PU.r7,MBMBM0;            uMU:LaaUFHGPHUKEORX:
             7OMBMOPr.M1:MWRBOWMBBBMWi      31  BMx;71PaXKDEEP00WRBP:
               ;3RBOsZX:sBOKUaWBMBBBMBMc,.:aM,  ;BOsrrJPXZEWORDDERRMR3
                 .7rir,EMRDXRR3;rJRBBMBMa::.     xBELrr7UXEGGZDEDDWORMWi
                   :  DMXBMBDu::.,:3OBBBBP:       JBKvrirx1KEWZZZEZKPRRRF:
                  .,.GBuSRiLF3Us:.i:iDBMBMF        LBD7;ric2GDWEZ0EKXXWWRDJ
                  irGBWURS.:1ZPEGc:. :MBMBM;        MBMrii;L2ZDOGZZEKDGWWOW0.
                  :7BRaOBK. .sHPMU  ..aBBMBJ;::7x   .MBKir7:r3KDOEZED0ZZWDOWR:
                  uLSZXBL. ,,i1ZMH ,,.sRRBMa::r7     BRMcii7:ruKDOZEDWEGEGDOGRr
                 ,2EKxB7    , ;EB: .  vPBMBu         MMMJr;ir:iJKWOEDDOEWDGGOEOi
                 JRBMRM      : 3P   ..;HRBBu        iBMB2ii;ii:;LZOOZGDGEODOWD0Dr
                  :0GZu       :;; ..;ir1RMBF.r7    .BBBDBi:;;ii:;vEOWZOGDDOWWZZ0Dr
                      :.      .r.;::.;7URBR0    .rKBMB2LMW::;iii:7JDODGOEWEWWOKE0W:
                       .       . :   ,7PBMWZ;;uOBMBRBZr;cMK:;;;irr11WWODWDWGOOGZGKH
                              .  .  ,:vDMRZMODMBMWMBRL:;:rGS::r277U0SODODWGWDODDE0FO
                              c ..,:LrPMOSKOORBMXKBDLir:::;22:H    LKKWGWGWDOWWEDDPua
                             .xr:::i:cRJsL1R0DMBEPXvir..:,..:7U:    3EZDGODWGOGWEWD7,3
                             G: r   ,JWGxrOMDRRRRDOG1;7SWRBPHMBMBJ; ;KEPOGWDWDDGDZWs ::    7
                            ;B: .   ;MWOaGRRMMGRRROBMBMBMBMBMBBBBBMBPSZPZDDOGOEDDZKU. c  .7B
                           ;MM7 :.;,1; 7RBRORWRRBMRDBBBMMMRZZERMBMBMBUPXGGWGOWDGWPUUR1xJ:,.
                           7MBi::r71r  .7ZOEBORMRMRGMMWBMWSKGRRRMBMBMK1ZZWEDDOZOZXJSMBWa0Xc   U:
                           ;BBc.7r.s:  ;J1MORDMMBM0DBMMMBHEGMMBMBW0OBS1ZEGWEWEDDZHJaBMBMOMBMF:.:
                         rvUMBM7:i:rKcxS1EMMWOWMOPGBMBO1RBMBMBRMMs :.rHGKWGDEDEEXK3URBMBMBMMMB7
                         MMMBOBMOSucZMMORMBWRMWDRRBMO7   RBBBMRRB;   ,ZGZGODDKWPZHF..;c3WBBMMWBWa
                         MMBMBMBMMJv;uMBBBMEDBMRu7Lcr     iRMBRBMB7Z . OKDGOZ0KKKPJ      ;2BMBZB0:
                         MBMBMBE  ,:: .SBMBRMMBMWSL..        OBRBMBM  .OEKDGGPKXXH7        ,WMBO3M
                       .0BMBMW:  :rr.    :OBBBORMBM7         rsOMBMO   :ZEZEKXXKUG,          RMMuBr
                      ;PRBBMD   .i:         :FRMOMBM.        2MMBRO    ;EZDHZHZSHu           cBJRMU
                    7OBBBBMX    i,          L7DBMBMR        :BMBMBZ    ;ZOZ0HXHH2.           ,.rMB;
                  sMBBWHs:     :       ;iLaMH:ZOBMM        ZBMO7,OB    .OOEPHaPS:          . .7BBM
            ,cJ7LXMBMs               ,rZuH3:.rMBB:       7MBBBD  .u    aWOXXSPXL          O. XMBM.
           7BOBM0M2.                       cxOBM     7BMBMBBBJ        .GWZX1KO         ;.,a OMBR
           :  :;,                  i7vSJrFMRMBWX     .L:Far           :OE0UaIF         .X ; OMBs
                                  HMBMBMHMBML                         vEKSaZ;          R  iBBB.
                                  v    Pai.                          .EESSQ          rr uBMr
                                       .                             ;DHDF;           L WBR
                                                                    .SRGA:          ::OBx
                                                                    .;DX.          .WGB;
                                                                    ;DW:          SGB.
                                                                   ;SM           ;XZ;
                                                                   :J            B;
                                                                   :            ;
*/
#include   <stdio.h>
#include  <stdlib.h>
#include  <string.h>
#include <windows.h>
#include <wininet.h>
#include <shlwapi.h>
#include<shellapi.h>
#include  <direct.h>
#include   <conio.h>
#include    <time.h>
#include    <math.h>
#include      <io.h>
#pragma comment(lib, "wininet.lib")
#pragma comment(lib, "shlwapi.lib")

/*************宏量定义*************/
#define SAFE_SIZE  512
#define FILE_EXIST 0
#define FILE_EREAD 4

//定义帮助说明
#define HELP_INFORMATION "\
Usage: yum [option] [package name]...\n\
\n\
  -i, install\n\
              Install the package\n\
  -u, update\n\
              Update the package\n\
  -c, clean\n\
              Clean the package cache\n\
  -f, info\n\
              Show the package info\n\
  -s, search\n\
              Search the package\n\
  -l, list\n\
              List the package, type \"installed\" for installed infomation\n\
  -r, remove\n\
              Remove the package\n\
  -h, help\n\
              Show help information\n\
\n\
Version 1.0 2017-04-30 20:32 UTC\n\
Officil: www.bathome.net/thread-43961-1-1.html\n"

#define SEARCH_HEAD "\
=================================================================\n\
NAME                 VERSION                  YUM-SN        SIZE\n\
================================================================="

#define SEARCH_LOGO "\
-----------------------------------------------------------------\n\
\n\
_   _       _      _    |  The Batch Online Yum Tool,\n\
 \\ | |     | \\    / |   |  Copyright@2017~2019 By Happy, yum.exe\n\
\\ \\| |_   _| |\\  /| |   |  Type \"help\" for help informaiton .\n\
 \\_| | | | | | \\/ | |   |\n\
   | | |_| | |\\  /| |   |  Version 1.0 (2017-04-30 20:32 UTC)\n\
  _/ |\\__'_|_| VV |_|   |  Officil http://www.bathome.net/thread-\n\
 |__/                   |  43961-1-1.html\n\
\n\
-----------------------------------------------------------------\n\n"

/*************全局变量*************/
//分配读取配置容器
static char yumsURL[SAFE_SIZE];
static char inifDIR[SAFE_SIZE];
static char rootDIR[SAFE_SIZE];
static char rootENV[SAFE_SIZE];
static char exeFullPath[SAFE_SIZE];
static char keepCACHE[SAFE_SIZE];

//列举正则容器
static char resLIST[SAFE_SIZE];
//第三方下载地址容器
static char yumURL[SAFE_SIZE];
//第三方下载目录容器
static char binPATH[SAFE_SIZE];
//解压工具UNRAR容器
static char unrarPATH[SAFE_SIZE];
static char unrarDURL[SAFE_SIZE];

//搜索关键词
static char keywords[SAFE_SIZE];

//分配缓存区、行容器
static char LCache[SAFE_SIZE*2];

//分配表格容器
static char tmpp[6][SAFE_SIZE];

//Shell命令行容器
static char unrarCMD[SAFE_SIZE];

//进度条变量
static char proGRESS[64]= {0};
clock_t preTime;
ULONG preProgress;

/*************类型定义*************/
typedef HRESULT (WINAPI *PGETA)(LPUNKNOWN,LPCSTR,LPCSTR,DWORD,LPBINDSTATUSCALLBACK);
PGETA PgetUrlToFileA;

/*************功能函数*************/
//判断纯数字
BOOL isPureNumber(const char* nstr)
{
	char* p=(char*)nstr;
	while('0'<=*p && *p<='9')
	{
		p++;
	}
	return (*p=='\0')?TRUE:FALSE;
}
//子串查找忽略大小写
const char* stristr(const char* str, const char* subStr)
{
	int len = strlen(subStr);
	if(len == 0)
	{
		return NULL;
	}

	while(*str)
	{
		if(_strnicmp(str, subStr, len) == 0)
		{
			return str;
		}
		str++;
	}
	return NULL;
}
//删除目录树
BOOL RemoveDirectoryTreeA(const char* lpszPath)
{
	SHFILEOPSTRUCT FileOp;
	FileOp.fFlags = FOF_SILENT | FOF_NOCONFIRMATION;
	FileOp.hNameMappings = NULL;
	FileOp.hwnd = NULL;
	FileOp.lpszProgressTitle = NULL;
	FileOp.pFrom = lpszPath;
	FileOp.pTo = NULL;
	FileOp.wFunc = FO_DELETE;
	return (SHFileOperationA(&FileOp)==0)?TRUE:FALSE;
}
//遍历目录
void ListInstalledPackage(const char* inpath)
{
	struct _finddata_t data;
	long hnd=_findfirst(inpath, &data);

	if(hnd<0)
	{
		fprintf(stdout, "Not install any packages");
		return;
	}

	int nRet=(hnd<0)?-1:1;

	while(nRet>=0)
	{
		if(data.attrib==_A_SUBDIR && strcmp(data.name, ".")!=0 && strcmp(data.name, "..")!=0)
		{
			fprintf(stdout, "Already installed package \"%s\"\n\n", data.name);
		}
		else
		{
			char* ep=strrchr(data.name, '.');
			if(ep!=NULL && stricmp(ep, ".exe")==0)
			{
				*ep='\0';
				fprintf(stdout, "Already installed package \"%s\"\n\n", data.name);
			}
		}
		nRet=_findnext(hnd, &data);
	}
	_findclose(hnd);
}
//进度回调
class DownloadProgress :public IBindStatusCallback
{
public:
	HRESULT __stdcall QueryInterface(const IID &, void **)
	{
		return E_NOINTERFACE;
	}
	ULONG STDMETHODCALLTYPE AddRef(void)
	{
		return 1;
	}
	ULONG STDMETHODCALLTYPE Release(void)
	{
		return 1;
	}
	HRESULT STDMETHODCALLTYPE OnStartBinding(DWORD dwReserved, IBinding *pib)
	{
		return E_NOTIMPL;
	}
	virtual HRESULT STDMETHODCALLTYPE GetPriority(LONG *pnPriority)
	{
		return E_NOTIMPL;
	}
	virtual HRESULT STDMETHODCALLTYPE OnLowResource(DWORD reserved)
	{
		return S_OK;
	}
	virtual HRESULT STDMETHODCALLTYPE OnStopBinding(HRESULT hresult, LPCWSTR szError)
	{
		return E_NOTIMPL;
	}
	virtual HRESULT STDMETHODCALLTYPE GetBindInfo(DWORD *grfBINDF, BINDINFO *pbindinfo)
	{
		return E_NOTIMPL;
	}
	virtual HRESULT STDMETHODCALLTYPE OnDataAvailable(DWORD grfBSCF, DWORD dwSize, FORMATETC *pformatetc, STGMEDIUM *pstgmed)
	{
		return E_NOTIMPL;
	}
	virtual HRESULT STDMETHODCALLTYPE OnObjectAvailable(REFIID riid, IUnknown *punk)
	{
		return E_NOTIMPL;
	}
	virtual HRESULT __stdcall OnProgress(ULONG ulProgress, ULONG ulProgressMax, ULONG ulStatusCode, LPCWSTR szStatusText)
	{
		if (ulProgressMax != 0)
		{

			//进度百分数
			int downloadSpeed, restTime;
			int i=(int)ulProgress * 100.0 / (int)ulProgressMax;
			int j=i/2, disTime=(int)(clock()-preTime);
			if(disTime > 100)
			{
				downloadSpeed=((int)ulProgress-(int)preProgress) * CLOCKS_PER_SEC / disTime / 1000;
				restTime=(downloadSpeed!=0) ?((int)ulProgressMax-(int)ulProgress) / downloadSpeed :99999;
				preProgress=ulProgress, preTime+=disTime;
			}
			else if(i <100)
			{
				return S_OK;
			}

			if(i==100)
			{
				downloadSpeed=restTime=0;
			}

			for(int n=1; n<=j; n++)
			{
				proGRESS[n]='=';
			}

			//显示进度百分比
			fprintf(stdout, "\r%s%3d%%, %3dKB/s   ", proGRESS, i, downloadSpeed);
			fflush(stdout);
		}
		return S_OK;
	}
};
//搜索函数
int SearchStr(int iargc, char** iargv, char* inifile, int optionSN)
{
	FILE* fp=fopen(inifile, "r");
	if(fp==NULL)
	{
		fprintf(stdout, "Can not read yum settings.");
		exit(1);
	}

	BOOL kMARK=FALSE, sMARK=FALSE;
	int mode=0, yumSN=0, kLEN=strlen(iargv[2]), kpmode=atoi(keepCACHE);

	if(iargv[2][0]=='*')
	{
		//匹配任意位置
		mode=1;
		strcpy(keywords, iargv[2]+1);

	}
	else if(iargv[2][kLEN-1]=='*')
	{
		//匹配首部位置
		mode=2;
		strcpy(keywords, iargv[2]);
		keywords[kLEN-1]='\0';
	}
	else if(isPureNumber(iargv[2]))
	{
		//匹配YUM-SN
		mode=3;
		yumSN=atoi(iargv[2]);
	}
	else
	{
		//匹配全字
		mode=0;
		strcpy(keywords, iargv[2]);
	}

	if(optionSN==2)
	{
		fprintf(stdout, "%s\n", SEARCH_HEAD);
	}

	int i=0;
	while(!feof(fp))
	{
		//读入标准行
		fgets(LCache, SAFE_SIZE*2, fp);

		//辅助指针
		char* Line=LCache;

		while(*Line!='\n' && *Line!='\0')
		{
			Line++;
		}
		*Line='\0';

		//指针回首
		Line=LCache;

		//过滤行TAB缩进或前空格
		while(*Line=='\t'|| *Line==' ')
		{
			Line++;
		}

		if(!kMARK)
		{
			if(stristr(Line, "[table]")==Line)
			{
				kMARK=TRUE;
				continue;
			}
		}

		if(!kMARK || *Line==';' || *Line=='\0')
		{
			continue;
		}

		//行计数器
		i++;

		//开启search开关
		if(optionSN==4 && stristr(Line, keywords))
		{
			sMARK=TRUE;
		}

		char* textstrp=NULL;

		if(textstrp=strtok(Line, " "))
		{
			strcpy(tmpp[0], textstrp);
		}

		for(int j=1; j<=5; j++)
		{
			if(textstrp=strtok(NULL, " "))
			{
				strcpy(tmpp[j], textstrp);
			}
			else
			{
				tmpp[j][0]=0;
			}
		}

		if(
		    (optionSN==4 && sMARK==TRUE)                    ||
		    (mode==0 && stricmp(tmpp[0], keywords)==0)      ||
		    (mode==1 && stristr(tmpp[0], keywords))         ||
		    (mode==2 && stristr(tmpp[0], keywords)==tmpp[0])||
		    (mode==3 && atoi(tmpp[1]+1)==yumSN)
		)
		{
			if(iargc>3 && stristr(tmpp[2], iargv[3])==NULL)
			{
				continue;
			}

			//开启list开关
			if(optionSN==2)
			{
				//打印搜索结果
				fprintf(stdout, "%-20.20s %-24.24s %s     %7s\n", tmpp[0], tmpp[2], tmpp[1], tmpp[4]);
				continue;
			}

			//开启info开关
			if(optionSN==5||optionSN==4)
			{
				//打印搜索结果
				fprintf(stdout, "\nNAME       :%s\nYUMSN      :%s\nVERSION    :%s\nCOMPRESS   :%s\nSIZE       :%s\nDESCRIPTION:%s\n\n", tmpp[0], tmpp[1], tmpp[2], tmpp[3], tmpp[4], tmpp[5]);
				sMARK=FALSE;
				continue;
			}

			if (optionSN==1)
			{
				//进度回调入口
				DownloadProgress progress;
				IBindStatusCallback* callback=(IBindStatusCallback*)&progress;

				//进度归零
				memset(proGRESS, ' ', 51);
				proGRESS[0]='[', proGRESS[51]=']';

				preTime=clock();
				preProgress=(ULONG)0;

				//判断文件压缩类型
				int pressTYPE=atoi(tmpp[3]+1);

				strcpy(yumURL, yumsURL);
				strcat(yumURL, tmpp[2]);
				strcat(yumURL, "/");
				strcat(yumURL, tmpp[0]);
				if(pressTYPE==0)
				{
					strcat(yumURL, ".exe");
				}
				else
				{
					strcat(yumURL, ".rar");
				}

				strcpy(binPATH, rootDIR);
				strcat(binPATH, tmpp[0]);


				if(pressTYPE==0)
				{
					strcat(binPATH, ".exe");
				}
				else
				{
					strcat(binPATH, ".rar");
				}

				//显示下载信息
				fprintf(stdout, "Download package \"%s\" ...\n", tmpp[0]);

				//调用下载函数
				if (PgetUrlToFileA(NULL, yumURL, binPATH, 0, static_cast<IBindStatusCallback*>(&progress)) != S_OK)
				{
					//进度归零
					memset(proGRESS, ' ', 51);
					proGRESS[0]='[', proGRESS[51]=']';

					preTime=clock();
					preProgress=(ULONG)0;

					strcpy(yumURL, yumsURL);
					strcat(yumURL, tmpp[0]);
					if(pressTYPE==0)
					{
						strcat(yumURL, ".exe");
					}
					else
					{
						strcat(yumURL, ".rar");
					}

					if (PgetUrlToFileA(NULL, yumURL, binPATH, 0, static_cast<IBindStatusCallback*>(&progress)) != S_OK)
					{
						fprintf(stdout, "Download \"%s\" error.\n\n", tmpp[0]);
						continue;
					}

				}

				fprintf(stdout, "\n");

				if(pressTYPE==1)
				{

					if(unrarPATH[0]=='\0')
					{
						strcpy(unrarPATH, rootDIR);
						strcat(unrarPATH, "unrar.exe");
					}

					if(_access(unrarPATH, FILE_EXIST)!=0)
					{
						strcpy(unrarDURL, yumsURL);
						strcat(unrarDURL, "unrar.exe");
						if (PgetUrlToFileA(NULL, unrarDURL, unrarPATH, 0, 0) != S_OK)
						{
							fprintf(stdout, "Can not download the unrar tool, can not extract the package \"%s\".\n", tmpp[0]);
							exit(1);
						}
					}

					strcpy(unrarCMD, "x -o+ -y ");
					strcat(unrarCMD, binPATH);
					strcat(unrarCMD, " ");
					strcat(unrarCMD, rootDIR);
					fprintf(stdout, "Installation package \"%s\" ...\n\n", tmpp[0]);
					ShellExecuteA(GetConsoleWindow(), "runas", unrarPATH, unrarCMD, "", SW_HIDE);

					//删除安装包
					if(kpmode==0)
					{
						remove(binPATH);
					}
				}
				else
				{
					fprintf(stdout, "\n");
				}

				if(mode==0)
				{
					return 0;
				}

			}
		}
	}

	fclose(fp);

	return 0;
}

/*************MAIN函数*************/
int main(int argc, char** argv)
{
	if(argc==2)
	{
		if(stricmp(argv[1], "help")==0 || stricmp(argv[1], "-h")==0)
		{
			fprintf(stdout, HELP_INFORMATION);
			return 0;
		}
	}

	if(argc<3)
	{
		//展示LOGO
		fprintf(stdout, SEARCH_LOGO);
		//帮助指南
		fprintf(stdout, "Type \"yum help\" for help information .\n");
		return 1;
	}

	//设置开关模式变量
	int optionSN=0;

	if(stricmp(argv[1], "install")==0 || stricmp(argv[1], "update")==0 || stricmp(argv[1], "-i")==0 || stricmp(argv[1], "-u")==0)
	{
		optionSN=1;
	}
	else if(stricmp(argv[1], "list")==0 || stricmp(argv[1], "-l")==0)
	{
		optionSN=2;
		if(stricmp(argv[2], "installed")==0)
		{
			optionSN=7;
		}
	}
	else if(stricmp(argv[1], "remove")==0 || stricmp(argv[1], "-r")==0)
	{
		optionSN=3;
	}
	else if(stricmp(argv[1], "search")==0 || stricmp(argv[1], "-s")==0)
	{
		optionSN=4;
	}
	else if(stricmp(argv[1], "info"  )==0 || stricmp(argv[1], "-f")==0)
	{
		optionSN=5;
	}
	else if(stricmp(argv[1], "clean" )==0 || stricmp(argv[1], "-c")==0)
	{
		optionSN=6;
	}
	else
	{
		//展示LOGO
		fprintf(stdout, SEARCH_LOGO);
		fprintf(stdout, "Type \"yum help\" for help information .\n");
		return 1;
	}

	//装载DLL模块
	HMODULE hLib=LoadLibraryW(L"URLMON");
	if(hLib==NULL)
	{
		fprintf(stdout, "Load urlmon library error");
		return 1;
	}

	//过小红伞本地查杀
	PgetUrlToFileA=(PGETA)GetProcAddress(hLib, "URLDownloadToFileA");

	//获取可执行文件所在目录
	GetModuleFileNameA(NULL, exeFullPath, SAFE_SIZE);
	char *p=strrchr(exeFullPath, '\\');
	*(++p)='\0';

	//获取配置文件yum.ini路径
	strcpy(inifDIR, exeFullPath);
	strcat(inifDIR, ".\\yum.ini");

	if(_access(inifDIR, FILE_EREAD)!=0)
	{
		fprintf(stdout, "Can not read yum ini.\n");
		return 1;
	}

	//获取配置键值yumsurl
	GetPrivateProfileStringA("url", "yumsurl", "non", yumsURL, SAFE_SIZE, inifDIR);
	if(strcmp(yumsURL, "non")==0)
	{
		fprintf(stdout, "Needs yum url.\n");
		return 1;
	}
	else
	{
		if(strrchr(yumsURL, '/')+1 != '\0')
		{
			strcat(yumsURL, "/");
		}
	}

	//获取配置键值rootdir
	GetPrivateProfileStringA("dir", "rootdir", "non", rootDIR, SAFE_SIZE, inifDIR);
	if(strcmp(rootDIR, "non")==0)
	{
		strcpy(rootDIR, exeFullPath);
	}
	else
	{
		ExpandEnvironmentStringsA(rootDIR, rootENV,SAFE_SIZE);
		if(strchr(rootENV, ':')==NULL)
		{
			strcpy(rootDIR, exeFullPath);
			strcat(rootDIR, rootENV);
		}
		else
		{
			strcpy(rootDIR, rootENV);
		}

		//判断根目录是否存在
		if(!PathIsDirectory(rootDIR))
		{
			fprintf(stdout, "The root directory \"%s\" does not exist.\n", rootDIR);
			return 1;
		}

		strcat(rootDIR, ".\\");
	}

	//获取配置键值keepcache
	GetPrivateProfileStringA("setting", "keepcache", "non", keepCACHE, SAFE_SIZE, inifDIR);
	if(strcmp(keepCACHE, "non")==0)
	{
		strcpy(keepCACHE, "0");
	}

	//展示LOGO
	if(optionSN!=2)
	{
		fprintf(stdout, SEARCH_LOGO);
	}

	//删除包命令
	if(optionSN==3)
	{
		fprintf(stdout, "Remove package \"%s\" ...\n", argv[2]);

		BOOL isREMOVE=TRUE;
		char tmpT[SAFE_SIZE]= {0};

		strcpy(tmpT, rootDIR);
		strcat(tmpT, argv[2]);

		//删除包目录树
		if(PathIsDirectory(tmpT))
		{
			if(!RemoveDirectoryTreeA(tmpT))
			{
				isREMOVE=FALSE;
			}
		}

		strcat(tmpT, ".exe");
		if(_access(tmpT, FILE_EXIST)==0)
		{
			if(remove(tmpT)!=0)
			{
				isREMOVE=FALSE;
			}
		}

		char* tp=strrchr(tmpT, '.');

		if(isREMOVE)
		{
			fprintf(stdout, "The package \"%s\" has been removed.\n", argv[2]);
		}
		else
		{
			fprintf(stdout, "The package \"%s\" remove failed.\n", argv[2]);
		}

		return 0;
	}

	if(optionSN==6)
	{
		fprintf(stdout, "Clean package \"%s\"'s cache ...\n", argv[2]);

		char tmpT[SAFE_SIZE]= {0};
		strcpy(tmpT, rootDIR);
		strcat(tmpT, argv[2]);
		strcat(tmpT, ".rar");
		remove(tmpT);
		return 0;
	}

	if(optionSN==7)
	{
		strcpy(resLIST, rootDIR);
		strcat(resLIST, ".\\*");
		ListInstalledPackage(resLIST);
		return 0;
	}

	//调用核心函数
	SearchStr(argc, argv, inifDIR, optionSN);

	return 0;
}
复制代码