progress
This commit is contained in:
parent
475f707361
commit
72f94cc42e
4 changed files with 66 additions and 65 deletions
|
@ -53,7 +53,7 @@ remoteTypes =
|
||||||
]
|
]
|
||||||
|
|
||||||
{- Builds a list of all available Remotes.
|
{- Builds a list of all available Remotes.
|
||||||
- Since doing so can be expensive, the list is cached in the Annex. -}
|
- Since doing so can be expensive, the list is cached. -}
|
||||||
genList :: Annex [Remote Annex]
|
genList :: Annex [Remote Annex]
|
||||||
genList = do
|
genList = do
|
||||||
rs <- Annex.getState Annex.remotes
|
rs <- Annex.getState Annex.remotes
|
||||||
|
@ -130,7 +130,7 @@ remoteLog = do
|
||||||
getConfigs :: [Remote Annex] -> Annex [Remote Annex]
|
getConfigs :: [Remote Annex] -> Annex [Remote Annex]
|
||||||
getConfigs rs = do
|
getConfigs rs = do
|
||||||
m <- readRemoteLog
|
m <- readRemoteLog
|
||||||
return $ map (get m) rs
|
return $ map (get m) rs
|
||||||
where
|
where
|
||||||
get m r = r { config = M.lookup (uuid r) m }
|
get m r = r { config = M.lookup (uuid r) m }
|
||||||
|
|
||||||
|
|
|
@ -14,7 +14,7 @@ import Control.Exception.Extensible
|
||||||
import Control.Monad.State (liftIO)
|
import Control.Monad.State (liftIO)
|
||||||
import qualified Data.Map as M
|
import qualified Data.Map as M
|
||||||
import System.Cmd.Utils
|
import System.Cmd.Utils
|
||||||
import Control.Monad (filterM, liftM)
|
import Control.Monad (filterM, liftM, when)
|
||||||
import Data.String.Utils
|
import Data.String.Utils
|
||||||
import Maybe
|
import Maybe
|
||||||
|
|
||||||
|
@ -50,18 +50,19 @@ gen = do
|
||||||
- cached UUID value. -}
|
- cached UUID value. -}
|
||||||
let cheap = filter (not . Git.repoIsUrl) allremotes
|
let cheap = filter (not . Git.repoIsUrl) allremotes
|
||||||
let expensive = filter Git.repoIsUrl allremotes
|
let expensive = filter Git.repoIsUrl allremotes
|
||||||
expensive_todo <- filterM cachedUUID expensive
|
expensive_todo <- filterM noCachedUUID expensive
|
||||||
let skip = filter (`notElem` expensive_todo) expensive
|
let skip = filter (`notElem` expensive_todo) expensive
|
||||||
let todo = cheap++expensive_todo
|
let todo = cheap++expensive_todo
|
||||||
|
|
||||||
showNote $ "getting UUID for " ++ (join ", " $
|
when (not $ null expensive_todo) $
|
||||||
map Git.repoDescribe expensive_todo)
|
showNote $ "getting UUID for " ++ (join ", " $
|
||||||
|
map Git.repoDescribe expensive_todo)
|
||||||
done <- mapM tryGitConfigRead todo
|
done <- mapM tryGitConfigRead todo
|
||||||
|
|
||||||
generated <- mapM genRemote $ skip ++ done
|
generated <- mapM genRemote $ skip ++ done
|
||||||
return $ catMaybes generated
|
return $ catMaybes generated
|
||||||
where
|
where
|
||||||
cachedUUID r = liftM null $ getUUID r
|
noCachedUUID r = liftM null $ getUUID r
|
||||||
|
|
||||||
genRemote :: Git.Repo -> Annex (Maybe (Remote Annex))
|
genRemote :: Git.Repo -> Annex (Maybe (Remote Annex))
|
||||||
genRemote r = do
|
genRemote r = do
|
||||||
|
|
114
Remote/S3.hs
114
Remote/S3.hs
|
@ -7,6 +7,7 @@
|
||||||
|
|
||||||
module Remote.S3 (remote) where
|
module Remote.S3 (remote) where
|
||||||
|
|
||||||
|
import Control.Exception.Extensible (IOException)
|
||||||
import Network.AWS.AWSConnection
|
import Network.AWS.AWSConnection
|
||||||
import Network.AWS.S3Object
|
import Network.AWS.S3Object
|
||||||
import Network.AWS.S3Bucket
|
import Network.AWS.S3Bucket
|
||||||
|
@ -15,10 +16,9 @@ import qualified Data.ByteString.Lazy.Char8 as L
|
||||||
import qualified Data.Map as M
|
import qualified Data.Map as M
|
||||||
import Data.Maybe
|
import Data.Maybe
|
||||||
import Data.String.Utils
|
import Data.String.Utils
|
||||||
import Control.Monad (filterM, liftM, when)
|
import Control.Monad (filterM, when)
|
||||||
import Control.Monad.State (liftIO)
|
import Control.Monad.State (liftIO)
|
||||||
import System.Environment
|
import System.Environment
|
||||||
import Data.Char
|
|
||||||
|
|
||||||
import RemoteClass
|
import RemoteClass
|
||||||
import Types
|
import Types
|
||||||
|
@ -62,17 +62,21 @@ genRemote r = do
|
||||||
then return Nothing
|
then return Nothing
|
||||||
else do
|
else do
|
||||||
c <- remoteCost r
|
c <- remoteCost r
|
||||||
return $ Just $ Remote {
|
return $ Just $ newremote u c
|
||||||
uuid = u,
|
where
|
||||||
cost = c,
|
newremote u c = this
|
||||||
name = Git.repoDescribe r,
|
where
|
||||||
storeKey = error "TODO",
|
this = Remote {
|
||||||
retrieveKeyFile = error "TODO",
|
uuid = u,
|
||||||
removeKey = error "TODO",
|
cost = c,
|
||||||
hasKey = error "TODO",
|
name = Git.repoDescribe r,
|
||||||
hasKeyCheap = False,
|
storeKey = s3Store this,
|
||||||
config = Nothing
|
retrieveKeyFile = error "TODO retrievekey",
|
||||||
}
|
removeKey = error "TODO removekey",
|
||||||
|
hasKey = s3CheckPresent this,
|
||||||
|
hasKeyCheap = False,
|
||||||
|
config = Nothing
|
||||||
|
}
|
||||||
|
|
||||||
s3Connection :: M.Map String String -> IO AWSConnection
|
s3Connection :: M.Map String String -> IO AWSConnection
|
||||||
s3Connection c = do
|
s3Connection c = do
|
||||||
|
@ -93,10 +97,10 @@ s3Setup u c = do
|
||||||
case M.lookup "encryption" c of
|
case M.lookup "encryption" c of
|
||||||
Nothing -> error "Specify encryption=key or encryption=none"
|
Nothing -> error "Specify encryption=key or encryption=none"
|
||||||
Just "none" -> return ()
|
Just "none" -> return ()
|
||||||
Just k -> error "encryption keys not yet supported"
|
Just _ -> error "encryption keys not yet supported"
|
||||||
let fullconfig = M.union c defaults
|
let fullconfig = M.union c defaults
|
||||||
|
|
||||||
-- check bucket location to see if the bucket exists
|
-- check bucket location to see if the bucket exists, and create it
|
||||||
let datacenter = fromJust $ M.lookup "datacenter" fullconfig
|
let datacenter = fromJust $ M.lookup "datacenter" fullconfig
|
||||||
conn <- liftIO $ s3Connection fullconfig
|
conn <- liftIO $ s3Connection fullconfig
|
||||||
showNote "checking bucket"
|
showNote "checking bucket"
|
||||||
|
@ -105,7 +109,7 @@ s3Setup u c = do
|
||||||
Right _ -> return ()
|
Right _ -> return ()
|
||||||
Left err@(NetworkError _) -> error $ prettyReqError err
|
Left err@(NetworkError _) -> error $ prettyReqError err
|
||||||
Left (AWSError _ _) -> do
|
Left (AWSError _ _) -> do
|
||||||
showNote "creating bucket"
|
showNote $ "creating bucket in " ++ datacenter
|
||||||
res <- liftIO $ createBucketIn conn bucket datacenter
|
res <- liftIO $ createBucketIn conn bucket datacenter
|
||||||
case res of
|
case res of
|
||||||
Right _ -> return ()
|
Right _ -> return ()
|
||||||
|
@ -113,12 +117,13 @@ s3Setup u c = do
|
||||||
|
|
||||||
g <- Annex.gitRepo
|
g <- Annex.gitRepo
|
||||||
liftIO $ do
|
liftIO $ do
|
||||||
Git.run g "config" [Param ("remote." ++ name ++ ".annex-s3"), Param "true"]
|
Git.run g "config" [Param (configsetting "annex-s3"), Param "true"]
|
||||||
Git.run g "config" [Param ("remote." ++ name ++ ".annex-uuid"), Param u]
|
Git.run g "config" [Param (configsetting "annex-uuid"), Param u]
|
||||||
return fullconfig
|
return fullconfig
|
||||||
where
|
where
|
||||||
name = fromJust (M.lookup "name" c)
|
remotename = fromJust (M.lookup "name" c)
|
||||||
bucket = name ++ "-" ++ u
|
bucket = remotename ++ "-" ++ u
|
||||||
|
configsetting s = "remote." ++ remotename ++ "." ++ s
|
||||||
defaults = M.fromList
|
defaults = M.fromList
|
||||||
[ ("datacenter", "US")
|
[ ("datacenter", "US")
|
||||||
, ("storageclass", "STANDARD")
|
, ("storageclass", "STANDARD")
|
||||||
|
@ -127,44 +132,39 @@ s3Setup u c = do
|
||||||
, ("bucket", bucket)
|
, ("bucket", bucket)
|
||||||
]
|
]
|
||||||
|
|
||||||
{-
|
s3Action :: Remote Annex -> ((AWSConnection, String) -> Annex a) -> Annex a
|
||||||
|
s3Action r a = do
|
||||||
|
when (config r == Nothing) $
|
||||||
|
error $ "Missing configuration for special remote " ++ name r
|
||||||
|
conn <- liftIO $ s3Connection (fromJust $ config r)
|
||||||
|
let bucket = fromJust $ M.lookup "bucket" $ fromJust $ config r
|
||||||
|
a (conn, bucket)
|
||||||
|
|
||||||
{- The UUID of a S3 bucket is stored in a file "git-annex-uuid" in the
|
s3File :: Key -> FilePath
|
||||||
- bucket. Gets the UUID, or if there is none, sets a new UUID, possibly
|
s3File k = show k
|
||||||
- also creating the bucket. -}
|
|
||||||
getS3UUID :: Git.Repo -> Annex UUID
|
s3CheckPresent :: Remote Annex -> Key -> Annex (Either IOException Bool)
|
||||||
getS3UUID r = withS3Connection r disable $ \(c, b) -> do
|
s3CheckPresent r k = s3Action r $ \(conn, bucket) -> do
|
||||||
res <- liftIO $
|
let object = S3Object bucket (s3File k) "" [] L.empty
|
||||||
getObject c $ S3Object b uuidfile "" [] L.empty
|
showNote ("checking " ++ name r ++ "...")
|
||||||
|
res <- liftIO $ getObjectInfo conn object
|
||||||
case res of
|
case res of
|
||||||
Right o -> return $ L.unpack $ obj_data o
|
Right _ -> return $ Right True
|
||||||
Left _ -> do
|
Left (AWSError _ _) -> return $ Right False
|
||||||
location <- getS3Config r "s3-datacenter" (Just "EU")
|
Left e -> return $ Left (error $ prettyReqError e)
|
||||||
-- bucket may already exist, or not
|
|
||||||
_ <- liftIO $ createBucketIn c b location
|
|
||||||
u <- getUUID r
|
|
||||||
res' <- liftIO $ sendObject c $
|
|
||||||
S3Object b uuidfile "" [] $
|
|
||||||
L.pack u
|
|
||||||
case res' of
|
|
||||||
Right _ -> return u
|
|
||||||
Left e -> do
|
|
||||||
warning $ prettyReqError e
|
|
||||||
disable
|
|
||||||
|
|
||||||
where
|
|
||||||
uuidfile = "git-annex-uuid"
|
|
||||||
disable = return "" -- empty uuid will disable this remote
|
|
||||||
|
|
||||||
getS3Config :: Git.Repo -> String -> Maybe String-> Annex String
|
s3Store :: Remote Annex -> Key -> Annex Bool
|
||||||
getS3Config r s def = do
|
s3Store r k = s3Action r $ \(conn, bucket) -> do
|
||||||
e <- liftIO $ catch (liftM Just $ getEnv envvar) (const $ return def)
|
let object = setStorageClass storageclass $
|
||||||
v <- case e of
|
S3Object bucket (s3File k) "" [] (error "read content here")
|
||||||
Nothing -> getConfig r s ""
|
res <- liftIO $ sendObject conn object
|
||||||
Just d -> getConfig r s d
|
case res of
|
||||||
when (null v) $ error $ "set " ++ envvar ++ " or " ++ remoteConfig r s
|
Right _ -> return True
|
||||||
return v
|
Left e -> do
|
||||||
|
warning $ prettyReqError e
|
||||||
|
return False
|
||||||
where
|
where
|
||||||
envvar = "ANNEX_" ++ map (\c -> if c == '-' then '_' else toUpper c) s
|
storageclass =
|
||||||
|
case fromJust $ M.lookup "storageclass" $ fromJust $ config r of
|
||||||
-}
|
"REDUCED_REDUNDANCY" -> REDUCED_REDUNDANCY
|
||||||
|
_ -> STANDARD
|
||||||
|
|
|
@ -13,7 +13,7 @@ First, export your S3 credentials:
|
||||||
Next, create the S3 remote.
|
Next, create the S3 remote.
|
||||||
|
|
||||||
# git annex initremote mys3 type=S3 encryption=none
|
# git annex initremote mys3 type=S3 encryption=none
|
||||||
initremote mys3 (checking bucket) (creating bucket) ok
|
initremote mys3 (checking bucket) (creating bucket in US) ok
|
||||||
|
|
||||||
The configuration for the S3 remote is stored in git. So to make a different
|
The configuration for the S3 remote is stored in git. So to make a different
|
||||||
repository use the same S3 remote is easy:
|
repository use the same S3 remote is easy:
|
||||||
|
|
Loading…
Reference in a new issue