parent
24a5eae3f4
commit
174b1af658
7 changed files with 225 additions and 0 deletions
@ -0,0 +1,30 @@ |
||||
Copyright Gaël Depreeuw (c) 2018 |
||||
|
||||
All rights reserved. |
||||
|
||||
Redistribution and use in source and binary forms, with or without |
||||
modification, are permitted provided that the following conditions are met: |
||||
|
||||
* Redistributions of source code must retain the above copyright |
||||
notice, this list of conditions and the following disclaimer. |
||||
|
||||
* Redistributions in binary form must reproduce the above |
||||
copyright notice, this list of conditions and the following |
||||
disclaimer in the documentation and/or other materials provided |
||||
with the distribution. |
||||
|
||||
* Neither the name of Gaël Depreeuw nor the names of other |
||||
contributors may be used to endorse or promote products derived |
||||
from this software without specific prior written permission. |
||||
|
||||
THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS |
||||
"AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT |
||||
LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR |
||||
A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT |
||||
OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, |
||||
SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT |
||||
LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, |
||||
DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY |
||||
THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT |
||||
(INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE |
||||
OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. |
@ -0,0 +1 @@ |
||||
# client |
@ -0,0 +1,2 @@ |
||||
import Distribution.Simple |
||||
main = defaultMain |
@ -0,0 +1,25 @@ |
||||
name: client |
||||
version: 0.1.0.0 |
||||
-- synopsis: |
||||
-- description: |
||||
homepage: https://github.com/Mithror/client#readme |
||||
license: BSD3 |
||||
license-file: LICENSE |
||||
author: Gaël Depreeuw |
||||
maintainer: gael.depreeuw@gmail.com |
||||
copyright: Copyright (c) 2018 Gaël Depreeuw |
||||
category: daemon |
||||
build-type: Simple |
||||
cabal-version: >=1.10 |
||||
extra-source-files: README.md |
||||
|
||||
executable client |
||||
hs-source-dirs: src |
||||
main-is: Main.hs |
||||
default-language: Haskell2010 |
||||
build-depends: base >= 4.7 && < 5 |
||||
, network |
||||
, text |
||||
, bytestring |
||||
, transformers |
||||
, optparse-applicative |
@ -0,0 +1,102 @@ |
||||
{-# LANGUAGE OverloadedStrings #-} |
||||
|
||||
module Main where |
||||
|
||||
import qualified Control.Exception as E |
||||
import qualified Data.ByteString.Char8 as C |
||||
import System.IO |
||||
import Network.Socket hiding (recv) |
||||
import Network.Socket.ByteString (recv, sendAll) |
||||
import Data.Text (Text) |
||||
import Data.String (fromString) |
||||
import Control.Monad.Trans.Reader |
||||
import Control.Monad.IO.Class (liftIO) |
||||
import Options.Applicative |
||||
import Data.Semigroup ((<>)) |
||||
|
||||
type Host = String |
||||
type Port = String |
||||
|
||||
data Config = Config { configHost :: Host |
||||
, configPort :: Port |
||||
} deriving (Eq, Show) |
||||
|
||||
type ConfigT a = ReaderT Config IO a |
||||
|
||||
sendData :: C.ByteString -> ConfigT () |
||||
sendData byteString = do |
||||
config <- ask |
||||
liftIO $ withSocketsDo $ do |
||||
addr <- resolve (configHost config) (configPort config) |
||||
E.bracket (open addr) close (talk byteString) |
||||
where |
||||
resolve h p = do |
||||
let hints = defaultHints { addrSocketType = Stream } |
||||
addr:_ <- getAddrInfo (Just hints) (Just h) (Just p) |
||||
return addr |
||||
open addr = do |
||||
sock <- socket (addrFamily addr) (addrSocketType addr) defaultProtocol --(addrProtocol addr) |
||||
connect sock $ addrAddress addr |
||||
return sock |
||||
talk bs sock = do |
||||
sendAll sock $ bs |
||||
msg <- recv sock 1024 |
||||
C.putStrLn msg |
||||
|
||||
sendJSON :: FilePath -> ConfigT () |
||||
sendJSON filePath = |
||||
(liftIO $ readFile filePath) >>= (sendData . fromString . (++ "\0")) |
||||
|
||||
sendFinger :: String -> ConfigT () |
||||
sendFinger = sendData . fromString . (++ "\r\n") |
||||
|
||||
data Arguments = Finger String Config |
||||
| FileInput FilePath Config |
||||
deriving (Show, Eq) |
||||
|
||||
configInput :: Parser Config |
||||
configInput = Config |
||||
<$> strOption |
||||
( long "host" |
||||
<> short 'h' |
||||
<> metavar "HOST" |
||||
<> help "host to connect to") |
||||
<*> strOption |
||||
( long "port" |
||||
<> short 'p' |
||||
<> metavar "PORT" |
||||
<> help "port of daemon") |
||||
|
||||
fingerInput :: Parser Arguments |
||||
fingerInput = Finger |
||||
<$> strOption |
||||
( long "name" |
||||
<> short 'n' |
||||
<> metavar "NAME" |
||||
<> help "name of person to look up") |
||||
<*> configInput |
||||
|
||||
fileInput :: Parser Arguments |
||||
fileInput = FileInput |
||||
<$> strOption |
||||
( long "file" |
||||
<> short 'f' |
||||
<> metavar "FILENAME" |
||||
<> help "json file with config") |
||||
<*> configInput |
||||
|
||||
input :: Parser Arguments |
||||
input = fingerInput <|> fileInput |
||||
|
||||
handleArguments :: Arguments -> IO () |
||||
handleArguments (Finger n c) = runReaderT (sendFinger n) c |
||||
handleArguments (FileInput f c) = runReaderT (sendJSON f) c |
||||
|
||||
main :: IO () |
||||
main = handleArguments =<< execParser opts |
||||
where |
||||
opts = info (input <**> helper) |
||||
( fullDesc |
||||
<> progDesc "Finger a person or update the server's database." |
||||
<> header "client - a tool to talk to the fingerd" ) |
||||
|
@ -0,0 +1,65 @@ |
||||
# This file was automatically generated by 'stack init' |
||||
# |
||||
# Some commonly used options have been documented as comments in this file. |
||||
# For advanced use and comprehensive documentation of the format, please see: |
||||
# https://docs.haskellstack.org/en/stable/yaml_configuration/ |
||||
|
||||
# Resolver to choose a 'specific' stackage snapshot or a compiler version. |
||||
# A snapshot resolver dictates the compiler version and the set of packages |
||||
# to be used for project dependencies. For example: |
||||
# |
||||
# resolver: lts-3.5 |
||||
# resolver: nightly-2015-09-21 |
||||
# resolver: ghc-7.10.2 |
||||
# resolver: ghcjs-0.1.0_ghc-7.10.2 |
||||
# |
||||
# The location of a snapshot can be provided as a file or url. Stack assumes |
||||
# a snapshot provided as a file might change, whereas a url resource does not. |
||||
# |
||||
# resolver: ./custom-snapshot.yaml |
||||
# resolver: https://example.com/snapshots/2018-01-01.yaml |
||||
resolver: lts-11.15 |
||||
|
||||
# User packages to be built. |
||||
# Various formats can be used as shown in the example below. |
||||
# |
||||
# packages: |
||||
# - some-directory |
||||
# - https://example.com/foo/bar/baz-0.0.2.tar.gz |
||||
# - location: |
||||
# git: https://github.com/commercialhaskell/stack.git |
||||
# commit: e7b331f14bcffb8367cd58fbfc8b40ec7642100a |
||||
# - location: https://github.com/commercialhaskell/stack/commit/e7b331f14bcffb8367cd58fbfc8b40ec7642100a |
||||
# subdirs: |
||||
# - auto-update |
||||
# - wai |
||||
packages: |
||||
- . |
||||
# Dependency packages to be pulled from upstream that are not in the resolver |
||||
# using the same syntax as the packages field. |
||||
# (e.g., acme-missiles-0.3) |
||||
# extra-deps: [] |
||||
|
||||
# Override default flag values for local packages and extra-deps |
||||
# flags: {} |
||||
|
||||
# Extra package databases containing global packages |
||||
# extra-package-dbs: [] |
||||
|
||||
# Control whether we use the GHC we find on the path |
||||
# system-ghc: true |
||||
# |
||||
# Require a specific version of stack, using version ranges |
||||
# require-stack-version: -any # Default |
||||
# require-stack-version: ">=1.7" |
||||
# |
||||
# Override the architecture used by stack, especially useful on Windows |
||||
# arch: i386 |
||||
# arch: x86_64 |
||||
# |
||||
# Extra directories used by stack for building |
||||
# extra-include-dirs: [/path/to/dir] |
||||
# extra-lib-dirs: [/path/to/dir] |
||||
# |
||||
# Allow a newer minor version of GHC than the snapshot specifies |
||||
# compiler-check: newer-minor |
Binary file not shown.
Loading…
Reference in new issue