#include <stdio.h>
#include <stdbool.h>
#include <stdlib.h>
#include <string.h>
#include <assert.h>
#include <getch.h>
#define USER_MAX 100
typedef struct User
{
    char name[20];
    char passwd[7];
    char phone[12];
    char lock;
}User;
User user[USER_MAX];
size_t cnt;
char get_cmd(char start,char end)
{
    assert(start<=end);
    puts("--------------------");
    printf("请输入指令：");
    for(;;)
    {
        char cmd=getch();
        if(start<=cmd && cmd<=end)
        {
            printf("%c\n",cmd);
            return cmd;
        }
    }
}
void print_sec(const char* msg,float sec)
{
    printf("\33[01;32m %s \33[00m\n",msg);
    usleep(1000000*sec);
}
char* get_str(char* str,size_t size)
{
    assert(NULL!=str && size>1);
    size_t len=strlen(fgets(str,size,stdin));
    if('\n'==str[len-1])
        str[len-1]='\0';
    else
        while('\n'!=getchar());
    return str;

}
char* get_passwd(char* passwd,size_t size)
{
    int i=0;
    while(i<size-1)
    {
        passwd[i]=getch();
        if(127==passwd[i])
        {
            if(i>0)
            {

                i--;
                printf("\b \b");
            }
            continue;
        }
        i++;
        printf("*");
    }
    passwd[size-1]='\0';
    printf("\n");
    return passwd;
}
bool yes_or_no(void)
{
    printf("是否确认此操作（y/n）?");
    for(;;)
    {
        char cmd=getch();
        if('y'==cmd || 'Y'==cmd || 'n'==cmd || 'N'==cmd)
        {
            printf("%c\n",cmd);
            return 'y'==cmd || 'Y'==cmd;
        }
    }
}
int login_index=-1;
bool logoff_user(void)
{
    if(yes_or_no())
    {
    //把最后一个用户移动给要删除的用户位置
    user[login_index]=user[cnt-1];
    cnt--;
    login_index=-1;
    print_sec("注销用户成功！",0.75);
        return true;
    }
    else
    {
        print_sec("取消注销操作！",0.75);
        return false;

    }
}
void modify_passwd(void)
{
    char passwd[7];
    printf("请输入旧密码：");
    get_passwd(passwd,sizeof(passwd));
    if(strcmp(passwd,user[login_index].passwd))
    {
        print_sec("旧密码错误，无法修改密码！",0.75);
        return;
    }
    printf("请输入新密码：");
    get_passwd(passwd,sizeof(passwd));
    char repasswd[7];
    printf("请输入新密码：");
    get_passwd(repasswd,sizeof(repasswd));
    if(0==strcmp(passwd,repasswd))
    {
        strcpy(user[login_index].passwd,passwd);
        print_sec("修改成功",0.75);
    }
    else
        print_sec("修改失败",0.75);

}
void modify_user(void)
{
    printf("请输入用户电话：");
    get_str(user[login_index].phone,sizeof(user[login_index].phone));
    print_sec("修改用户信息成功！",0.75);
}
void sub_menu(void)
{
    for(;;)
    {
        system("clear");
        puts("****用户信息管理模块****");
        puts("1、注销用户");
        puts("2、修改密码");
        puts("3、修改信息");
        puts("4、退出登录");
        switch(get_cmd('1','4'))
        {
            case '1':
                if(logoff_user())
                    return;
                else
                    break;
            case '2':modify_passwd();break;
            case '3':modify_user();break;
            case '4':login_index=-1;return;
        }
    }
}
void menu(void)
{
    system("clear");
    puts("*****用户管理模块*****");
    puts("1、注册用户");
    puts("2、用户登录");
    puts("3、遍历用户");
    puts("4、退出系统");
}
void anykey_contonue(void)
{
    puts("按任意键继续...");
    getch();
}
int query_user(const char* name)
{
    for(int i=0;i<cnt;i++)
    {
        if(0==strcmp(user[i].name,name))
            return i;
    }
    return -1;
}
void register_user(void)
{
    //判断用户是否已满
    if(cnt>=USER_MAX)
    {
        print_sec("用户数量已满，无法添加！",0.75);
        return;
    }
    //输入用户名
    printf("请输入用户名：");
    get_str(user[cnt].name,sizeof(user[cnt].name));

    //检查是否被占用
    if(-1<query_user(user[cnt].name))
    {
        print_sec("该用户已被占用，无法注册！",0.75);
        return;
    }
    //输入密码
    printf("请输入密码：");
    get_passwd(user[cnt].passwd,sizeof(user[cnt].passwd));
    //再一次输入密码
    char repasswd[7];
    printf("请在一次输入密码：");
    get_passwd(repasswd,sizeof(repasswd));
    //确认密码
    if(strcmp(user[cnt].passwd,repasswd))
    {
        print_sec("两次输入的密码不符，注册失败！",0.75);
        return;
    }
    //输入电话号码
    printf("请输入用于找回密码的手机号：");
    get_str(user[cnt].phone,sizeof(user[cnt].phone));
    //初始化锁定标记
    user[cnt].lock=0;
    //用户数量+1
    cnt++;


}
void longin_user(void)
{
    //输入用户名
    char name[20];
    printf("请输入用户名：");
    get_str(name,sizeof(name));
    //查询用户名是否存在
    int index=query_user(name);
    if(-1==index)
    {
        print_sec("该用户不存在，登录失败！",0.75);
        return;
    }
    //判断是否被锁
    if(user[index].lock>=3)
    {
        print_sec("该用户已被锁定，请联系大师兄！",0.75);
        return;
    }
    char passwd[7];
    printf("请输入密码：");
    get_passwd(passwd,sizeof(passwd));
    //比较密码
    if(0==strcmp(user[index].passwd,passwd))
    {
        user[index].lock=0;
        login_index=index;
        sub_menu();
        return;
    }
    switch(++user[index].lock)
    {
        case 1:print_sec("密码错误，你还有2次机会！",0.75);break;
        case 2:print_sec("密码错误，你还有1次机会！",0.75);break;
        case 3:print_sec("密码错误，账号锁定！",0.75);break;
    }

}
void show_user(void)
{
    for(int i=0;i<cnt;i++)
    {
        printf("%s %s %s %c\n",user[i].name,user[i].passwd,user[i].phone,user[i].lock);
    }
    anykey_contonue();
}
int main(int arge,const char* argv[])
{
    for(;;)
    {
        menu();
        switch(get_cmd('1','4'))
        {
            case '1':register_user();break;
            case '2':longin_user();break;
            case '3':show_user();break;
            case '4':return 0;
        }
    }
    
}