///////////////////////////////////////////////////////////////////////////////
//
//  文	件: Backup.c
//		執行備分工作的相關函數
//
//  作  者: 江南孤峰 
//  聯  系：QQ: 403324669   
//  時  間: 2007--3--3
//
///////////////////////////////////////////////////////////////////////////////


#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <direct.h>
#include <io.h>
#include <conio.h>
#include "FilePass.h"

// 創建執行備分工作必要的目錄 
int	CreateBackupDirect(void){
	char strSavePath[_MAX_PATH] = {""};
	
	_getcwd(strSavePath,_MAX_PATH);
	if(!strcmp(strBackupDirect,""))	// 備分目錄的獲取不成功
		return FAILED;
	if(_chdir(strBackupDirect))	// 目錄不存在		
		_mkdir("backup");
	else
		_chdir(strSavePath);	// 恢復當前目錄	
	return SUCCESS;
}

// 創建執行備分工作必要的文件
int	CreateBackupFile(void){
	char strSavePath[_MAX_PATH] = {""};
	FILE *fp;

	_getcwd(strSavePath,_MAX_PATH);
	_chdir(strBackupDirect);
	if(_access("BackFile.ini",0)){    // 文件BackFile.ini不存在
		if((fp = fopen("BackFile.ini","w+")) != NULL){
			fprintf(fp,"###");// 文件結束標記
			fclose(fp);
		}
		else
			return FAILED;
	}
	if(_access("FileTimes.ini",0)){	
		if((fp = fopen("FileTimes.ini","w+")) != NULL){
			fprintf(fp,"0");  // 重命名因子初始化為 0
			fclose(fp);
		}
		else
			return FAILED;
	}
	_chdir(strSavePath);
	return SUCCESS;
}

// 創建執行備分工作必要的目錄，文件
void	CreateBackupDirectAndFile(void){
	if(CreateBackupDirect() == FAILED){
		puts("創建執行備分工作必要的目錄失敗\n");
		return ;
	}
	if(CreateBackupFile() == FAILED)
		puts("創建執行備分工作必要文件失敗\n");
}

// 讀取備分目錄中的文件數目
int	ReadFileCounts(void){
	FILE *fp;
	int  count = 0;

	if((fp = fopen("FileTimes.ini","r+")) == NULL){
		puts("文件 FileTimes.ini 不存在\n");
		if(CreateBackupFile() == SUCCESS){
			fp = fopen("FileTimes.ini","r+");
			puts("文件 FileTimes.ini 已創建\n");
		}
		else
			return count;
	}
	fscanf(fp,"%d",&count);
	rewind(fp);
	fprintf(fp,"%d",++count);
	fclose(fp);
	return count;
}

// 將當前目錄下的文件 [strFileName] 移動到目錄 [strDestDirect]
int	MyMoveFile(char *strFileName,char *strDestDirect){
	FILE *fp;
	long fileSize = 0;
	char *p;
	char strSavePath[_MAX_PATH];

	if((fp = fopen(strFileName,"r")) == NULL)
		return FAILED;
	fileSize = _filelength(_fileno(fp));
	if((p = (char*)malloc(fileSize)) == NULL){
		puts("內存不足 !\n");
		return FAILED;
	}
	fread(p,fileSize,1,fp);
	fclose(fp);
	if(remove(strFileName) != SUCCESS)
		printf("刪除文件 %s 失敗\n",strFileName);
	_getcwd(strSavePath,_MAX_PATH);	
	_chdir(strDestDirect);
	if((fp = fopen(strFileName,"w+")) == NULL){
		printf("丟失文件 %s\n",strFileName);
		free(p);
		return FAILED;
	}
	fwrite(p,fileSize,1,fp);
	fclose(fp);
	_chdir(strSavePath);
	free(p);
	return SUCCESS;
}


