/* $Id$ */
#include "bbs.h"
#define LOGPASS BBSHOME "/etc/winguess.log"

static void
show_table(char TABLE[], char ifcomputer)
{
    int             i;

    move(0, 35);
    prints(gettext[1139]);
    move(8, 1);
    prints(gettext[1140]);
    prints("\033[1;33m=================\033[m\n");
    if (ifcomputer) {
	prints(gettext[1141]);
	prints(gettext[1142]);
    } else {
	for (i = 1; i <= 6; i++)
	    prints(gettext[1143], i, TABLE[i]);
    }
    prints("\033[33m=================\033[m");
}

static long int
get_money(void)
{
    int             money, i;
    char            data[20];

    move(1, 0);
    prints(gettext[1144], cuser.money);
    do {
	getdata(2, 0, gettext[1145], data, 9, LCECHO);
	money = 0;
	if (data[0] == 'q' || data[0] == 'Q') {
	    unlockutmpmode();
	    return 0;
	}
	for (i = 0; data[i]; i++)
	    if (data[i] < '0' || data[i] > '9') {
		money = -1;
		break;
	    }
	if (money != -1) {
	    money = atoi(data);
	    reload_money();
	    if (money > cuser.money || money <= 4 || money > 10 ||
		money < 1)
		money = -1;
	}
    } while (money == -1);
    move(1, 0);
    clrtoeol();
    reload_money();
    prints(gettext[1146], cuser.money - money);
    return money;
}

static int
check_data(char *str)
{
    int             i, j;

    if (strlen(str) != 4)
	return -1;
    for (i = 0; i < 4; i++)
	if (str[i] < '0' || str[i] > '9')
	    return -1;
    for (i = 0; i < 4; i++)
	for (j = i + 1; j < 4; j++)
	    if (str[i] == str[j])
		return -1;
    return 1;
}

static char    *
get_data(char data[5], int count)
{
    while (1) {
	getdata(6, 0, gettext[1147], data, 5, LCECHO);
	if (check_data(data) == 1)
	    break;
    }
    return data;
}

static int
guess_play(char *data, char *answer, int count)
{
    int             A_num = 0, B_num = 0;
    int             i, j;

    for (i = 0; i < 4; i++) {
	if (data[i] == answer[i])
	    A_num++;
	for (j = 0; j < 4; j++)
	    if (i == j)
		continue;
	    else if (data[i] == answer[j]) {
		B_num++;
		break;
	    }
    }
    if (A_num == 4)
	return 1;
    move(count + 8, 55);
    prints("%s => \033[1;32m%dA %dB\033[m", data, A_num, B_num);
    return 0;
}

static int
result(int correct, int number)
{
    char            a = 0, b = 0, i, j;
    char            n1[5], n2[5];

    snprintf(n1, sizeof(n1), "%04d", correct);
    snprintf(n2, sizeof(n2), "%04d", number);
    for (i = 0; i < 4; i++)
	for (j = 0; j < 4; j++)
	    if (n1[(int)i] == n2[(int)j])
		b++;
    for (i = 0; i < 4; i++)
	if (n1[(int)i] == n2[(int)i]) {
	    b--;
	    a++;
	}
    return 10 * a + b;
}

static int
legal(int number)
{
    char            i, j;
    char            temp[5];

    snprintf(temp, sizeof(temp), "%04d", number);
    for (i = 0; i < 4; i++)
	for (j = i + 1; j < 4; j++)
	    if (temp[(int)i] == temp[(int)j])
		return 0;
    return 1;
}

static void
initcomputer(char flag[])
{
    int             i;

    for (i = 0; i < 10000; i++)
	if (legal(i))
	    flag[i] = 1;
	else
	    flag[i] = 0;
}

static int
computer(int correct, int total, char flag[], int n[])
{
    int             guess;
    static int      j;
    int             k, i;
    char            data[5];

    if (total == 1) {
	do {
	    guess = rand() % 10000;
	} while (!legal(guess));
    } else
	guess = n[rand() % j];
    k = result(correct, guess);
    if (k == 40) {
	move(total + 8, 25);
	snprintf(data, sizeof(data), "%04d", guess);
	prints(gettext[1148], data);
	return 1;
    } else {
	move(total + 8, 25);
	snprintf(data, sizeof(data), "%04d", guess);
	prints("%s => \033[1;32m%dA %dB\033[m", data, k / 10, k % 10);
    }
    j = 0;
    for (i = 0; i < 10000; i++)
	if (flag[i]) {
	    if (result(i, guess) != k)
		flag[i] = 0;
	    else
		n[j++] = i;
	}
    return 0;
}

static void
Diff_Random(char *answer)
{
    register int    i = 0, j, k;

    while (i < 4) {
	k = rand() % 10 + '0';
	for (j = 0; j < i; j++)
	    if (k == answer[j])
		break;
	if (j == i) {
	    answer[j] = k;
	    i++;
	}
    }
    answer[4] = 0;
}

