#include "btree.h"

//菜單的顯示,并接受用戶返回的選項
char menu_selete(){
	char ch;
	system("cls");	
	printf("\t**************** 基于 B- 樹實現(xiàn)的圖書管理系統(tǒng) ***************\n\n");
    printf("\t\t\t1.輸入書信息. \n");
    printf("\t\t\t2.查找書信息. \n");
    printf("\t\t\t3.顯示書信息. \n");
    printf("\t\t\t4.刪除書信息. \n");
	printf("\t\t\t5.借閱. \n");
	printf("\t\t\t6.還書. \n");
	printf("\t\t\t7.查找某作者的全部書籍\n");
	printf("\t\t\t8.退出. \n");
    printf("\t*************************************************************\n\n");
	//gotoxy(25,23);
	printf("\t\t請選擇你所需要的操作(1~8):");
	do{		
		ch = getch(); 
	}while(ch < '1' || ch > '8');
	return ch;
}


void main(){
	BTree root = NULL; //樹根
	KeyType k;
	Result rs;  //查找的結果

	writeLog("歡迎使用該圖書管理系統(tǒng),程序開始"); //寫入日記文件中

	while(1){
		switch(menu_selete()){
		case '1':
			system("cls");
	        printf("\t\t------------------ 錄入書信息 -----------------\n");
			InBookMess(k);
			InsertBTree(root,k);
			printf("\t\t\t--------- 錄入結束 ---------\n\n");
			printf("\t\t\t當前書庫的庫存信息如下：\n");
			writeLog("當前的庫存后的庫存如下:(下面顯示的只是書號)\n     " + display(root,0));
			printf("\t\t\t\t按任意鍵返回");
	        getch();
			break;
		case '2':
			system("cls");
	        printf("\t\t------------------ 查找書信息 -----------------\n\n");
			InBookKey(k);
			printf("\n");
			rs = SearchBTree(root,k);			
			if(rs.tag == 1){
				ShowBookMess(rs.pt->key[rs.i]);
				writeLog("查找書號" + itos(k.key) + "成功!");  //將會話添加到日記文件中
			}
			else{
				printf("\t\t\t\t你要查找的書號 %d 不存在!\n\n",k.key);	
				writeLog("查找書號" + itos(k.key) + "不存在!");  //將會話添加到日記文件中
			}
			printf("\n\t\t------------------ 查找結束 -----------------\n");
			printf("\t\t\t\t按任意鍵返回");
			getch();
			break;
		case '3':
			system("cls");
	        printf("\t\t------------------ 全部的書 -----------------\n\n");
			writeLog("顯示書的全部信息如下:(下面顯示的只是書號)\n     " + display(root,0));
			printf("\t\t------------------ 顯示完畢 -----------------\n");
			printf("\t\t\t\t按任意鍵返回");
			getch();
			break;
		case '4':
			system("cls");
			printf("\t\t------------------ 刪除書 -----------------\n\n");
			InBookKey(k);
			if(DeleteBTree(root,k) == OK){
				printf("\t\t\t\t刪除 %d 成功!\n\n",k.key);
				printf("\t\t當前書庫中有如下書:\n");
				writeLog("刪除書號" + itos(k.key) + "成功!\n\t當前的庫存后的庫存如下:(下面顯示的只是書號)\n     " + display(root,0));
			}
			else{
				printf("\t\t\t\t刪除 %d 失敗!\n\n",k.key);
				printf("\t\t當前書庫中有如下書:\n");
				writeLog("當前的庫存后的庫存如下:(下面顯示的只是書號)\n     " + display(root,0));
			}
			printf("\n\t\t\t\t按任意鍵返回");
			getch();
			break;	
		case '5':
			system("cls");
			printf("\t\t------------------ 借閱 -----------------\n\n");
			InBookKey(k);
			if(BorrowBook(root,k)){
				printf("\n\t\t\t借閱成功!\n");
			}
			else{
				printf("\n\t\t\t借閱失敗!\n");
			}
			printf("\n\t\t\t\t按任意鍵返回");
			getch();			
			break;
		case '6':
			system("cls");
			printf("\t\t------------------ 還書 -----------------\n\n");
			InBookKey(k);
			if(ReturnBook(root,k)){
				printf("\n\t\t\t歸還成功!\n");
			}			
			else{
				printf("\n\t\t\t歸還失敗!\n");
			}
			printf("\n\t\t\t\t按任意鍵返回");
			getch();
			break;
		case '7':
			system("cls");
			searchAuthorB(root);  //查找和作者相關的書籍
			printf("\n\t\t\t\t按任意鍵返回");
			getch();			
			break;
		case '8':
			printf("\n\n\n\t\t\t");
			writeLog("退出程序\n\n\n");
			exit(0);
		}
	}
}