2 files changed, 18 insertions, 32 deletions

diff --git a/src/git_remote_helper.rs b/src/git_remote_helper.rs
index 68aa681..08a59c1 100644
--- a/src/git_remote_helper.rs
+++ b/src/git_remote_helper.rs
@@ -6,7 +6,7 @@
 
 use core::str;
 use std::{
-    collections::HashSet,
+    collections::{HashMap, HashSet},
     env,
     io::{self, Stdin},
     path::PathBuf,
@@ -261,41 +261,25 @@ async fn generate_updated_state(
     git_repo: &Repo,
     repo_ref: &RepoRef,
     refspecs: &Vec<String>,
-) -> Result<Vec<(String, String)>> {
+) -> Result<HashMap<String, String>> {
     let new_state = {
         if let Ok(mut repo_state) = get_state_from_cache(git_repo.get_path()?, repo_ref).await {
             for refspec in refspecs {
                 let (from, to) = refspec_to_from_to(refspec)?;
-                if to.is_empty() {
+                if from.is_empty() {
                     // delete
-                    repo_state.state.retain(|(name, _)| !name.eq(to));
-                } else if repo_state.state.iter().any(|(name, _)| name.eq(from)) {
-                    // update
-                    repo_state.state = repo_state
-                        .state
-                        .iter()
-                        .map(|(name, value)| {
-                            (
-                                name.clone(),
-                                if name.eq(to) {
-                                    reference_to_ref_value(&git_repo.git_repo, to).unwrap()
-                                } else {
-                                    value.to_string()
-                                },
-                            )
-                        })
-                        .collect();
+                    repo_state.state.remove(to);
                 } else {
-                    // add
-                    repo_state.state.push((
+                    // add or update
+                    repo_state.state.insert(
                         to.to_string(),
                         reference_to_ref_value(&git_repo.git_repo, to).unwrap(),
-                    ));
+                    );
                 }
             }
             repo_state.state
         } else {
-            let mut state = vec![];
+            let mut state = HashMap::new();
             let git_server_url = repo_ref
                 .git_server
                 .first()
@@ -303,14 +287,14 @@ async fn generate_updated_state(
             let mut git_server_remote = git_repo.git_repo.remote_anonymous(git_server_url)?;
             git_server_remote.connect(git2::Direction::Fetch)?;
             for head in git_server_remote.list()? {
-                state.push((
+                state.insert(
                     head.name().to_string(),
                     if let Some(symbolic_ref) = head.symref_target() {
-                        format!("ref: {}", symbolic_ref)
+                        format!("ref: {symbolic_ref}")
                     } else {
                         head.oid().to_string()
                     },
-                ));
+                );
             }
             git_server_remote.disconnect()?;
             state
@@ -460,7 +444,7 @@ fn get_refspecs_from_push_batch(stdin: &Stdin, initial_refspec: &str) -> Result<
 impl RepoState {
     pub async fn build(
         identifier: String,
-        state: Vec<(String, String)>,
+        state: HashMap<String, String>,
         signer: &NostrSigner,
     ) -> Result<RepoState> {
         let mut tags = vec![Tag::identifier(identifier.clone())];
diff --git a/src/repo_state.rs b/src/repo_state.rs
index 0c1aa30..a5cebab 100644
--- a/src/repo_state.rs
+++ b/src/repo_state.rs
@@ -1,9 +1,11 @@
+use std::collections::HashMap;
+
 use anyhow::{Context, Result};
 use git2::Oid;
 
 pub struct RepoState {
     pub identifier: String,
-    pub state: Vec<(String, String)>,
+    pub state: HashMap<String, String>,
     pub event: nostr::Event,
 }
 
@@ -11,7 +13,7 @@ impl RepoState {
     pub fn try_from(mut state_events: Vec<nostr::Event>) -> Result<Self> {
         state_events.sort_by_key(|e| e.created_at);
         let event = state_events.first().context("no state events")?;
-        let mut state = vec![];
+        let mut state = HashMap::new();
         for tag in &event.tags {
             if let Some(name) = tag.as_vec().first() {
                 if ["refs/heads/", "refs/tags", "HEAD"]
@@ -19,8 +21,8 @@ impl RepoState {
                     .any(|s| name.starts_with(*s))
                 {
                     if let Some(value) = tag.as_vec().get(1) {
-                        if Oid::from_str(value).is_ok() {
-                            state.push((name.to_owned(), value.to_owned()));
+                        if Oid::from_str(value).is_ok() || value.contains("ref: refs/") {
+                            state.insert(name.to_owned(), value.to_owned());
                         }
                     }
                 }