/* $Id: poststat.c,v 1.2 2002/04/05 14:36:25 in2 Exp $ */
/* 統計今日、週、月、年熱門話題 */

#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include <fcntl.h>
#include <time.h>
#include <sys/types.h>
#include <sys/ipc.h>
#include <sys/shm.h>

#ifdef __FreeBSD__
#include <machine/param.h>
#endif

#include "config.h"
#include "pttstruct.h"

char *myfile[] =
{"day", "week", "month", "year"};
int mycount[4] =
{7, 4, 12};
int mytop[] =
{10, 50, 100, 100};
char *mytitle[] =
{"日十", "週五十", "月百", "年度百"};


#define HASHSIZE 1024
#define TOPCOUNT 200


struct postrec
{
    char author[13];		/* author name */
    char board[13];		/* board name */
    char title[66];		/* title name */
    time_t date;		/* last post's date */
    int number;			/* post number */
    struct postrec *next;	/* next rec */
}
*bucket[HASHSIZE];


/* 100 bytes */
struct posttop
{
    char author[13];		/* author name */
    char board[13];		/* board name */
    char title[66];		/* title name */
    time_t date;		/* last post's date */
    int number;			/* post number */
}
top[TOPCOUNT], *tp;



/*
   woju
   Cross-fs rename()
 */

int Rename(char *src, char *dst)
{

    if (rename(src, dst) == 0)
	return 0;
/*
  sprintf(cmd, "/bin/mv %s %s", src, dst);
  return system(cmd);
*/
    return 0;
}



/*-------------------------------------------------------*/
/* .BOARDS cache                                           */
/*-------------------------------------------------------*/

struct bcache_t *brdshm;
boardheader_t *bcache;
int numboards = -1;


static void
attach_err(shmkey, name)
    int shmkey;
    char *name;
{
    fprintf(stderr, "[%s error] key = %x\n", name, shmkey);
    exit(1);
}


static void *
attach_shm(shmkey, shmsize)
    int shmkey, shmsize;
{
    void *shmptr;
    int shmid;

    shmid = shmget(shmkey, shmsize, 0);
    if (shmid < 0)
    {
	shmid = shmget(shmkey, shmsize, IPC_CREAT | 0600);
	if (shmid < 0)
	    attach_err(shmkey, "shmget");
	shmptr = (void *) shmat(shmid, NULL, 0);
	if (shmptr == (void *) -1)
	    attach_err(shmkey, "shmat");
	memset(shmptr, 0, shmsize);
    }
    else
    {
	shmptr = (void *) shmat(shmid, NULL, 0);
	if (shmptr == (void *) -1)
	    attach_err(shmkey, "shmat");
    }
    return shmptr;
}



void
resolve_boards()
{
    if (brdshm == NULL)
    {
	brdshm = attach_shm(BRDSHM_KEY, sizeof(*brdshm));
	if (brdshm->touchtime == 0)
	    brdshm->touchtime = 1;
	bcache = brdshm->bcache;
    }

    while (brdshm->uptime < brdshm->touchtime)
    {
	if (brdshm->busystate)
	{
	    sleep(1);
	}
	else
	{
	    int fd;

	    brdshm->busystate = 1;

	    if ((fd = open(".BRD", O_RDONLY)) > 0)
	    {
		brdshm->number = read(fd, bcache, MAX_BOARD * sizeof(boardheader_t))
		    / sizeof(boardheader_t);
		close(fd);
	    }

	    /* 等所有 boards 資料更新後再設定 uptime */

	    brdshm->uptime = brdshm->touchtime;
	    brdshm->busystate = 0;
	}
    }
    numboards = brdshm->number;
}


int
ci_strcmp(s1, s2)
    register char *s1, *s2;
{
    register int c1, c2, diff;

    do
    {
	c1 = *s1++;
	c2 = *s2++;
	if (c1 >= 'A' && c1 <= 'Z')
	    c1 |= 32;
	if (c2 >= 'A' && c2 <= 'Z')
	    c2 |= 32;
	if((diff = c1 - c2))
	    return (diff);
    }
    while (c1);
    return 0;
}

int
get_record(fpath, rptr, size, id)
    char *fpath;
    char *rptr;
    int size, id;
{
    int fd;

    if ((fd = open(fpath, O_RDONLY, 0)) != -1)
    {
	if (lseek(fd, size * (id - 1), SEEK_SET) != -1)
	{
	    if (read(fd, rptr, size) == size)
	    {
		close(fd);
		return 0;
	    }
	}
	close(fd);
    }
    return -1;
}


int
getbnum(bname)
    char *bname;
{
    register int i;
    register boardheader_t *bhdr;

    resolve_boards();
    for (i = 0, bhdr = bcache; i++ < numboards; bhdr++)
	/* if (Ben_Perm(bhdr)) */
	if (!ci_strcmp(bname, bhdr->brdname))
	    return i;
    return 0;
}


int
hash(key)
    char *key;
{
    int i, value = 0;

    for (i = 0; key[i] && i < 80; i++)
	value += key[i] < 0 ? -key[i] : key[i];

    value = value % HASHSIZE;
    return value;
}


/* ---------------------------------- */
/* hash structure : array + link list */
/* ---------------------------------- */


void
search(t)
    struct posttop *t;
{
    struct postrec *p, *q, *s;
    int i, found = 0;