void	BackupFile(char *strFileName){
	char	strBkFileName[FILE_LENGTH + 6];
	char	strSavePath[_MAX_PATH + 2];
	int	count = 0;
	FILE	*fp;

	if(MyMoveFile(strFileName,strBackupDirect) == FAILED){
		printf("文件 %s 備分失敗\n",strFileName);
		return ;
	}
	_getcwd(strSavePath,_MAX_PATH);
	// 在備分目錄下重命名文件，并將原目錄信息保存
	_chdir(strBackupDirect);
	count = ReadFileCounts();
	sprintf(strBkFileName,"%s#%d",strFileName,count);
	rename(strFileName,strBkFileName);
	if( (fp = fopen("BackFile.ini","a")) == NULL &&
	    (CreateBackupFile() == FAILED) ){
		printf("備分文件時,丟失文件 %s 的原路徑\n",strFileName);
		return;
	}
	fprintf(fp,"%s\n",strBkFileName); 
	fprintf(fp,"%s\n",strSavePath);
	fprintf(fp,"###");		// 文件結束標記
	printf("文件 %s 備分完成\n",strFileName);
	fclose(fp);
	_chdir(strSavePath);
}

// 從備分目錄中恢復文件到原位置
void	GetBackFile(
		char *strAddSuffix ,	// 加密文件的后綴
		char *strSourceSuffix   // 原文件的后綴
	){
	char	strSavePath[_MAX_PATH + 2] = {""};
	char	strFileName[FILE_LENGTH + 3] = {""};
	char	strOldFileName[FILE_LENGTH] = {""};
	char	strFileDirect[_MAX_PATH + 4] = {""};
	char	strFileDelete[FILE_LENGTH + 2] = {""};
	int	iCountSuccess = 0;
	FILE	*fp;

	_getcwd(strSavePath,_MAX_PATH);	
	_chdir(strBackupDirect);	
	if((fp = fopen("BackFile.ini","r")) == NULL){
		puts("打開文件 BackFile.ini 失敗, 文件無法恢復到原位置\n");
		return;
	}
	while(TRUE){
		// 獲取文件名和目錄信息
		fgets(strFileName,FILE_LENGTH,fp);
		if(!strcmp(strFileName,"###"))	     // ### 為結束標記
			break;
		strcpy(strFileName,strFileName+3);   // 忽略 ###
		*strchr(strFileName,'\n') = '\0';    // 去掉讀入的回車符
		strcpy(strOldFileName,strFileName);
		*strrchr(strOldFileName,'#') = '\0'; // 取得備分文件的原名字

		fgets(strFileDirect,_MAX_PATH,fp);
		*strchr(strFileDirect,'\n') = '\0';
		// 移動文件
		if(MyMoveFile(strFileName,strFileDirect) == FAILED){
			printf("文件 \"%s\" 恢復失敗\n\n",strOldFileName);
			continue;
		}
		_chdir(strFileDirect);
		if(rename(strFileName,strOldFileName) != SUCCESS){
			remove(strFileName);	   
			printf("發現同名文件 %s\n",strOldFileName);
		}
		else{
			printf("文件 \"%s\" 恢復成功\n",strOldFileName);
			iCountSuccess ++;
		}
		// 刪除加密文件
		if(  GetDestFileName(
			strFileDelete,
			strOldFileName,
			strSourceSuffix,
			strAddSuffix) == SUCCESS &&
		     !access(strFileDelete,0)
		  ){	 
			if(remove(strFileDelete) == SUCCESS)
				printf("刪除文件 \"%s\" 完成\n",strFileDelete);
			else
				printf("刪除文件 \"%s\" 失敗\n",strFileDelete);
		}
		_chdir(strBackupDirect);
		putchar('\n');
	}
	fclose(fp);
	_chdir(strBackupDirect);
	if(system("del *.*/q/f") == FAILED)
		puts("清空備分目錄失敗\n");
	else
		puts("清空備分目錄完成\n");
	CreateBackupFile();
	printf("共恢復文件:%d\n",iCountSuccess);
	_chdir(strSavePath);
}