#include "person.h"

//首頁
int Print_Head(int * num)
{
 printf("\t***********************************************\n");
 printf("\t|     職工管理系統    |\n");
 printf("\t*---------------------------------------------*\n");
 printf("\t|   【1】新增  【2】刪改   |\n");
 printf("\t*   【3】查詢  【4】退出   *\n");
 printf("\t***********************************************\n");
 printf("\t◎請輸入想進行操作的序號【 】\b\b");
 *num=getchar();
 return SUCCESS;
}

//刪改畫面
int Print_DelRev(int* num)
{
 printf("\t******************************************\n");
 printf("\t*          *\n");
 printf("\t* 【1】刪除 【2】修改 【3】返回 *\n");
 printf("\t*          *\n");
 printf("\t******************************************\n");
 printf("\t◎請輸入想進行操作的序號【 】\b\b");
 *num=getchar();
 return SUCCESS;
}

//查詢畫面
int Print_Que(int* num)
{
 printf("\t******* ************************************\n");
 printf("\t*          *\n");
 printf("\t*  【1】全部查詢 【2】工號查詢  *\n");
 printf("\t*          *\n");
 printf("\t*  【3】模糊查詢 【4】返回菜單  *\n");
 printf("\t*          *\n");
 printf("\t******************************************\n");
 printf("\t◎請輸入想進行操作的序號【 】\b\b");
 *num=getchar();
 return SUCCESS;
}

//增加畫面
int Print_Add(int* num)
{
 int n=0;
 printf("\t******************************************\n");
 printf("\t*          *\n");
 printf("\t* 【1】新增 【2】添加 【3】返回 *\n");
 printf("\t*          *\n");
 printf("\t******************************************\n");
 printf("\t◎請輸入想進行操作的序號【 】\b\b");
 *num=getchar();
 return SUCCESS;
}

//標題頭
int Print_Tittle()
{
 printf("-------------------------------------------\n");
 printf(" 工號 | 姓名 |性別|年齡| 電話  \n");
 printf("-------------------------------------------\n");
 return SUCCESS;
}

//選擇將要打開的文件
int File_name()
{
 printf("\n\t◎請輸入您想要打開的文件:");
 //memset(filename,0,sizeof(filename));
 if(scanf("%s", filename)!=1){
 printf("\a 選擇文件錯誤!");
 return FAILED;
 }
 return SUCCESS;
}


//生成鏈表，參數n是add記錄數量
struct Employee *Creat() {
 printf("\n\t◎請輸入要添加的數據個數:");
 int n;
 if(scanf("%d", &n)!=1) {
 printf("\a error!");
 }
 struct Employee *head;
 struct Employee *p1, *p2;

 system("cls");
 for(int i=1;i<n+1;i++) {
 p1 = (struct Employee*)malloc(SIZE);
 Print_Tittle();
 scanf("%s%s%s%d%s",p1->num,p1->name,p1->sex,
 p1->age,p1->tel);
 p1->next = NULL;
 if(i==1) {
 head = p2 = p1;
 }
 else {
 p2->next = p1;
 p2 = p1;
 }
 }
 return(head);
}

//建立新文件
int WriteData_wb(struct Employee *head) {
 FILE *fp;
 struct Employee *p;
 if((fp = fopen(filename, "wb"))==NULL)
 printf("\a 打開文件錯誤!"); 
 p = head;
 while(p!=NULL) {
 if(fwrite(p,SIZE,1,fp)!=1) {
 printf("\b 寫入數據出錯\n");
 fclose(fp);
 return FAILED;
 }
 p=p->next;
 }
 fclose(fp);
 return SUCCESS;
}

//在已有文件添加
int WriteData_ab(struct Employee *head) {
 FILE *fp;
 struct Employee *p;
 if((fp = fopen(filename, "ab"))==NULL)
 printf("\a 打開文件錯誤!");
 p = head;
 while(p!=NULL) {
 if(fwrite(p,SIZE,1,fp)!=1) {
 printf("\b 寫入數據出錯\n");
 fclose(fp);
 return FAILED;
 }
 p=p->next;
 }
 fclose(fp);
 return SUCCESS;
}


//增加
int Emp_Add()
{
 system("cls");
 getchar();
 int add;
 while(1){
 system("cls");
 Print_Add(&add);
 while(add!='1' && add!='2' && add!='3') {
 putchar('\a');
 printf("◎請重新輸入有效序號(1~3):【 】\b\b");
 add=getchar();
 }
 switch(add){
 case '1':
 WriteData_wb(Creat());
 printf("\n◎新建文件成功數據已保存√\n");
 system("pause");
 system("cls");
 Emp_Add();
 break;
 case '2':
 WriteData_ab(Creat());
 printf("\n◎數據已成功添加√\n");
 system("pause");
 system("cls");
 Emp_Add();
 break;
 case '3':
 system("cls");
 getchar();
 Emp_Return();
 break;
 }
 }
 return SUCCESS;
}

