快译通 _0317

//dictionary.c

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <assert.h>

#define WORD_LEN    128
#define EX_LEN        1024

typedef struct word
{
    char word[50];
    long loc;
    struct word *next;
}node_t;

int menu(void)
{
    int choice;

    printf("1.search for a word\n2.add a word\n9.quit\nPlease make a choice:");
    scanf("%d", &choice);
    getchar();

    return choice;
}

node_t *make_node(char *w, long loc)
{
    node_t *cur;
    cur = malloc(sizeof(node_t));
    assert(cur);

    strcpy(cur->word, w);
    cur->loc = loc;
    cur->next = NULL;

    return cur;
}

node_t *insert_to_link(node_t *head, char *w, long loc)
{
    node_t *cur = make_node(w, loc);

    if(head == NULL)
        head = cur;
    else
        cur->next = head;

    return cur;
}

node_t *create_link(void)
{
    FILE *fp;
    char ch, w[WORD_LEN];
    node_t *head = NULL;
    long pos = 0;

    fp = fopen("dict.txt", "r+");
    assert(fp);

    while((ch = getc(fp)) != EOF)
    {
        if(ch == '#')
        {
            pos = ftell(fp);

            fgets(w, WORD_LEN-1, fp);
            head = insert_to_link(head, w, pos);
        }
    }
    fclose(fp);
    return head;
}

void print_exp(char *exp)
{
    char *token, *savep;
    int i;

    exp += 6;
    for(i = 1; ; exp = NULL, i ++)
    {
        token = strtok_r(exp, "@", &savep);

        if(! token)
            break;
        printf("%d : %s\n", i, token);
    }
}

void get_explanation(long loc)
{
    FILE *fp;
    char buffer[1024];

    fp = fopen("dict.txt", "r");
    assert(fp);

    fseek(fp, loc, SEEK_SET);
    fgets(buffer, WORD_LEN, fp);
    fgets(buffer, EX_LEN, fp);

    print_exp(buffer);

    fclose(fp);
}

int search3(node_t *head, char *w)
{
    node_t *cur;

    for(cur = head; cur != NULL; cur = cur->next)
    {
        if(strcmp(cur->word, w) == 0)
        {
            get_explanation(cur->loc);
            return 0;
        }
    }

    return -1;
}

void print_similar(node_t *head, char *w)
{
    int i = 1;
    char buffer[20];
    node_t *cur;

    for(cur = head; cur != NULL; cur = cur->next)
    {
        if(strncmp(cur->word, w, strlen(w) - 1) == 0)
        {
            printf("%s", cur->word);
            if(((i++)%20) == 0)
            {
                printf("press enter to continue...(press 'q' to quit):");
                fgets(buffer, 19, stdin);

                if(buffer[0] == 'q')
                    break;
            }
        }
    }
}

void search_for_word(node_t *head)
{
    char word[WORD_LEN];

    while(1)
    {
        printf("Please input the word:(input 1 to quit): ");
        fgets(word, WORD_LEN-1, stdin);
        if(word[0] == '1')
            break;

        if(word[strlen(word) - 2] == '*')
        {
            word[strlen(word) - 2] = '\n';
            word[strlen(word) - 1] = '\0';
            print_similar(head, word);
        }
        else
        {
            if(search3(head, word) < 0)
                print_similar(head, word);
        }
    }
}

int get_info(char *w, char *exp)
{
    char exp_buf[1024];
    w[0] = '#';
    strcpy(exp, "Trans:");

    printf("please input word:");
    fgets(w+1, WORD_LEN, stdin);
    int i = 1;
    while(1)
    {
        printf("input explanation %d(input 1 to quit): ", i++);
        fgets(exp_buf,EX_LEN, stdin);

        if(*exp_buf == '1')
            break;

        if(i > 2)
        {
            exp[strlen(exp)-1] = '@';
        }
        strcat(exp, exp_buf);
    }
    return 1;
}

void move_to_file(long location, char *w, char *exp)
{
    FILE *fp, *fp_tmp;
    char ch;
    long loc;

    //对fp文件操作
    fp= fopen("dict.txt", "r+");
    assert(fp);

    fseek(fp, location, SEEK_SET);//定位到应该插入的前面一个单词处
    while((ch = getc(fp)) != EOF)
        if(ch == '#')
            break;
    loc = ftell(fp) - 1;               //找到应该插入的位置

    //对fp_tmp文件操作
    fp_tmp= fopen("tmp.txt", "w+");
    assert(fp_tmp);

    fwrite(w, 1, strlen(w), fp_tmp);    //将新添加单词和译文 写入临时文件fp_tmp中
    fwrite(exp, 1, strlen(exp), fp_tmp);
    fseek(fp, loc, SEEK_SET);
    while((ch = getc(fp)) != EOF)       //将文件fp中的内容，读入到临时文件fp_tmp中
        putc(ch, fp_tmp);

    //把临时文件中的内容，写回到fp文件中
    rewind(fp_tmp);
    fseek(fp, loc, SEEK_SET);
    while((ch = getc(fp_tmp)) != EOF)
        putc(ch, fp);

    fclose(fp);
    fclose(fp_tmp);
}

node_t *update_link(node_t *head)
{
    node_t *cur, *nextp;

    for(cur = head; cur != NULL; cur = nextp)
    {
        nextp = cur->next;
        free(cur);
    }
    cur = create_link();
    return cur;
}

node_t *insert_to_file(node_t *head, char *w, char *exp)
{
    node_t *cur;

    for(cur = head; cur != NULL; cur = cur->next)
    {
       if(strcmp(cur->word, w + 1) < 0)
       {
            move_to_file(cur->loc, w, exp);
            head = update_link(head);
            break;
       }
    }
    return head;
}


node_t *add_word(node_t *head)
{
    char w[WORD_LEN];
    char exp[EX_LEN];

    if(get_info(w, exp) > 0)
        head = insert_to_file(head, w, exp);

    return head;
}


int main(void)
{
    int choice;
    node_t *head = NULL;

    head = create_link();
    while(1)
    {
        choice = menu();
        if(choice == 1)
        {
            search_for_word(head);
        }
        else if(choice == 2)
        {
            head = add_word(head);
        }
        else if(choice == 9)
        {
            break;
        }
    }
    return 0;
}