From 50cfcdf54b828fbeab532b712e00063ae9e82581 Mon Sep 17 00:00:00 2001
From: Joey Hess <joey@kitenet.net>
Date: Sun, 17 Apr 2011 01:13:21 -0400
Subject: make encrypted remotes have slightly higher costs

---
 Config.hs             | 17 ++++++++++++++++-
 Remote/Encryptable.hs |  4 +++-
 test.hs               |  2 ++
 3 files changed, 21 insertions(+), 2 deletions(-)

diff --git a/Config.hs b/Config.hs
index 53f1a455f..a324427d4 100644
--- a/Config.hs
+++ b/Config.hs
@@ -52,10 +52,25 @@ remoteCost r def = do
 cheapRemoteCost :: Int
 cheapRemoteCost = 100
 semiCheapRemoteCost :: Int
-semiCheapRemoteCost = 150
+semiCheapRemoteCost = 110
 expensiveRemoteCost :: Int
 expensiveRemoteCost = 200
 
+{- Adjust's a remote's cost to reflect it being encrypted. -}
+encryptedRemoteCostAdj :: Int
+encryptedRemoteCostAdj = 50
+
+{- Make sure the remote cost numbers work out. -}
+prop_cost_sane :: Bool
+prop_cost_sane = False `notElem`
+	[ expensiveRemoteCost > 0
+	, cheapRemoteCost < semiCheapRemoteCost
+	, semiCheapRemoteCost < expensiveRemoteCost
+	, cheapRemoteCost + encryptedRemoteCostAdj > semiCheapRemoteCost
+	, cheapRemoteCost + encryptedRemoteCostAdj < expensiveRemoteCost
+	, semiCheapRemoteCost + encryptedRemoteCostAdj < expensiveRemoteCost
+	]
+
 {- Checks if a repo should be ignored, based either on annex-ignore
  - setting, or on command-line options. Allows command-line to override
  - annex-ignore. -}
diff --git a/Remote/Encryptable.hs b/Remote/Encryptable.hs
index a9a7472fb..aa7c2a569 100644
--- a/Remote/Encryptable.hs
+++ b/Remote/Encryptable.hs
@@ -15,6 +15,7 @@ import RemoteClass
 import Crypto
 import qualified Annex
 import Messages
+import Config
 
 {- Encryption setup for a remote. The user must specify whether to use
  - an encryption key, or not encrypt. An encrypted cipher is created, or is
@@ -48,7 +49,8 @@ encryptableRemote c storeKeyEncrypted retrieveKeyFileEncrypted r =
 		storeKey = store,
 		retrieveKeyFile = retrieve,
 		removeKey = withkey $ removeKey r,
-		hasKey = withkey $ hasKey r
+		hasKey = withkey $ hasKey r,
+		cost = cost r + encryptedRemoteCostAdj
 	}
 	where
 		store k = do
diff --git a/test.hs b/test.hs
index 30ebe6e59..cdec4ea61 100644
--- a/test.hs
+++ b/test.hs
@@ -39,6 +39,7 @@ import qualified Remote
 import qualified Content
 import qualified Command.DropUnused
 import qualified Key
+import qualified Config
 
 main :: IO ()
 main = do
@@ -61,6 +62,7 @@ quickcheck = TestLabel "quickcheck" $ TestList
 	, qctest "prop_idempotent_shellEscape_multiword" Utility.prop_idempotent_shellEscape_multiword
 	, qctest "prop_parentDir_basics" Utility.prop_parentDir_basics
 	, qctest "prop_relPathDirToDir_basics" Utility.prop_relPathDirToDir_basics
+	, qctest "prop_cost_sane" Config.prop_cost_sane
 	]
 
 blackbox :: Test
-- 
cgit v1.2.3