#pragma once

#include <stdint.h>

#include "ztypes.h"

#define SYS0(name)                         \
  struct Z##name##Req {};                  \
  [[nodiscard]] inline z_err_t Z##name() { \
    Z##name##Req req{};                    \
    return SysCall1(kZion##name, &req);    \
  }

#define SYS1(name, t1, a1)                      \
  struct Z##name##Req {                         \
    t1 a1;                                      \
  };                                            \
  [[nodiscard]] inline z_err_t Z##name(t1 a1) { \
    Z##name##Req req{                           \
        .a1 = a1,                               \
    };                                          \
    return SysCall1(kZion##name, &req);         \
  }

#define SYS2(name, t1, a1, t2, a2)                     \
  struct Z##name##Req {                                \
    t1 a1;                                             \
    t2 a2;                                             \
  };                                                   \
  [[nodiscard]] inline z_err_t Z##name(t1 a1, t2 a2) { \
    Z##name##Req req{                                  \
        .a1 = a1,                                      \
        .a2 = a2,                                      \
    };                                                 \
    return SysCall1(kZion##name, &req);                \
  }

#define SYS3(name, t1, a1, t2, a2, t3, a3)                    \
  struct Z##name##Req {                                       \
    t1 a1;                                                    \
    t2 a2;                                                    \
    t3 a3;                                                    \
  };                                                          \
  [[nodiscard]] inline z_err_t Z##name(t1 a1, t2 a2, t3 a3) { \
    Z##name##Req req{                                         \
        .a1 = a1,                                             \
        .a2 = a2,                                             \
        .a3 = a3,                                             \
    };                                                        \
    return SysCall1(kZion##name, &req);                       \
  }

#define SYS4(name, t1, a1, t2, a2, t3, a3, t4, a4)                   \
  struct Z##name##Req {                                              \
    t1 a1;                                                           \
    t2 a2;                                                           \
    t3 a3;                                                           \
    t4 a4;                                                           \
  };                                                                 \
  [[nodiscard]] inline z_err_t Z##name(t1 a1, t2 a2, t3 a3, t4 a4) { \
    Z##name##Req req{                                                \
        .a1 = a1,                                                    \
        .a2 = a2,                                                    \
        .a3 = a3,                                                    \
        .a4 = a4,                                                    \
    };                                                               \
    return SysCall1(kZion##name, &req);                              \
  }

#define SYS5(name, t1, a1, t2, a2, t3, a3, t4, a4, t5, a5)                  \
  struct Z##name##Req {                                                     \
    t1 a1;                                                                  \
    t2 a2;                                                                  \
    t3 a3;                                                                  \
    t4 a4;                                                                  \
    t5 a5;                                                                  \
  };                                                                        \
  [[nodiscard]] inline z_err_t Z##name(t1 a1, t2 a2, t3 a3, t4 a4, t5 a5) { \
    Z##name##Req req{                                                       \
        .a1 = a1,                                                           \
        .a2 = a2,                                                           \
        .a3 = a3,                                                           \
        .a4 = a4,                                                           \
        .a5 = a5,                                                           \
    };                                                                      \
    return SysCall1(kZion##name, &req);                                     \
  }

z_err_t SysCall1(uint64_t code, const void* req);

SYS1(ProcessExit, uint64_t, code);
SYS5(ProcessSpawn, z_cap_t, proc_cap, z_cap_t, bootstrap_cap, z_cap_t*,
     new_proc_cap, z_cap_t*, new_vmas_cap, z_cap_t*, new_bootstrap_cap);

SYS2(ThreadCreate, z_cap_t, proc_cap, z_cap_t*, thread_cap);
SYS4(ThreadStart, z_cap_t, thread_cap, uint64_t, entry, uint64_t, arg1,
     uint64_t, arg2);
SYS0(ThreadExit);

SYS4(AddressSpaceMap, z_cap_t, vmas_cap, uint64_t, vmas_offset, z_cap_t,
     vmmo_cap, uint64_t*, vaddr);

SYS2(MemoryObjectCreate, uint64_t, size, z_cap_t*, vmmo_cap);
SYS3(MemoryObjectCreatePhysical, uint64_t, paddr, uint64_t, size, z_cap_t*,
     vmmo_cap);
SYS3(MemoryObjectCreateContiguous, uint64_t, size, z_cap_t*, vmmo_cap,
     uint64_t*, paddr);
SYS2(TempPcieConfigObjectCreate, z_cap_t*, vmmo_cap, uint64_t*, vmmo_size);

SYS2(ChannelCreate, z_cap_t*, channel1, z_cap_t*, channel2);
SYS5(ChannelSend, z_cap_t, chan_cap, uint64_t, num_bytes, const void*, data,
     uint64_t, num_caps, z_cap_t*, caps);
SYS5(ChannelRecv, z_cap_t, chan_cap, uint64_t*, num_bytes, void*, data,
     uint64_t*, num_caps, z_cap_t*, caps);

SYS1(PortCreate, z_cap_t*, port_cap);
SYS5(PortSend, z_cap_t, port_cap, uint64_t, num_bytes, const void*, data,
     uint64_t, num_caps, z_cap_t*, caps);
SYS5(PortRecv, z_cap_t, port_cap, uint64_t*, num_bytes, void*, data, uint64_t*,
     num_caps, z_cap_t*, caps);
SYS5(PortPoll, z_cap_t, port_cap, uint64_t*, num_bytes, void*, data, uint64_t*,
     num_caps, z_cap_t*, caps);

SYS2(IrqRegister, uint64_t, irq_num, z_cap_t*, port_cap);

SYS2(CapDuplicate, z_cap_t, cap_in, z_cap_t*, cap_out);

SYS1(Debug, const char*, message);