hellnet-0.1: hell-insert.hs

--------------------------------------------------------------------------------
--     This file is part of Hellnet
--
--     Hellnet is free software: you can redistribute it and/or modify
--     it under the terms of the GNU General Public License as published by
--     the Free Software Foundation, either version 3 of the License, or
--     (at your option) any later version.
--
--     Hellnet is distributed in the hope that it will be useful,
--     but WITHOUT ANY WARRANTY; without even the implied warranty of
--     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
--     GNU General Public License for more details.
--
--     You should have received a copy of the GNU General Public License
--     along with Hellnet.  If not, see <http://www.gnu.org/licenses/>.
--------------------------------------------------------------------------------

import Codec.Utils
import Control.Monad
import Hellnet
import Hellnet.Files
import Hellnet.Storage
import Hellnet.URI
import Hellnet.Utils
import qualified Data.ByteString as BS
import qualified Data.ByteString.Lazy as BSL
import qualified Data.ByteString.Char8 as BS8 (unpack,pack)
import System.Console.GetOpt
import System.Environment (getArgs)
import System.FilePath
import System.IO

data Opts = Opts {encKey :: Maybe Key, encrypt :: Bool,  chunk :: Bool}

options :: [OptDescr (Opts -> Opts)]
options = [
	Option ['e'] ["encrypt"]
		(OptArg (\s o -> o {encKey = maybe (Nothing) (Just . hexToHash) s, encrypt = True}) "key") "Encrypt (optionally with specified key)",
	Option ['c'] ["chunk"]
		(NoArg (\o -> o {chunk = True})) "Add file as single chunk (Only for files < 256 kB)"
	]

defaultOptions = Opts {encKey = Nothing, encrypt = False, chunk = False}


insertFilePrintHash :: Maybe [Octet] ->  FilePath -> IO ()
insertFilePrintHash encKey fname = do
	let filename = last (splitPath fname)
	hsh <- insertFile encKey fname
	let url = FileURI hsh encKey (Just filename)
	putStrLn $ show url

insertChunkPrintHash :: Maybe [Octet] -> FilePath -> IO ()
insertChunkPrintHash encKey fname = do
	let filename = last (splitPath fname)
	dat <- BSL.readFile fname
	when (not $ BSL.null $ BSL.drop (fromIntegral chunkSize) dat) (fail $ "Too large to insert as chunk: " ++ fname)
	hsh <- insertChunk encKey dat
	let
	let url = ChunkURI hsh encKey (Just filename)
	putStrLn $ show url

main = do
	args <- getArgs
 	let (opts, argz, errs) = getOpt Permute options args
	let optz = processOptions defaultOptions opts
	when (or [(not . null $ errs), (null argz)]) (fail $ (usageInfo "Usage: hell-insert [file] file1 [file2...]" options) ++ concat errs)
	theKey <- if encrypt optz then
		maybe (genKey >>= return . Just) (return . Just) (encKey optz)
		else
		return Nothing
	if chunk optz then
		mapM (insertChunkPrintHash theKey) argz
		else
		mapM (insertFilePrintHash theKey) argz