    i = hash(t->title);
    q = NULL;
    p = bucket[i];
    while (p && (!found))
    {
	if (!strcmp(p->title, t->title) && !strcmp(p->board, t->board))
	    found = 1;
	else
	{
	    q = p;
	    p = p->next;
	}
    }
    if (found)
    {
	p->number += t->number;
	if (p->date < t->date)	/* 取較近日期 */
	    p->date = t->date;
    }
    else
    {
	s = (struct postrec *) malloc(sizeof(struct postrec));
	memcpy(s, t, sizeof(struct posttop));
	s->next = NULL;
	if (q == NULL)
	    bucket[i] = s;
	else
	    q->next = s;
    }
}


int
sort(pp, count)
    struct postrec *pp;
{
    int i, j;

    for (i = 0; i <= count; i++)
    {
	if (pp->number > top[i].number)
	{
	    if (count < TOPCOUNT - 1)
		count++;
	    for (j = count - 1; j >= i; j--)
		memcpy(&top[j + 1], &top[j], sizeof(struct posttop));

	    memcpy(&top[i], pp, sizeof(struct posttop));
	    break;
	}
    }
    return count;
}


void
load_stat(fname)
    char *fname;
{
    FILE *fp;

    if((fp = fopen(fname, "r")))
    {
	int count = fread(top, sizeof(struct posttop), TOPCOUNT, fp);
	fclose(fp);
	while (count)
	    search(&top[--count]);
    }
}


int
filter(board)
    char *board;
{
    boardheader_t bh;
    int bid;

    bid = getbnum(board);
    if (get_record(".BRD", &bh, sizeof(bh), bid) == -1)
	return 1;
    if (bh.brdattr & BRD_NOCOUNT)
	return 1;
/*
  if (bh.brdattr & BRD_POSTMASK)
  return 0;
  return (bh.brdattr & ~BRD_POSTMASK) ||
  >= 32;
*/
    return 0;
}


void
poststat(mytype)
    int mytype;
{
    static char *logfile = ".post";
    static char *oldfile = ".post.old";

    FILE *fp;
    char buf[40], curfile[40] = "etc/day.0", *p;
    struct postrec *pp;
    int i, j;

    if (mytype < 0)
    {
	/* --------------------------------------- */
	/* load .post and statictic processing     */
	/* --------------------------------------- */

	remove(oldfile);
	Rename(logfile, oldfile);
	if ((fp = fopen(oldfile, "r")) == NULL)
	    return;
	mytype = 0;
	load_stat(curfile);

	while (fread(top, sizeof(struct posttop), 1, fp))
	    search(top);
	fclose(fp);
    }
    else
    {
	/* ---------------------------------------------- */
	/* load previous results and statictic processing */
	/* ---------------------------------------------- */

	i = mycount[mytype];
	p = myfile[mytype];
	while (i)
	{
	    sprintf(buf, "etc/%s.%d", p, i);
	    sprintf(curfile, "etc/%s.%d", p, --i);
	    load_stat(curfile);
	    Rename(curfile, buf);
	}
	mytype++;
    }

/* ---------------------------------------------- */
/* sort top 100 issue and save results            */
/* ---------------------------------------------- */

    memset(top, 0, sizeof(top));
    for (i = j = 0; i < HASHSIZE; i++)
    {
	for (pp = bucket[i]; pp; pp = pp->next)
	{
#ifdef  DEBUG
	    printf("Title : %s, Board: %s\nPostNo : %d, Author: %s\n"
		   ,pp->title
		   ,pp->board
		   ,pp->number
		   ,pp->author);
#endif

	    j = sort(pp, j);
	}
    }

    p = myfile[mytype];
    sprintf(curfile, "etc/%s.0", p);
    if((fp = fopen(curfile, "w")))
    {
	fwrite(top, sizeof(struct posttop), j, fp);
	fclose(fp);
    }

    sprintf(curfile, "etc/%s", p);
    if((fp = fopen(curfile, "w")))
    {
	int max, cnt;

	fprintf(fp, "\t\t-----===== 本%s大熱門話題 =====-----\n\n", mytitle[mytype]);

	max = mytop[mytype];
	p = buf + 4;
	for (i = cnt = 0; (cnt < max) && (i < j); i++)
	{
	    tp = &top[i];
	    if (filter(tp->board))
		continue;

	    strcpy(buf, ctime(&(tp->date)));
	    buf[20] = 0;
	    fprintf(fp,
		    "%3d. 看板 : %-16s《 %s》%4d 篇%16s\n"
		    "     標題 : %-60.60s\n"
		    ,++cnt, tp->board, p, tp->number, tp->author, tp->title);
	}
	fclose(fp);
    }

/* free statistics */

    for (i = 0; i < HASHSIZE; i++)
    {
	struct postrec *pp0;

	pp = bucket[i];
	while (pp)
	{
	    pp0 = pp;
	    pp = pp->next;
	    free(pp0);
	}

	bucket[i] = NULL;
    }
}


int main(argc, argv)
    char *argv[];
{
    time_t now;
    struct tm *ptime;

    if (argc < 2)
    {
	printf("Usage:\t%s bbshome [day]\n", argv[0]);
	return (-1);
    }
    chdir(argv[1]);

    if (argc == 3)
    {
	poststat(atoi(argv[2]));
	return (0);
    }
    time(&now);
    ptime = localtime(&now);
    if (ptime->tm_hour == 0)
    {
	if (ptime->tm_mday == 1)
	    poststat(2);

	if (ptime->tm_wday == 0)
	    poststat(1);
	poststat(0);
    }
    poststat(-1);
    return 0;
}