//讀取文件的數據到鏈表中返回鏈表head指針
struct Employee *Read() {
 struct Employee *head = NULL;
 struct Employee *p1, *p2;//s = p1;p = p2;

 FILE *fp;
 if((fp=fopen(filename,"rb+"))==NULL)
 {
 printf("打開文件出錯\n");
 exit(0);
 }
 while(!feof(fp)) {
 if((p1=(struct Employee*)malloc(SIZE))==NULL){
 printf("內存申請出錯\n");
 fclose(fp);
 exit(0);
 }
 if(fread(p1,SIZE,1,fp)!=1){
 free(p1);
 break;
 }
 if(head==NULL)
 head=p2=p1;
 else{
 p2->next=p1;
 p2=p1;
 }
 }
 fclose(fp);
 return (head);
}

//刪除
int Emp_Del()
{
 struct Employee* head;
 struct Employee* pt1,*pt2;
 char str_num[10];
 memset(str_num,0,sizeof(str_num));
 printf("\n◎請輸入你要刪除的學號信息：");
 scanf("%s",str_num);
 pt1=Read();
 pt2=pt1->next;
 head=pt1;
 while(pt2!=NULL){
 if(!strcmp(pt1->num,str_num)){
 WriteData_wb(pt2);
 }else if(!strcmp(pt2->num,str_num)){
 pt1->next=pt2->next;
 WriteData_wb(head);
 }
 pt2=pt2->next;
 pt1=pt1->next;
 }
 if(pt2!=NULL){
 printf("\t◎沒有你要刪除的數據");
 }
 return SUCCESS;
}

//修改
int Emp_Rev()
{
 struct Employee* pt1,*pt2,*head;
 char str_num[10];
 printf("\t◎請輸入要修改的學號信息：");
 scanf("%s",str_num);
 pt1=Read();
 pt2=pt1->next;
 head=pt1;
 while(pt2!=NULL){
 if(strcmp(pt1->num,str_num)==0) {
 Print_Tittle();
 scanf("%s%s%s%d%s",pt1->num,pt1->name,pt1->sex,
 pt1->age,pt1->tel);
 WriteData_wb(head);
 }
 else if(strcmp(pt2->num,str_num)==0) {
 Print_Tittle();
 scanf("%s%s%s%d%s",pt1->num,pt1->name,pt1->sex,
 pt1->age,pt1->tel);
 WriteData_wb(head);
 }
 pt2 = pt2->next;
 pt1 = pt1->next;
 }
 if(pt2!=NULL)
 printf("數據庫中沒有存儲您要刪除的數據！\n");
 return 0;
}

//刪改
int Emp_DelRev()
{
 
 getchar();
 int delrev=0;
 while(1){
 system("cls");
 Print_DelRev(&delrev);
 while(delrev!='1' && delrev!='2' && delrev!='3'){
 putchar('\a');
 //getchar();
 printf("○請重新輸入有效序號(1~3):【 】\b\b");
 delrev=getchar();
 }
 switch(delrev){
 case '1':
 Emp_Del();
 printf("\n◎已成功刪除指定數據!");
 system("pause");
 getchar();
 break;
 case '2':
 Emp_Rev();
 printf("\n◎已成功修改指定數據!");
 system("pause");
 getchar();
 break;
 case '3':
 system("cls");
 getchar();
 Emp_Return();
 break;
 }
 }
 return SUCCESS;
}

//總體查詢
int Emp_QueAll()
{
 struct Employee *pt;
 pt = Read();
 Print_Tittle();
 do {
 printf("%2s%5s%4s%2d%3s\n",
 pt->num,pt->name,pt->sex,pt->age,pt->tel);
 pt = pt->next;
 }while(pt!=NULL);
 printf("\n\n");
 return SUCCESS;
}

//工號查詢
int Emp_QueNum() 
{
 struct Employee *pt;
 char str_num[10];
 printf("\t◎請輸入您要查詢的學號:");
 scanf("%s", str_num);
 pt = Read();
 Print_Tittle();
 do {
 if(!strcmp(pt->num,str_num)) {
 printf("%2s%5s%4s%2d%3s\n",
 pt->num,pt->name,pt->sex,pt->age,pt->tel);
 printf("\n\n");
 return 0;
 }
 pt = pt->next;
 }while(pt!=NULL);
 printf("\t數據庫中沒有存儲您要查詢的數據！\n");
 printf("\n\n");
 return SUCCESS;
}

