add torrent get raw

1 files changed, 146 insertions, 15 deletions

diff --git a/src/request.zig b/src/request.zig
index 950a1fc..8641776 100644
--- a/src/request.zig
+++ b/src/request.zig
@@ -48,7 +48,7 @@ const Method = enum {
     // Transmission RPC uses hyphens instead of underscores, this creates a
     // table of method names using the same enum field names but with
     // underscores.
-    pub const Fields = util.RPCFields(Self);
+    pub const Fields = util.enumFieldsToStringSlice(Self);
 
     pub fn str(self: Self) []const u8 {
         return Fields[@enumToInt(self)];
@@ -83,7 +83,7 @@ pub const SessionSetFields = struct {
 
         const Self = @This();
 
-        pub const Fields = util.RPCFields(Self);
+        pub const Fields = util.enumFieldsToStringSlice(Self);
 
         pub fn str(self: Self) []const u8 {
             return Fields[@enumToInt(self)];
@@ -126,6 +126,100 @@ pub const SessionSetFields = struct {
     utp_enabled: ?bool = null,
 };
 
+pub const TorrentGetFields = enum {
+    activityDate,
+    addedDate,
+    availability,
+    bandwidthPriority,
+    comment,
+    corruptEver,
+    creator,
+    dateCreated,
+    desiredAvailable,
+    doneDate,
+    downloadDir,
+    downloadedEver,
+    downloadLimit,
+    downloadLimited,
+    editDate,
+    @"error",
+    errorString,
+    eta,
+    etaIdle,
+    file_count,
+    files,
+    fileStats,
+    group,
+    hashString,
+    haveUnchecked,
+    haveValid,
+    honorsSessionLimits,
+    id,
+    isFinished,
+    isPrivate,
+    isStalled,
+    labels,
+    leftUntilDone,
+    magnetLink,
+    manualAnnounceTime,
+    maxConnectedPeers,
+    metadataPercentComplete,
+    name,
+    peer_limit,
+    peers,
+    peersConnected,
+    peersFrom,
+    peersGettingFromUs,
+    peersSendingToUs,
+    percentComplete,
+    percentDone,
+    pieces,
+    pieceCount,
+    pieceSize,
+    priorities,
+    primary_mime_type,
+    queuePosition,
+    rateDownload,
+    rateUpload,
+    recheckProgress,
+    secondsDownloading,
+    secondsSeeding,
+    seedIdleLimit,
+    seedIdleMode,
+    seedRatioLimit,
+    seedRatioMode,
+    sequentialDownload,
+    sizeWhenDone,
+    startDate,
+    status,
+    trackers,
+    trackerList,
+    trackerStats,
+    totalSize,
+    torrentFile,
+    uploadedEver,
+    uploadLimit,
+    uploadLimited,
+    uploadRatio,
+    wanted,
+    webseeds,
+    webseedsSendingToUs,
+
+    const Self = @This();
+
+    pub const Fields = util.enumFieldsToStringSlice(Self);
+
+    pub fn str(self: Self) []const u8 {
+        return Fields[@enumToInt(self)];
+    }
+
+    // We have to add our own custom stringification because the default
+    // one will nest the enum inside the method name.
+    pub fn jsonStringify(self: Self, options: std.json.StringifyOptions, out_stream: anytype) !void {
+        try std.json.stringify(self.str(), options, out_stream);
+    }
+};
+
 pub const SessionGetFields = enum {
     alt_speed_down,
     alt_speed_enabled,
@@ -185,7 +279,7 @@ pub const SessionGetFields = enum {
 
     const Self = @This();
 
-    pub const Fields = util.RPCFields(Self);
+    pub const Fields = util.enumFieldsToStringSlice(Self);
 
     pub fn str(self: Self) []const u8 {
         return Fields[@enumToInt(self)];
@@ -218,27 +312,41 @@ pub const TorrentActionFields = struct {
     ids: TorrentIDs,
 };
 
+pub const TorrentStart = TorrentActionFields;
+pub const TorrentStartNow = TorrentActionFields;
+pub const TorrentStop = TorrentActionFields;
+pub const TorrentVerify = TorrentActionFields;
+pub const TorrentReannounce = TorrentActionFields;
+
+pub const TorrentGet = struct {
+    ids: ?TorrentIDs = null,
+    fields: []const TorrentGetFields,
+};
+
+pub const SessionGet = struct {
+    fields: []const SessionGetFields,
+};
+
+pub const SessionSet = SessionSetFields;
+
 pub const Request = struct {
     method: Method,
     arguments: union(Method) {
-        torrent_start: TorrentActionFields,
-        torrent_start_now: TorrentActionFields,
-        torrent_stop: TorrentActionFields,
-        torrent_verify: TorrentActionFields,
-        torrent_reannounce: TorrentActionFields,
+        torrent_start: TorrentStart,
+        torrent_start_now: TorrentStartNow,
+        torrent_stop: TorrentStop,
+        torrent_verify: TorrentVerify,
+        torrent_reannounce: TorrentReannounce,
 
         torrent_set: u8,
-        torrent_get: u8,
+        torrent_get: TorrentGet,
         torrent_add: u8,
         torrent_remove: u8,
         torrent_set_location: u8,
         torrent_rename_path: u8,
 
-        session_get: struct {
-            fields: []const SessionGetFields,
-        },
-
-        session_set: SessionSetFields,
+        session_get: SessionGet,
+        session_set: SessionSet,
 
         const Self = @This();
 
@@ -255,6 +363,8 @@ pub const Request = struct {
                 .torrent_reannounce,
                 => |v| try std.json.stringify(v, options, out_stream),
 
+                .torrent_get => |v| try std.json.stringify(v, options, out_stream),
+
                 .session_set => |v| try std.json.stringify(v, options, out_stream),
 
                 .session_get => |v| try std.json.stringify(v, options, out_stream),
@@ -274,7 +384,9 @@ test "json request encoding" {
             var req = std.ArrayList(u8).init(std.testing.allocator);
             defer req.deinit();
             try std.json.stringify(self.request, .{}, req.writer());
-            try std.testing.expect(std.mem.eql(u8, req.items, self.expected));
+            std.testing.expect(std.mem.eql(u8, req.items, self.expected)) catch {
+                std.debug.panic("{s}\n", .{req.items});
+            };
         }
     };
 
@@ -357,6 +469,25 @@ test "json request encoding" {
             \\{"method":"torrent-reannounce","arguments":{"ids":"recently-active"}}
             ,
         },
+
+        .{
+            .name = "torrent-get",
+            .request = .{
+                .method = .torrent_get,
+                .arguments = .{
+                    .torrent_get = .{
+                        .fields = &[_]TorrentGetFields{
+                            .id,
+                            .name,
+                        },
+                        .ids = .recently_active,
+                    },
+                },
+            },
+            .expected =
+            \\{"method":"torrent-get","arguments":{"ids":"recently-active","fields":["id","name"]}}
+            ,
+        },
     };
 
     for (test_cases) |tc| try tc.run();