#define lockreturn0(unmode, state) if(lockutmpmode(unmode, state)) return 0

int
guess_main()
{
    char            data[5];
    long int money;
    char            computerwin = 0, youwin = 0;
    int             count = 0, c_count = 0;
    char            ifcomputer[2];
    char            answer[5];
    int            *n = NULL;
    char            yournum[5];
    char           *flag = NULL;
    char            TABLE[] = {0, 10, 8, 4, 2, 1, 0, 0, 0, 0, 0};
    FILE           *file;

    clear();
    showtitle(gettext[1149], BBSName);
    lockreturn0(GUESSNUM, LOCK_MULTI);

    reload_money();
    if (cuser.money < 5) {
	clear();
	move(12, 35);
	prints(gettext[1150]);
	unlockutmpmode();
	pressanykey();
	return 1;
    }
    if ((money = get_money()) == 0)
	return 1;
    vice(money, gettext[1151]);

    Diff_Random(answer);
    move(2, 0);
    clrtoeol();
    prints(gettext[1152], money);

    getdata_str(4, 0, gettext[1153],
		ifcomputer, sizeof(ifcomputer), LCECHO, "y");
    if (ifcomputer[0] == 'y') {
	ifcomputer[0] = 1;
	show_table(TABLE, 1);
    } else {
	ifcomputer[0] = 0;
	show_table(TABLE, 0);
    }
    if (ifcomputer[0]) {
	do {
	    getdata(5, 0, gettext[1154],
		    yournum, sizeof(yournum), LCECHO);
	} while (!legal(atoi(yournum)));
	move(8, 25);
	prints(gettext[1155]);
	flag = malloc(sizeof(char) * 10000);
	n = malloc(sizeof(int) * 1500);
	initcomputer(flag);
    }
    move(8, 55);
    prints(gettext[1156]);
    while (((!computerwin || !youwin) && count < 10 && (ifcomputer[0])) ||
	   (!ifcomputer[0] && count < 10 && !youwin)) {
	if (!computerwin && ifcomputer[0]) {
	    ++c_count;
	    if (computer(atoi(yournum), c_count, flag, n))
		computerwin = 1;
	}
	move(20, 55);
	prints(gettext[1157], count + 1);
	if (!youwin) {
	    ++count;
	    if (guess_play(get_data(data, count), answer, count))
		youwin = 1;
	}
    }
    move(17, 35);
    free(flag);
    free(n);
    if (ifcomputer[0]) {
	if (count > c_count) {
	    prints(gettext[1158]);
	    move(18, 35);
	    prints(gettext[1159], money);
	    if ((file = fopen(LOGPASS, "a"))) {
		fprintf(file, gettext[1160], c_count);
		if (youwin)
		    fprintf(file, gettext[1161], cuser.userid, count);
		else
		    fprintf(file, gettext[1162], cuser.userid);
		fprintf(file, gettext[1163], cuser.userid, money);
		fclose(file);
	    }
	} else if (count < c_count) {
	    prints(gettext[1164]);
	    move(18, 35);
	    prints(gettext[1165], money * 2);
	    demoney(money * 2);
	    if ((file = fopen(LOGPASS, "a"))) {
		fprintf(file, gettext[1166], cuser.userid, count,
			c_count, money * 2);
		fclose(file);
	    }
	} else {
	    prints(gettext[1167], money);
	    demoney(money);
	    if ((file = fopen(LOGPASS, "a"))) {
		fprintf(file, gettext[1168], cuser.userid);
		fclose(file);
	    }
	}
	unlockutmpmode();
	pressanykey();
	return 1;
    }
    if (youwin) {
	demoney(TABLE[count] * money);
	if (count < 5) {
	    prints(gettext[1169]);
	    if ((file = fopen(LOGPASS, "a"))) {
		fprintf(file, gettext[1170],
			cuser.userid, count, TABLE[count] * money);
		fclose(file);
	    }
	} else if (count > 5) {
	    prints(gettext[1171]);
	    if ((file = fopen(LOGPASS, "a"))) {
		fprintf(file, gettext[1172],
			cuser.userid, count, money);
		fclose(file);
	    }
	} else {
	    prints(gettext[1173]);
	    move(18, 35);
	    clrtoeol();
	    prints(gettext[1174], money);
	    if ((file = fopen(LOGPASS, "a"))) {
		fprintf(file, gettext[1175],
			cuser.userid, count, money);
		fclose(file);
	    }
	}
	unlockutmpmode();
	pressanykey();
	return 1;
    }
    move(17, 35);
    prints(gettext[1176], answer);
    move(18, 35);
    prints(gettext[1177]);
    if ((file = fopen(BBSHOME "/etc/loseguess.log", "a"))) {
	fprintf(file, gettext[1178], cuser.userid, money);
	fclose(file);
    }
    unlockutmpmode();
    pressanykey();
    return 1;
}