//模糊查詢
int Emp_QueVague()
{
 struct Employee *pt;
 char str_vague[20];
 int m=0;
 printf("\t◎請輸入您要查詢的關鍵詞:");
 scanf("%s", str_vague);
 pt = Read();
 Print_Tittle();
 do {
 if(strstr(pt->num,str_vague)!=0||strstr(pt->name,str_vague)!=0
 ||strstr(pt->sex,str_vague)!=0||strstr(pt->age,str_vague)!=0
 ||strstr(pt->tel,str_vague)!=0) {
 printf("%2s%5s%4s%2d%3s\n",
 pt->num,pt->name,pt->sex,pt->age,pt->tel);
 m = 1;
 }
 pt = pt->next;
 }while(pt!=NULL);
 if(!m)
 printf("數據庫中沒有存儲您要查詢的數據！\n");
 printf("\n\n");
 return SUCCESS;
}

//查詢
int Emp_Que()
{
 system("cls");
 int que;
 while(1){
 system("cls");
 Print_Que(&que);
 while(que!='1' && que!='2' && que!='3'){
 putchar('\a');
 printf("\t○請重新輸入有效序號(1~3):【 】\b\b");
 que=getchar();
 }
 switch(que){
 case '1':
 Emp_QueAll();
 system("pause");
 getchar();
 break;
 case '2':
 Emp_QueNum();
 system("pause");
 getchar();
 break;
 case '3':
 Emp_QueVague();
 system("pause");
 getchar();
 break;
 case '4':
 Emp_Return();
 system("pause");
 getchar();
 break;
 }
 }
 return SUCCESS;
}

//返回
int Emp_Return()
{
 Emp_Menu();
 return SUCCESS;
}

//退出
void Emp_Quit()
{
 printf("\n\t◎謝謝使用！");
 system("pause");
 exit(0);
}

//menu函數
int Emp_Menu()
{
 //int* Menu=NULL;
 int menu=0;
 //Print_Head(&Menu);
 //menu=*Menu;
 //delete Menu;
 //printf("%d\n",menu);
 Print_Head(&menu);
 
 while(menu!='1' && menu!='2' && menu!='3'&& menu!='4') {
 printf("error! please input the right number!\n");
 putchar('\a');
 //getchar();
 printf("◎請重新輸入有效序號(1~4):【 】\b\b");
 menu=getchar();
 }
 switch(menu){
 case '1': 
 File_name();
 Emp_Add();
 break;
 case '2': 
 File_name();
 Emp_DelRev();
 break;
 case '3': 
 File_name();
 Emp_Que();
 break;
 case '4': 
 Emp_Quit();
 break;
 }
 return SUCCESS;
}
/*void Sig(int n)
{
 printf("\n\t◎程序將要退出，操作已保存！");
 system("pause");
 exit(0);
}*/
int main()
{
 //printf("\t◎使用Ctrl+C終止現在程序的運行！\n");
 //signal(SIGINT,Sig);
 Emp_Menu();
 return SUCCESS;
}

#ifndef __PERSON_H__
#define __PERSON_H_
#define _CRT_SECURE_NO_DEPRECATE
#define _CRT_SECURE_CPP_OVERLOAD_STANDARD_NAMES 1
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <conio.h>
#include <signal.h>
#define SUCCESS  0
#define FAILED (!SUCCESS)
#define SIZE sizeof(struct Employee)
//全局變量
int flag=0;//返回標志
char filename[30];//要打開的文件名
//員工信息結構體
struct Employee{
 char name[20];
 char tel[15];
 char sex[5]; 
 char num[10];
 char age[3];
 struct Employee* next;
};

//建立一個鏈表
struct Employee* creat(int n);

//主函數
int Emp_Menu();
//首頁打印
int Print_Head();
//刪改打印
int Print_DelRev();
//增加打印
int Print_Add();
//表頭打印
int Print_Tittle();
//新建文件寫入
int WriteData_wb(struct Employee *head);
//在已有的文件寫入
int WriteData_ab(struct Employee *head);
//添加
int Emp_Add();
//刪除
int Emp_Del();
//修改
int Emp_Rev();
//查詢
int Emp_Que();
//返回
int Emp_Return();
//退出
void Emp_Quit();

#endif //__PERSON_H__