/*
	REGULAR COPY TOOLS, COPYRIGHT@2016~2018 BY HAPPY, VERSION 1.1
	REPY.EXE
*/

#include   "deelx.h"
#include <windows.h>
#include  <direct.h>
#include   <stdio.h>
#include  <assert.h>
#include  <stdlib.h>
#include  <string.h>
#include      <io.h>

/***************定义全局变量*************/
//百分比标记
char perMARKS[51];
//进度条容器
const char proGRESS[128]="[                                                      ]\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b";
//定义路径名最大长度（字节）
#define  PATH_SIZE   512
//定义文件名最大长度（字节）
#define  FNAME_SIZE 1024
//主窗口句柄
HANDLE handle_out;

//定义帮助说明
#define HELP_INFORMATION "\
-----------------------------------------------------------------\n\
REGULAR COPY TOOLS, COPYRIGHT@2017~2019 BY HAPPY, VERSION 1.1\n\
-----------------------------------------------------------------\n\
repy [match expression] [format expression]\n\
-----------------------------------------------------------------\n\
     /h  Show help information\n\
     /r  Recursive subdirectories\n\
     /f  Force copy\n\
-----------------------------------------------------------------\n\
2017-02-25"

/***************功能函数群***************/
char* strrstr(const char *dst, const char *src)  
{  
	assert(dst);assert(src);  
	const char *pdst=dst, *psrc=src; 
	char *ret=NULL;  
	while(*dst){
		while(*pdst==*psrc){ 
 
			if(*pdst=='\0'){
				return ret=(char*)dst;
			}else{  
				pdst++;  
				psrc++;  
			}  
		}  
		if(*psrc=='\0'){ret=(char*)dst;}
		pdst=++dst;  
		psrc=src;  
	}  
	return ret;  
}
//进度条回调函数
DWORD CALLBACK CopyProgressRunTim(

	LARGE_INTEGER tFileSize, 
	LARGE_INTEGER tTransferred,
	LARGE_INTEGER StreamSize,
	LARGE_INTEGER Streamferred,
	DWORD  StreamNumber,
	DWORD  CallbackReason,
	HANDLE SourceFile,
	HANDLE DestinationFile,
	LPVOID lpData
)
{
	//计算进度
	int i=(tFileSize.LowPart<=0) ?100 :(tTransferred.LowPart*1.0)/(tFileSize.LowPart)*100, j=i>>1;

	//显示递增进度
	while((j--)>perMARKS[0] && !(perMARKS[j])){perMARKS[j]=TRUE; fputc('=', stdout);}
	perMARKS[0]=i>>1;

	//显示进度百分比
	fprintf(stdout, "%02d%%\b\b\b", i);

	//回调
	return PROGRESS_CONTINUE;
}
//正则命名函数 
void ReCopy(
	int FLAG,             //复制模式开关
	const char* Org_Path, //工作目录
	char* RE_Match,       //要匹配的正则
	char* RE_Format       //要更名的正则
)
{
	//设置工作目录
	_chdir(Org_Path);
	
	struct _finddata_t FileS;	
	intptr_t handle=_findfirst("*",&FileS);
	bool M=(handle==-1)?TRUE:FALSE;
	
	while(!M){
		if(
			(strcmp(FileS.name, ".")==0) ||
			(strcmp(FileS.name,"..")==0)
		){
			M=(_findnext(handle, &FileS)!=0)?TRUE:FALSE;
			continue;
		}

		//匹配正则
		CRegexpT<char> regexp(RE_Match); 
		MatchResult result=regexp.Match(FileS.name);

		//执行复制
		if(result.IsMatched()){
    			//正则替换
			char* tpSTR=regexp.Replace(FileS.name, RE_Format);

			//绘进度条
			SetConsoleTextAttribute(handle_out, 1|4|8);  //紫色
			fprintf(stdout, "Copy... ");
			SetConsoleTextAttribute(handle_out, 2|8);    //绿色
			fprintf(stdout, "%s -> %s\n", FileS.name, tpSTR);
			SetConsoleTextAttribute(handle_out, 2|4|8);  //黄色
 			fputs(proGRESS, stdout);

			//初始化进度数组 
			memset(perMARKS, FALSE, 50);

			//强制删除
			if(FLAG==0x01){remove(tpSTR);}

			//复制文件
			if(!CopyFileEx(FileS.name, tpSTR, (LPPROGRESS_ROUTINE)CopyProgressRunTim, NULL, FALSE, COPY_FILE_FAIL_IF_EXISTS)){
				SetConsoleTextAttribute(handle_out, 4|8);
				fprintf(stdout, "Failed to copy, Check if the target path exists");
			}
			SetConsoleTextAttribute(handle_out, 1|2|4|8);//白色
			fputs("\n\n", stdout);
			//释放字串
			CRegexpT<char>::ReleaseString(tpSTR);
		}

		M=(_findnext(handle, &FileS)!=0)?TRUE:FALSE;
	}
	_findclose(handle);
	
	//递归子目
	if(FLAG==0x02){
		handle=_findfirst("*.*", &FileS);
		M=(handle==-1)?true:false;
		while(!M){
			if(FileS.attrib & _A_SUBDIR){
				if(
					(strcmp(FileS.name, ".")!=0) &&
					(strcmp(FileS.name,"..")!=0)
				){
					//正则复制
					ReCopy(FLAG, FileS.name, RE_Match, RE_Format);
				}
			}
			M=(_findnext(handle, &FileS)!=0);
		}
		_findclose(handle);
	}

	//还原目录、释放内存
	_chdir(Org_Path);
}

