#pragma once

#include "avltree.h"
#include "memmap.h"
#include "priorityqueue.h"
#include "queue.h"
#include "mmgr.h"
#include "syscalls.h"
#include "types/syscallid.h"
#include <stddef.h>

#define MAX_SYSCALL_ID 256
#define module_limit 8

#define IO_OP 1 << 0
#define IO_SYNC 0 << 0
#define IO_ASYNC 1 << 0

#define IO_RECIPIENT_TYPE 3 << 1
#define IO_PID 0 << 1
#define IO_PORT 1 << 1
#define IO_MAILBOX 2 << 1

struct process_context_t;

struct module_t
{
    physaddr_t start;
    physaddr_t end;
    char str[64 - 2 * sizeof(physaddr_t)];
};

struct boot_info_t
{
    char *bootloader;
    char *parameters;
    size_t module_count;
    struct memory_map_t map;
    struct module_t modules[module_limit];
};

struct message_t
{
    unsigned long sender;
    unsigned long code;
    unsigned long args[6];
};

enum process_state_t
{
    PROCESS_ACTIVE,
    PROCESS_REQUESTING,
    PROCESS_SENDING
};

struct process_t
{
    unsigned long pid;
    int priority;
    physaddr_t page_table;
    enum process_state_t state;
    struct queue_t sending_queue;
    struct queue_t message_queue;
    struct message_t *message_buffer;
    struct process_context_t *ctx;
};

struct port_t
{
    unsigned long id;
    unsigned long owner_pid;
};

struct kernel_t
{
    struct syscall_t syscall_table[MAX_SYSCALL_ID];
    struct priority_queue_t priority_queue;
    struct avltree_t *port_table;
    struct avltree_t *process_table;
    struct process_t *active_process;
    int next_pid;
};

void kernel_initialize(struct boot_info_t *boot_info);

int set_syscall(int id, int arg_count, int pid, void *func_ptr);

size_t do_syscall(enum syscall_id_t id, syscall_arg_t arg1, syscall_arg_t arg2, syscall_arg_t arg3, void *pc, void *stack, unsigned long flags);

int load_module(struct module_t *module);

int active_process();

int add_process(void *program_entry, int priority, physaddr_t address_space);

struct process_context_t *next_process();

int terminate_process(size_t process_id);

int store_active_context(struct process_context_t *context, size_t size);

struct process_context_t *get_active_context();

int open_port(unsigned long id);

int close_port(unsigned long id);

int send_message(int recipient, struct message_t *message, int flags);

int receive_message(struct message_t *buffer, int flags);

void panic(const char *message) __attribute__ ((noreturn));