1 files changed, 36 insertions, 2 deletions

diff --git a/lib/Types.hs b/lib/Types.hs
index 9aa0340..767ee98 100644
--- a/lib/Types.hs
+++ b/lib/Types.hs
@@ -1,12 +1,13 @@
 {-# LANGUAGE DeriveFunctor #-}
 module Types where
 
-import           Data.Map           as Map (Map)
+import           Data.Map as Map (Map)
+import qualified Data.Map as Map
 import           Data.OpenPGP
 import           Data.OpenPGP.Util
 import           FunctorToMaybe
-import qualified System.Posix.Types as Posix
 import qualified Data.ByteString.Lazy as L
+import qualified System.Posix.Types as Posix
 
 -- | This type describes an idempotent transformation (merge or import) on a
 -- set of GnuPG keyrings and other key files.
@@ -199,6 +200,26 @@ data OriginMapped a = MappedPacket
 instance Functor OriginMapped where
     fmap f (MappedPacket x ls) = MappedPacket (f x) ls
 
+origin :: Packet -> Int -> OriginFlags
+origin p n = OriginFlags ispub n
+ where
+    ispub = case p of
+                    SecretKeyPacket {} -> False
+                    _                  -> True
+
+mappedPacket :: FilePath -> Packet -> MappedPacket
+mappedPacket filename p = MappedPacket
+    { packet = p
+    , locations = Map.singleton filename (origin p (-1))
+    }
+
+mappedPacketWithHint :: FilePath -> Packet -> Int -> MappedPacket
+mappedPacketWithHint filename p hint = MappedPacket
+    { packet = p
+    , locations = Map.singleton filename (origin p hint)
+    }
+
+
 -- | This type is used to indicate success or failure
 -- and in the case of success, return the computed object.
 -- The 'FunctorToMaybe' class is implemented to facilitate
@@ -252,6 +273,11 @@ isKey (PublicKeyPacket {}) = True
 isKey (SecretKeyPacket {}) = True
 isKey _                    = False
 
+isSecretKey :: Packet -> Bool
+isSecretKey (SecretKeyPacket {}) = True
+isSecretKey _                    = False
+
+
 isUserID :: Packet -> Bool
 isUserID (UserIDPacket {}) = True
 isUserID _                 = False
@@ -260,4 +286,12 @@ isTrust :: Packet -> Bool
 isTrust (TrustPacket {}) = True
 isTrust _                = False
 
+-- matchpr computes the fingerprint of the given key truncated to
+-- be the same lenght as the given fingerprint for comparison.
+--
+-- matchpr fp = Data.List.Extra.takeEnd (length fp)
+--
+matchpr :: String -> Packet -> String
+matchpr fp k = reverse $ zipWith const (reverse (fingerprint k)) fp
+