From 0ecd8fb0741cbb93e8e451d9404a6630372967ad Mon Sep 17 00:00:00 2001
From: Alexis Sellier <self@cloudhead.io>
Date: Thu, 27 Apr 2023 10:54:19 +0200
Subject: [PATCH] cli: Add `--draft` to `patch list` command

---
 radicle-cli/src/commands/patch.rs |  6 +++++-
 radicle/src/cob/patch.rs          | 13 +++++++++----
 2 files changed, 14 insertions(+), 5 deletions(-)

diff --git a/radicle-cli/src/commands/patch.rs b/radicle-cli/src/commands/patch.rs
index 9b58e90d..7d068267 100644
--- a/radicle-cli/src/commands/patch.rs
+++ b/radicle-cli/src/commands/patch.rs
@@ -40,7 +40,7 @@ pub const HELP: Help = Help {
 Usage
 
     rad patch [<option>...]
-    rad patch list [--all|--merged|--open|--archived] [<option>...]
+    rad patch list [--all|--merged|--open|--archived|--draft] [<option>...]
     rad patch show <patch-id> [<option>...]
     rad patch open [--draft] [<option>...]
     rad patch archive <patch-id> [<option>...]
@@ -68,6 +68,7 @@ List options
         --archived             Show only archived patches
         --merged               Show only merged patches
         --open                 Show only open patches (default)
+        --draft                Show only draft patches
 
 Ready options
 
@@ -209,6 +210,9 @@ impl Args for Options {
                 Long("all") => {
                     filter = None;
                 }
+                Long("draft") => {
+                    filter = Some(patch::State::Draft);
+                }
                 Long("archived") => {
                     filter = Some(patch::State::Archived);
                 }
diff --git a/radicle/src/cob/patch.rs b/radicle/src/cob/patch.rs
index f9f76de5..c5d1826e 100644
--- a/radicle/src/cob/patch.rs
+++ b/radicle/src/cob/patch.rs
@@ -276,12 +276,17 @@ impl Patch {
 
     /// Check if the patch is open.
     pub fn is_open(&self) -> bool {
-        matches!(self.state.get().get(), State::Open)
+        matches!(self.state(), State::Open)
     }
 
     /// Check if the patch is archived.
     pub fn is_archived(&self) -> bool {
-        matches!(self.state.get().get(), &State::Archived)
+        matches!(self.state(), State::Archived)
+    }
+
+    /// Check if the patch is a draft.
+    pub fn is_draft(&self) -> bool {
+        matches!(self.state(), State::Draft)
     }
 }
 
@@ -982,7 +987,7 @@ impl<'a, 'g> PatchMut<'a, 'g> {
 
     /// Mark a patch as ready to be reviewed. Returns `false` if the patch was not a draft.
     pub fn ready<G: Signer>(&mut self, signer: &G) -> Result<bool, Error> {
-        if self.state() != State::Draft {
+        if !self.is_draft() {
             return Ok(false);
         }
         self.lifecycle(State::Open, signer)?;
@@ -992,7 +997,7 @@ impl<'a, 'g> PatchMut<'a, 'g> {
 
     /// Mark an open patch as a draft. Returns `false` if the patch was not open.
     pub fn unready<G: Signer>(&mut self, signer: &G) -> Result<bool, Error> {
-        if self.state() != State::Open {
+        if !self.is_open() {
             return Ok(false);
         }
         self.lifecycle(State::Draft, signer)?;