From 0d85a2a182a7756ac3e6803d8bd58cb8be03ffbf Mon Sep 17 00:00:00 2001
From: Hardhat Chad <hardhatchad@gmail.com>
Date: Thu, 15 Feb 2024 18:05:51 +0000
Subject: [PATCH] cleanup

---
 src/processor/create_proof.rs |  4 ++--
 src/processor/initialize.rs   | 17 ++++++++++-------
 src/state/bus.rs              |  9 +++++----
 src/state/hash.rs             |  1 -
 src/state/mod.rs              | 13 -------------
 src/state/proof.rs            | 10 ++++++----
 src/state/treasury.rs         | 10 ++++++----
 src/utils.rs                  | 13 +++++++++++++
 8 files changed, 42 insertions(+), 35 deletions(-)

diff --git a/src/processor/create_proof.rs b/src/processor/create_proof.rs
index 50998e9..7b1509c 100644
--- a/src/processor/create_proof.rs
+++ b/src/processor/create_proof.rs
@@ -8,8 +8,8 @@ use solana_program::{
 use crate::{
     instruction::CreateProofArgs,
     loaders::*,
-    state::{Discriminator, Proof},
-    utils::create_pda,
+    state::Proof,
+    utils::{create_pda, Discriminator},
     PROOF,
 };
 
diff --git a/src/processor/initialize.rs b/src/processor/initialize.rs
index 9f84403..a7490ac 100644
--- a/src/processor/initialize.rs
+++ b/src/processor/initialize.rs
@@ -1,20 +1,23 @@
 use std::mem::size_of;
 
-use solana_program::program_pack::Pack;
-use solana_program::{self, sysvar};
 use solana_program::{
-    account_info::AccountInfo, entrypoint::ProgramResult, program_error::ProgramError,
-    pubkey::Pubkey, system_program,
+    account_info::AccountInfo,
+    entrypoint::ProgramResult,
+    program_error::ProgramError,
+    program_pack::Pack,
+    pubkey::Pubkey,
+    system_program, {self, sysvar},
 };
 use spl_token::state::Mint;
 
-use crate::state::Discriminator;
-use crate::{instruction::*, BUS, INITIAL_DIFFICULTY, MINT_ADDRESS, TREASURY_ADDRESS};
 use crate::{
+    instruction::*,
     loaders::*,
     state::{Bus, Treasury},
     utils::create_pda,
-    BUS_COUNT, INITIAL_REWARD_RATE, MINT, TOKEN_DECIMALS, TREASURY,
+    utils::Discriminator,
+    BUS, BUS_COUNT, INITIAL_DIFFICULTY, INITIAL_REWARD_RATE, MINT, MINT_ADDRESS, TOKEN_DECIMALS,
+    TREASURY, TREASURY_ADDRESS,
 };
 
 pub fn process_initialize<'a, 'info>(
diff --git a/src/state/bus.rs b/src/state/bus.rs
index cfa403b..58e2c51 100644
--- a/src/state/bus.rs
+++ b/src/state/bus.rs
@@ -1,8 +1,9 @@
 use bytemuck::{Pod, Zeroable};
 
-use crate::{impl_account_from_bytes, impl_to_bytes};
-
-use super::{AccountDiscriminator, Discriminator};
+use crate::{
+    impl_account_from_bytes, impl_to_bytes,
+    utils::{AccountDiscriminator, Discriminator},
+};
 
 #[repr(C)]
 #[derive(Clone, Copy, Debug, PartialEq, Pod, Zeroable)]
@@ -18,7 +19,7 @@ pub struct Bus {
 }
 
 impl Discriminator for Bus {
-    fn discriminator() -> super::AccountDiscriminator {
+    fn discriminator() -> AccountDiscriminator {
         AccountDiscriminator::Bus
     }
 }
diff --git a/src/state/hash.rs b/src/state/hash.rs
index 5a51937..9a55030 100644
--- a/src/state/hash.rs
+++ b/src/state/hash.rs
@@ -24,4 +24,3 @@ impl From<Hash> for KeccakHash {
 }
 
 impl_to_bytes!(Hash);
-// impl_account_from_bytes!(Hash);
diff --git a/src/state/mod.rs b/src/state/mod.rs
index a646e53..64809b4 100644
--- a/src/state/mod.rs
+++ b/src/state/mod.rs
@@ -5,18 +5,5 @@ mod treasury;
 
 pub use bus::*;
 pub use hash::*;
-use num_enum::{IntoPrimitive, TryFromPrimitive};
 pub use proof::*;
 pub use treasury::*;
-
-#[repr(u8)]
-#[derive(Clone, Copy, Debug, Eq, PartialEq, IntoPrimitive, TryFromPrimitive)]
-pub enum AccountDiscriminator {
-    Bus = 100,
-    Proof = 101,
-    Treasury = 102,
-}
-
-pub trait Discriminator {
-    fn discriminator() -> AccountDiscriminator;
-}
diff --git a/src/state/proof.rs b/src/state/proof.rs
index 0bd30bc..4ac6be0 100644
--- a/src/state/proof.rs
+++ b/src/state/proof.rs
@@ -1,9 +1,11 @@
 use bytemuck::{Pod, Zeroable};
 use solana_program::pubkey::Pubkey;
 
-use crate::{impl_account_from_bytes, impl_to_bytes};
-
-use super::{AccountDiscriminator, Discriminator, Hash};
+use crate::{
+    impl_account_from_bytes, impl_to_bytes,
+    state::Hash,
+    utils::{AccountDiscriminator, Discriminator},
+};
 
 #[repr(C)]
 #[derive(Clone, Copy, Debug, PartialEq, Pod, Zeroable)]
@@ -28,7 +30,7 @@ pub struct Proof {
 }
 
 impl Discriminator for Proof {
-    fn discriminator() -> super::AccountDiscriminator {
+    fn discriminator() -> AccountDiscriminator {
         AccountDiscriminator::Proof
     }
 }
diff --git a/src/state/treasury.rs b/src/state/treasury.rs
index 97dcf72..639d653 100644
--- a/src/state/treasury.rs
+++ b/src/state/treasury.rs
@@ -1,9 +1,11 @@
 use bytemuck::{Pod, Zeroable};
 use solana_program::pubkey::Pubkey;
 
-use crate::{impl_account_from_bytes, impl_to_bytes};
-
-use super::{AccountDiscriminator, Discriminator, Hash};
+use crate::{
+    impl_account_from_bytes, impl_to_bytes,
+    state::Hash,
+    utils::{AccountDiscriminator, Discriminator},
+};
 
 #[repr(C)]
 #[derive(Clone, Copy, Debug, PartialEq, Pod, Zeroable)]
@@ -28,7 +30,7 @@ pub struct Treasury {
 }
 
 impl Discriminator for Treasury {
-    fn discriminator() -> super::AccountDiscriminator {
+    fn discriminator() -> AccountDiscriminator {
         AccountDiscriminator::Treasury
     }
 }
diff --git a/src/utils.rs b/src/utils.rs
index 044e777..1200b31 100644
--- a/src/utils.rs
+++ b/src/utils.rs
@@ -1,3 +1,4 @@
+use num_enum::{IntoPrimitive, TryFromPrimitive};
 use solana_program::{
     account_info::AccountInfo, entrypoint::ProgramResult, pubkey::Pubkey, rent::Rent,
     sysvar::Sysvar,
@@ -32,6 +33,18 @@ pub fn create_pda<'a, 'info>(
     Ok(())
 }
 
+#[repr(u8)]
+#[derive(Clone, Copy, Debug, Eq, PartialEq, IntoPrimitive, TryFromPrimitive)]
+pub enum AccountDiscriminator {
+    Bus = 100,
+    Proof = 101,
+    Treasury = 102,
+}
+
+pub trait Discriminator {
+    fn discriminator() -> AccountDiscriminator;
+}
+
 #[macro_export]
 macro_rules! impl_to_bytes {
     ($struct_name:ident) => {