summaryrefslogtreecommitdiffhomepage
path: root/Storage/Network.hs
blob: 41d1ff1590d7a2b7fe2711ca7371938bb9c51d9f (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
{- Copyright 2016 Joey Hess <id@joeyh.name>
 -
 - Licensed under the GNU AGPL version 3 or higher.
 -}

{-# LANGUAGE OverloadedStrings #-}

module Storage.Network (
	networkStorage,
	networkStorageOverride,
) where

import Types
import Types.Storage
import Types.Server
import Storage.Local
import HTTP.Client
import HTTP.ProofOfWork
import System.FilePath

networkStorage :: StorageLevel -> Maybe LocalStorageDirectory -> Server -> Storage
networkStorage storagelevel localdir server = Storage
	{ storeShare = store server
	, retrieveShare = retrieve server
	, obscureShares = obscure server
	, countShares = count server
	, moveShares = move server
	, uploadQueue = Just $ localStorage storagelevel (storageDir localdir)
		("uploadqueue" </> name)
	, storageLevel = storagelevel
	, getServer = Just server
	}
  where
	ServerName name = serverName server

networkStorageOverride :: Maybe LocalStorageDirectory -> HostName -> Port -> IO (Maybe Storage)
networkStorageOverride lsd h p = return $ Just $
	networkStorage LocallyPreferred lsd $ Server
		{ serverName = ServerName h
		, serverAddress = [ServerAddress h p]
		, serverDesc = h
		}

store :: Server -> StorableObjectIdent -> Share -> IO StoreResult
store srv i (Share _n o) = 
	serverRequest srv StoreFailure id i $ \pow ->
		putObject i pow o

retrieve :: Server -> ShareNum -> StorableObjectIdent -> IO RetrieveResult
retrieve srv n i = 
	serverRequest srv RetrieveFailure (RetrieveSuccess . Share n) i $
		getObject i

-- | Servers should automatically obscure, so do nothing.
-- (Could upload chaff.)
obscure :: Server -> IO ObscureResult
obscure _ = return ObscureSuccess

count :: Server -> IO CountResult
count srv = serverRequest srv CountFailure id NoPOWIdent countObjects

-- | Not needed for servers.
move :: Server -> Storage -> IO [StoreResult]
move _ _ = error "move is not implemented for servers"