From af39b03ef2b6ac5a1b6fbc3ab0045f73d9fa44af Mon Sep 17 00:00:00 2001
From: TotallyNot <44345987+TotallyNot@users.noreply.github.com>
Date: Mon, 20 Nov 2023 13:17:26 +0100
Subject: [PATCH] use torn-api 0.6 for the key pool

---
 torn-key-pool/Cargo.toml      |  2 +-
 torn-key-pool/src/lib.rs      | 14 ++++++++++++++
 torn-key-pool/src/postgres.rs |  2 +-
 3 files changed, 16 insertions(+), 2 deletions(-)

diff --git a/torn-key-pool/Cargo.toml b/torn-key-pool/Cargo.toml
index 3fc246e..41ac733 100644
--- a/torn-key-pool/Cargo.toml
+++ b/torn-key-pool/Cargo.toml
@@ -1,6 +1,6 @@
 [package]
 name = "torn-key-pool"
-version = "0.6.2"
+version = "0.7.0"
 edition = "2021"
 authors = ["Pyrit [2111649]"]
 license = "MIT"
diff --git a/torn-key-pool/src/lib.rs b/torn-key-pool/src/lib.rs
index e942a3c..8a8d09b 100644
--- a/torn-key-pool/src/lib.rs
+++ b/torn-key-pool/src/lib.rs
@@ -29,6 +29,20 @@ where
     Response(ResponseError),
 }
 
+impl<S, C> KeyPoolError<S, C>
+where
+    S: std::error::Error,
+    C: std::error::Error,
+{
+    #[inline(always)]
+    pub fn api_code(&self) -> Option<u8> {
+        match self {
+            Self::Response(why) => why.api_code(),
+            _ => None,
+        }
+    }
+}
+
 pub trait ApiKey: Sync + Send + std::fmt::Debug + Clone {
     type IdType: PartialEq + Eq + std::hash::Hash + Send + Sync + std::fmt::Debug + Clone;
 
diff --git a/torn-key-pool/src/postgres.rs b/torn-key-pool/src/postgres.rs
index c05997a..8899259 100644
--- a/torn-key-pool/src/postgres.rs
+++ b/torn-key-pool/src/postgres.rs
@@ -1,4 +1,4 @@
-
+use async_trait::async_trait;
 use indoc::indoc;
 use sqlx::{FromRow, PgPool, Postgres, QueryBuilder};
 use thiserror::Error;