#include <Message.h>
#include <Entry.h>
#include <File.h>
#include <NodeMonitor.h>
#include <kernel/fs_info.h>
#include <kernel/fs_query.h>

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

#ifndef B_BAD_DATA
#       define B_BAD_DATA B_ERROR
#endif

#define DUMPED_BLOCK_SIZE 16
#define Print printf

void
dumpBlock(const char *buffer,int size)
{
        for(int i = 0;i < size;) {
                int start = i;

                for(;i < start+DUMPED_BLOCK_SIZE;i++) {
                        if (!(i % 4))
                                Print(" ");

                        if (i >= size)
                                Print("  ");
                        else
                                Print("%02x",*(unsigned char *)(buffer+i));
                }
                Print("  ");

                for(i = start;i < start + DUMPED_BLOCK_SIZE;i++) {
                        if (i < size) {
                                char c = *(buffer+i);

                                if (c < 30)
                                        Print(".");
                                else
                                        Print("%c",c);
                        }
                        else
                                break;
                }
                Print("\n");
        }
}


void
createFile(int32 num)
{
        char name[B_FILE_NAME_LENGTH];
        sprintf(name,"./_query_test_%ld",num);

        BFile file(name,B_CREATE_FILE | B_WRITE_ONLY);
}


BEntry *
getEntry(int32 num)
{
        char name[B_FILE_NAME_LENGTH];
        sprintf(name,"./_query_test_%ld",num);

        return new BEntry(name);
}


status_t
waitForMessage(port_id port,const char *string,int32 op,char *name)
{
        puts(string);

        int32 msg;
        char buffer[1024];
        ssize_t bytes = read_port_etc(port,&msg,buffer,sizeof(buffer),B_TIMEOUT,1000000);
        if (op == B_TIMED_OUT && bytes == B_TIMED_OUT) {
                puts("  passed, timed out!\n");
                return bytes;
        }
        if (bytes < B_OK) {
                printf("-> %s\n\n", strerror(bytes));
                return bytes;
        }

        BMessage message;
        if (message.Unflatten(buffer) < B_OK) {
                printf("could not unflatten message:\n");
                dumpBlock(buffer, bytes);
                return B_BAD_DATA;
        }

        if (message.what != B_QUERY_UPDATE
                || message.FindInt32("opcode") != op) {
                printf("Expected what = %x, opcode = %ld, got:", B_QUERY_UPDATE, op);
                message.PrintToStream();
                putchar('\n');
                return message.FindInt32("opcode");
        }
        puts("  passed!\n");
        return op;
}


int
main(int argc,char **argv)
{
        port_id port = create_port(100, "query port");
        printf("port id = %ld\n", port);
        printf("  B_ENTRY_REMOVED = %d, B_ENTRY_CREATED = %d\n\n", B_ENTRY_REMOVED, B_ENTRY_CREATED);

        dev_t device = dev_for_path(".");
        DIR *query = fs_open_live_query(device, "name=_query_test_*", B_LIVE_QUERY, port, 12345);

        createFile(1);
        waitForMessage(port, "File 1 created:", B_ENTRY_CREATED, "_query_test_1");

        createFile(2);
        waitForMessage(port, "File 2 created:", B_ENTRY_CREATED, "_query_test_2");

        BEntry *entry = getEntry(2);
        if (entry->InitCheck() < B_OK) {
                fprintf(stderr, "Could not get entry for file 2\n");
                fs_close_query(query);
                return -1;
        }
        entry->Rename("_some_other_name_");
        waitForMessage(port,"File 2 renamed (should fall out of query):", B_ENTRY_REMOVED, NULL);

        entry->Rename("_some_other_");
        waitForMessage(port,"File 2 renamed again (should time out):", B_TIMED_OUT, NULL);

        entry->Rename("_query_test_2");
        waitForMessage(port,"File 2 renamed back (should be added to query):",
                B_ENTRY_CREATED, "_query_test_2");

        entry->Rename("_query_test_2_and_more");
        status_t status = waitForMessage(port,"File 2 renamed (should stay in query, time out):",
                                                        B_TIMED_OUT, "_query_test_2_and_more");
        if (status == B_ENTRY_REMOVED) {
                waitForMessage(port,"Received B_ENTRY_REMOVED, now expecting file 2 to be added again:",
                        B_ENTRY_CREATED, NULL);
        }

        entry->Remove();
        delete entry;
        waitForMessage(port,"File 2 removed:", B_ENTRY_REMOVED, NULL);

        entry = getEntry(1);
        if (entry->InitCheck() < B_OK) {
                fprintf(stderr, "Could not get entry for file 1\n");
                fs_close_query(query);
                return -1;
        }

        entry->Rename("_some_other_name_");
        waitForMessage(port, "File 1 renamed (should fall out of query):", B_ENTRY_REMOVED, NULL);

        entry->Remove();
        delete entry;
        waitForMessage(port, "File 1 removed (should time out):", B_TIMED_OUT, NULL);

        fs_close_query(query);

        return 0;
}