Merge branch 'switch-to-deriving_yojson' into 'master'

Switch to deriving yojson

See merge request Mishio595/disml!12

1 files changed, 118 insertions, 0 deletions

diff --git a/lib/models/channel_t.ml b/lib/models/channel_t.ml
new file mode 100644
index 0000000..dce1d54
--- /dev/null
+++ b/lib/models/channel_t.ml
@@ -0,0 +1,118 @@
+open Core
+
+exception Invalid_channel of Yojson.Safe.json
+
+type group = {
+    id: Snowflake.t;
+    last_message_id: Snowflake.t option [@default None];
+    last_pin_timestamp: string option [@default None];
+    icon: string option [@default None];
+    name: string option [@default None];
+    owner_id: Snowflake.t;
+    recipients: User_t.t list [@default []];
+} [@@deriving sexp, yojson { strict = false}]
+
+type dm = {
+    id: Snowflake.t;
+    last_message_id: Snowflake.t option [@default None];
+    last_pin_timestamp: string option [@default None];
+} [@@deriving sexp, yojson { strict = false}]
+
+type guild_text = {
+    id: Snowflake.t;
+    last_message_id: Snowflake.t option [@default None];
+    last_pin_timestamp: string option [@default None];
+    category_id: Snowflake.t option [@default None][@key "parent_id"];
+    guild_id: Snowflake.t option [@default None];
+    name: string;
+    position: int;
+    topic: string option [@default None];
+    nsfw: bool;
+    slow_mode_timeout: int option [@default None];
+} [@@deriving sexp, yojson { strict = false}]
+
+type guild_voice = {
+    id: Snowflake.t;
+    category_id: Snowflake.t option [@default None][@key "parent_id"];
+    guild_id: Snowflake.t option [@default None];
+    name: string;
+    position: int;
+    user_limit: int [@default -1];
+    bitrate: int option [@default None];
+} [@@deriving sexp, yojson { strict = false}]
+
+type category = {
+    id: Snowflake.t;
+    guild_id: Snowflake.t option [@default None];
+    position: int;
+    name: string;
+} [@@deriving sexp, yojson { strict = false}]
+
+type t =
+| Group of group
+| Private of dm
+| GuildText of guild_text
+| GuildVoice of guild_voice
+| Category of category
+[@@deriving sexp, yojson { strict = false}]
+
+type channel_wrapper = {
+    id: Snowflake.t;
+    kind: int [@key "type"];
+    guild_id: Snowflake.t option [@default None];
+    position: int option [@default None];
+    name: string option [@default None];
+    topic: string option [@default None];
+    nsfw: bool option [@default None];
+    last_message_id: Snowflake.t option [@default None];
+    bitrate: int option [@default None];
+    user_limit: int option [@default None];
+    slow_mode_timeout: int option [@default None];
+    recipients: User_t.t list option [@default None];
+    icon: string option [@default None];
+    owner_id: Snowflake.t option [@default None];
+    application_id: Snowflake.t option [@default None];
+    category_id: Snowflake.t option [@default None][@key "parent_id"];
+    last_pin_timestamp: string option [@default None];
+} [@@deriving sexp, yojson { strict = false}]
+
+let unwrap_as_guild_text {id;guild_id;position;name;topic;nsfw;last_message_id;slow_mode_timeout;category_id;last_pin_timestamp;_} =
+    let position = Option.value_exn position in
+    let name = Option.value_exn name in
+    let nsfw = Option.value ~default:false nsfw in
+    { id; guild_id; position; name; topic; nsfw; last_message_id; slow_mode_timeout; category_id; last_pin_timestamp }
+
+let unwrap_as_guild_voice {id;guild_id;position;name;bitrate;user_limit;category_id;_} =
+    let position = Option.value_exn position in
+    let name = Option.value_exn name in
+    let user_limit = Option.value ~default:(-1) user_limit in
+    { id; guild_id; position; name; user_limit; bitrate ; category_id; }
+
+let unwrap_as_dm {id;last_message_id;last_pin_timestamp;_} =
+    { id; last_message_id; last_pin_timestamp; }
+
+let unwrap_as_group {id;name;last_message_id;recipients;icon;owner_id;last_pin_timestamp;_} =
+    let recipients = Option.value ~default:[] recipients in
+    let owner_id = Option.value_exn owner_id in
+    { id; name; last_message_id; recipients; icon; owner_id; last_pin_timestamp; }
+
+let unwrap_as_category {id;guild_id;position;name;_} =
+    let position = Option.value_exn position in
+    let name = Option.value_exn name in
+    { id; guild_id; position; name; }
+
+let wrap s =
+    match s.kind with
+    | 0 -> GuildText (unwrap_as_guild_text s)
+    | 1 -> Private (unwrap_as_dm s)
+    | 2 -> GuildVoice (unwrap_as_guild_voice s)
+    | 3 -> Group (unwrap_as_group s)
+    | 4 -> Category (unwrap_as_category s)
+    | _ -> raise (Invalid_channel (channel_wrapper_to_yojson s))
+
+let get_id = function
+| Group g -> g.id
+| Private p -> p.id
+| GuildText t -> t.id
+| GuildVoice v -> v.id
+| Category c -> c.id
\ No newline at end of file