Lab09실습 - 파일 입출력을 이용한 전화번호부

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
 
#define DELETED "DELETED"
#define FILENAME "phone.txt"
 
/*PHONE 구조체 정의*/
typedef struct phone {
    char name[20];
    char phone[20];
} PHONE;
 
/*메뉴*/
void displaymenu()
{
    printf("\n***전화번호부***\n");
    printf("(1) 목록\n");
    printf("(2) 검색\n");
    printf("(3) 추가\n");
    printf("(4) 변경\n");
    printf("(5) 삭제\n");
    printf("(99) 종료\n");
    printf("-----------------\n");
    printf("선택하시오 : ");
}
 
/*레코드 수(데이터 수)*/
getnumberofdata(FILE* fp)
{
    int filesize;
    int n;
    
    fseek(fp, 0, SEEK_END);//파일 끝으로 이동
    filesize = ftell(fp);//파일 끝의 위치(바이트형식)
    
    n = filesize / sizeof(PHONE);//파일 끝의 위치(바이트형식)를 구조체의 크기(40byte)로 나누면 레코드 개수 구함
    return n;//레코드 개수
}
 
/*data하나 보여줌*/
void disp_one(PHONE* data)
{
    printf("이름 : %s\t전화 : %s\n", data->name, data->phone);
}
 
/*전체 데이터 보여줌*/
void list(FILE* fp)
{
    PHONE data;//PHONE 구조체 변수 선언
    int ndata;
    ndata = getnumberofdata(fp);//레코드 수 얻음
 
    printf("\n***전체 목록***\n");
    
    if (ndata == 0) {//레코드 수가 0이면 데이터 없음
        printf("데이터가 없습니다.\n");
    }
 
    fseek(fp, 0, SEEK_SET);//처음으로 이동
 
    while (ndata--) {//ndata가 0이 되면 반복문 탈출
        fread(&data, sizeof(PHONE), 1, fp);//파일에서 데이터 읽어서 구조체data에 저장
        if (strcmp(data.name, DELETED) == 0 && strcmp(data.phone, DELETED) == 0) {//만약 삭제된 레코드라면
            continue;//건너뛴다
        }
        disp_one(&data);//저장된 구조체data를 하나씩 출력
    }
}
 
/**/
int search(FILE* fp)
{
    PHONE data;
    char name[20];//찾을 이름
    int n;
    int ndata;//전체 레코드 수
    ndata = getnumberofdata(fp);//레코드 수 얻음
 
    printf("\n***자료 찾기***\n");
    if (ndata == 0) {//전체 레코드 수가 없다면
        printf("데이터가 없음.\n");
        return -1;//-1반환
    }
    printf("찾을 이름 : ");
    scanf("%s", name);
    
    fseek(fp, 0, SEEK_SET);//파일의 처음으로 이동해서 0번 건너뜀
    
    for (n = 0; n < ndata; n++) {
        fread(&data, sizeof(PHONE), 1, fp);//파일에서 데이터 한개씩 가져온다
        
        if (strcmp(data.name, DELETED) == 0 && strcmp(data.phone, DELETED) == 0) {//삭제된 레코드면 skip
            continue;
        }
        if (strcmp(data.name, name) == 0) {//strcmp는 참이면 0반환
            break;//중지
        }
    }
    
    if (n >= ndata) {
        printf("찾는 데이터가 없습니다.\n");
        return -1;
    }
 
    disp_one(&data);
    return n;//찾은 데이터의 레코드 번호(레코드번호 0번부터 시작)
}
 
/*데이터 넣기*/
void inputone(PHONE* data)
{
    printf("이름 : ");
    scanf("%s", data->name);
    printf("전화번호 : ");
    scanf("%s", data->phone);
}
 
/*빈 레코드 찾는다(삭제된 레코드)*/
int getblank(FILE* fp)
{
    PHONE data;
    int n;
    int ndata = getnumberofdata(fp);
 
    if (ndata == 0) {
        return 0;//처음 데이터 넣을 때 첫번째 레코드에서 시작
    }
 
    for (n = 0; n < ndata; n++) {
        fread(&data, sizeof(PHONE), 1, fp);
        if (strcmp(data.name, DELETED) == 0 && strcmp(data.phone, DELETED) == 0) {//삭제된것
            return n;//삭제된 레코드 번호 리턴
        }
    }
 
    return -1;//삭제된 레코드가 없다
}
 