/*************MAIN主函数入口*************/
int main(int argc, char ** argv)
{
	//检查输入参数是否存在错误
	if(
		(argc!=3 && argc!=4        )   ||
		(argc==3 && argv[1][0]=='/')   ||
		(argc==4 && argv[1][0]!='/')   || 
		(
			argc>1          && 
			argv[1][0]=='/' && 
			(
				argv[1][1]=='?'||
				argv[1][1]=='h'||
				argv[1][1]=='H'
			)
		)
	){
		fputs(HELP_INFORMATION, stdout);
		exit(0);
	}

	//模式变量
	int i=1, FLAG=0x00;

	//解析开关
	if(argv[i][0]=='/' && argc==4){

		switch(argv[i][1])
		{
		case 'f':
		case 'F':
			FLAG=0x01, i++;
			break;
		case 'r':
		case 'R':
			FLAG=0x02, i++;
			break;
		default:
			//无法识别的开关
			fprintf(stderr, "Unrecognized switch '%s'\n", argv[i]);
			exit(1);
		}
	}

	handle_out=GetStdHandle(STD_OUTPUT_HANDLE);

	char* Org_Path=(char*)calloc(PATH_SIZE, sizeof(char));
	char* p=strrstr(argv[i], "\\\\");
	if(p!=NULL){
		strncpy_s(Org_Path, PATH_SIZE, (const char*)argv[i], (int)p-(int)argv[i]);
		ReCopy(FLAG,  Org_Path, p+2, argv[i+1]);
	}else{
		ReCopy(FLAG,   ".", argv[i], argv[i+1]);
	}

	return 0;
}
复制代码

RegCloseKey  
RegConnectRegistry  
RegCreateKey  
RegCreateKeyEx  
RegDeleteKey  
RegDeleteValue  
RegEnumKey  
RegEnumKeyEx  
RegEnumValue  
RegFlushKey  
RegGetKeySecurity  
RegLoadKey  
RegNotifyChangeKeyValue  
RegOpenKey  
RegOpenKeyEx  
RegQueryInfoKey  
RegQueryMultipleValues  
RegQueryValue  
RegQueryValueEx  
RegReplaceKey  
RegRestoreKey  
RegSaveKey  
RegSetKeySecurity  
RegSetValue  
RegSetValueEx  
RegUnLoadKey
复制代码