/*데이터 추가*/
void add(FILE* fp)
{
    PHONE data;
    int blank_n = getblank(fp);//빈 공간이 어디에 있는지(레코드 번호)
 
    printf("\n***자료 추가***\n");
 
    //구조체에 추가할 데이터 내용을 저장
    printf("이름 : ");
    scanf("%s", &data.name);
    printf("전화 번호 : ");
    scanf("%s", &data.phone);
 
    if (blank_n > 0) {//blank_n이 -1이 아니면 삭제된 곳이 있다는 뜻
        fseek(fp, blank_n*sizeof(PHONE), SEEK_SET);//처음에서 삭제된 레코드로 이동
        fwrite(&data, sizeof(PHONE), 1, fp);//구조체data를 파일에 쓴다
    }
    else if(blank_n == 0) {
        fwrite(&data, sizeof(PHONE), 1, fp);
    }
    else {//삭제된 곳이 없다면
        fseek(fp, 0, SEEK_END);//끝으로 이동해서
        fwrite(&data, sizeof(PHONE), 1, fp);//구조체data를 파일에 쓴다
    }
}
 
/*데이터 변경하기*/
void updateone(FILE* fp, int n)
{
    PHONE data;
    printf("\n***변경***\n");
    if (n == -1) {//n=-1이면 검색먼저 해야함
        printf("검색 먼저 해주세요\n");
        return;
    }
 
    fseek(fp, (n)*sizeof(PHONE), SEEK_SET);//처음에서 n(레코드 번호)으로 이동해서
    fread(&data, sizeof(PHONE), 1, fp);//파일에서 데이터를 읽어와서 구조체data에 넣는다
    disp_one(&data);//구조체data의 내용을 출력
 
    //변경하고자 하는 내용을 입력
    printf("이름 : ");
    scanf("%s", &data.name);
    printf("전화번호 : ");
    scanf("%s", &data.phone);
 
    fseek(fp, (n) * sizeof(PHONE), SEEK_SET);//처음에서 레코드 번호로 이동
    fwrite(&data, sizeof(PHONE), 1, fp);//구조체의 데이터를 파일에 쓴다
}
 
/*데이터 삭제하기*/
void deletone(FILE* fp, int n)
{
    PHONE deletdata = { DELETED, DELETED };
    PHONE data;
    int ans;
    int ndata;
 
    printf("\n***삭제***\n");
    
    if (n == -1) {//n이 -1이면 검색부터
        printf("검색부터 해주세요\n");
        return;
    }
 
    fseek(fp, (n) * sizeof(PHONE), SEEK_SET);//삭제하려는 위치로 이동
    fread(&data, sizeof(PHONE), 1, fp);//삭제하려는 레코드의 정보를 구조체data에 담는다
    disp_one(&data);//삭제하려는 레코드의 정보를 출력
 
    printf("정말로 삭제하시겠습니까?(1/0)\n");
    scanf("%d", &ans);//삭제하려면 1 , 아니라면 0
 
    if (ans == 1) {
        fwrite(&deletdata, sizeof(PHONE), 1, fp);//삭제하려는 위치로 이동한 상태에서 구조체 deletdata를 파일에 쓴다
        printf("삭제완료\n");
        return;
    }
    else {
        printf("삭제취소\n");
        return;
    }
}
 
int main(void)
{
    FILE* fp = NULL;
    int record_number;//레코드 번호 어떤 동작을 할지 결정
    int menu;//메뉴번호
 
    record_number = -1;//-1로 기본설정
 
    if ((fp = fopen(FILENAME, "r+b")) == NULL) {//읽기모드로 열었을 때 파일열기 실패하면
        fp = fopen(FILENAME, "w+b");//쓰기모드로 연다
        if (fp == NULL) {//쓰기모드로 여는 것도 실패하면
            printf("파일을 열 수 없습니다.\n");//오류 메세지 출력
            exit(0);
        }
    }
 
    while (1) {
        do {
            displaymenu();//메뉴보이기
            scanf("%d", &menu);//메뉴 입력
        } while (menu != 99 && (menu < 1 || menu > 5));//menu가 99가 아니고 1보다 작거나 5보다 큰 수면 계속 반복
 
        if (menu == 99) {
            break;
        }
 
        switch (menu) {
        case 1://목록
            list(fp);
            break;
        case 2://검색
            record_number = search(fp);//레코드 번호(0부터)
            break;
        case 3://추가
            add(fp);
            record_number = -1;
            break;
        case 4://변경
            updateone(fp, record_number);
            record_number = -1;
            break;
        case 5://삭제
            deletone(fp, record_number);
            record_number = -1;
            break;
        }
    }
    fclose(fp);
    return 0;
}