Init

Author: NickSeagull

.gitignore

@@ -0,0 +1,33 @@
+.stack-work/
+aws-lambda-haskell-runtime.cabal
+*~
+
+# Created by https://www.gitignore.io/api/haskell
+# Edit at https://www.gitignore.io/?templates=haskell
+
+### Haskell ###
+dist
+dist-*
+cabal-dev
+*.o
+*.hi
+*.chi
+*.chs.h
+*.dyn_o
+*.dyn_hi
+.hpc
+.hsenv
+.cabal-sandbox/
+cabal.sandbox.config
+*.prof
+*.aux
+*.hp
+*.eventlog
+.stack-work/
+cabal.project.local
+cabal.project.local~
+.HTF/
+.ghc.environment.*
+
+# End of https://www.gitignore.io/api/haskell
+.DS_Store

.hlint.yaml

@@ -0,0 +1,59 @@
+steps:
+  - simple_align:
+      cases: true
+      top_level_patterns: true
+      records: true
+
+  # Import cleanup
+  - imports:
+      align: none
+      list_align: after_alias
+      pad_module_names: false
+      long_list_align: inline
+      empty_list_align: inherit
+      list_padding: 4
+      separate_lists: true
+      space_surround: false
+
+  - language_pragmas:
+      style: vertical
+      remove_redundant: true
+
+  # Remove trailing whitespace
+  - trailing_whitespace: {}
+
+columns: 100
+
+newline: native
+
+language_extensions:
+  - BangPatterns
+  - ConstraintKinds
+  - DataKinds
+  - DefaultSignatures
+  - DeriveAnyClass
+  - DeriveDataTypeable
+  - DeriveGeneric
+  - DerivingStrategies	
+  - FlexibleContexts
+  - FlexibleInstances
+  - FunctionalDependencies
+  - GADTs
+  - GeneralizedNewtypeDeriving
+  - InstanceSigs
+  - KindSignatures
+  - LambdaCase
+  - MultiParamTypeClasses
+  - MultiWayIf
+  - NamedFieldPuns
+  - NoImplicitPrelude
+  - OverloadedStrings
+  - QuasiQuotes
+  - RecordWildCards
+  - ScopedTypeVariables
+  - StandaloneDeriving
+  - TemplateHaskell
+  - TupleSections
+  - TypeApplications
+  - TypeFamilies
+  - ViewPatterns

.stylish-haskell.yaml

@@ -0,0 +1,3 @@
+# Changelog for aws-lambda-haskell-runtime
+
+## Unreleased changes

ChangeLog.md

@@ -0,0 +1,30 @@
+Copyright Author name here (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 Author name here 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.

LICENSE

@@ -0,0 +1 @@
+# aws-lambda-haskell-runtime

README.md

@@ -0,0 +1,2 @@
+import Distribution.Simple
+main = defaultMain

Setup.hs

@@ -0,0 +1,48 @@
+name: aws-lambda-haskell-runtime
+version: 0.1.0
+github: "theam/aws-lambda-haskell-runtime"
+license: Apache-2.0
+author: Nikita Tchayka
+maintainer: hackers@theagilemonkeys.com
+copyright: 2018 The Agile Monkeys SL
+
+extra-source-files:
+  - README.md
+  - ChangeLog.md
+
+synopsis: Haskell runtime for AWS Lambda
+category: AWS
+description: Please see the README on GitHub at <https://github.com/githubuser/aws-lambda-haskell-runtime#readme>
+
+dependencies:
+  - base >= 4.7 && < 5
+  - aeson
+  - loch-th
+  - relude
+  - time
+
+library:
+  source-dirs: src
+
+tests:
+  aws-lambda-haskell-runtime-test:
+    main: Spec.hs
+    source-dirs: test
+    ghc-options:
+      - -threaded
+      - -rtsopts
+      - -with-rtsopts=-N
+    dependencies:
+      - aws-lambda-haskell-runtime
+      - hspec
+
+default-extensions:
+  - TemplateHaskell
+  - OverloadedStrings
+  - RecordWildCards
+  - ScopedTypeVariables
+  - NoImplicitPrelude
+
+ghc-options:
+  - -Wall
+  - -Werror

package.yaml

@@ -0,0 +1,49 @@
+module AWS.Lambda.Context where
+
+import           Relude                  hiding ( identity )
+
+import           Data.Time.Clock.POSIX
+
+import           AWS.Lambda.Env
+
+
+-- TODO: Declare this in its own module
+data ClientContext
+
+-- TODO: Declare this in its own module
+data CognitoIdentity
+
+
+data Context = Context
+  { memoryLimitInMb :: Int
+  , functionName :: String
+  , functionVersion :: String
+  , invokedFunctionArn :: String
+  , awsRequestId :: String
+  , xrayTraceId :: String
+  , logStreamName :: String
+  , logGroupName :: String
+  , clientContext :: Maybe ClientContext
+  , identity :: Maybe CognitoIdentity
+  , deadline :: Word
+  }
+
+makeContext :: FunctionSettings -> Context
+makeContext FunctionSettings {..} = Context
+  { xrayTraceId        = ""
+  , memoryLimitInMb    = memorySize
+  , functionName       = functionName
+  , functionVersion    = version
+  , logStreamName      = logStream
+  , logGroupName       = logGroup
+  , invokedFunctionArn = ""
+  , awsRequestId       = ""
+  , clientContext      = Nothing
+  , identity           = Nothing
+  , deadline           = 0
+  }
+
+getTimeRemainingMillis :: Context -> IO Int
+getTimeRemainingMillis Context {..} = do
+  millis <- getPOSIXTime
+  return (fromIntegral deadline - round millis)

src/AWS/Lambda/Context.hs

@@ -0,0 +1,53 @@
+module AWS.Lambda.Env where
+
+import           Relude
+import           System.Environment
+
+import qualified AWS.Lambda.Error              as Error
+
+data FunctionSettings = FunctionSettings
+  { functionName :: String
+  , memorySize :: Int
+  , version :: String
+  , logStream :: String
+  , logGroup :: String
+  }
+
+data ProviderCapability = ProviderCapability
+  { getFunctionSettings :: IO (Either Error.RuntimeError FunctionSettings)
+  , getRuntimeApiEndpoint :: IO (Either Error.RuntimeError String)
+  }
+
+envConfig :: ProviderCapability
+envConfig = ProviderCapability
+  { getFunctionSettings   = do
+    functionName <- lookupEnv "AWS_LAMBDA_FUNCTION_NAME"
+    version      <- lookupEnv "AWS_LAMBDA_FUNCTION_VERSION"
+    logStream    <- lookupEnv "AWS_LAMBDA_LOG_STREAM_NAME"
+    logGroup     <- lookupEnv "AWS_LAMBDA_LOG_GROUP_NAME"
+    memoryStr    <- lookupEnv "AWS_LAMBDA_FUNCTION_MEMORY_SIZE"
+    let parsedMemory = memoryStr >>= readMaybe
+    case parsedMemory of
+      Nothing -> do
+        err <-
+          Error.newRuntimeError
+          $  "Could not parse memory value: "
+          <> (memoryStr ?: "<NOTHING>")
+          <> "\nMemory value from environment is not an 'Int'"
+        return $ Left err
+
+      Just (mem :: Int) -> return $ Right $ FunctionSettings
+        { functionName = fromMaybe "" functionName
+        , version      = fromMaybe "" version
+        , logStream    = fromMaybe "" logStream
+        , logGroup     = fromMaybe "" logGroup
+        , memorySize   = mem
+        }
+  , getRuntimeApiEndpoint = do
+    endpoint <- lookupEnv "AWS_LAMBDA_RUNTIME_API"
+    case endpoint of
+      Nothing -> do
+        err <- Error.newRuntimeError "Could not read endpoint, was it set?"
+        return $ Left err
+      Just ep -> return $ Right ep
+  }

src/AWS/Lambda/Env.hs

@@ -0,0 +1,41 @@
+module AWS.Lambda.Error where
+
+import           Relude
+import           System.Environment
+import           Debug.Trace.LocationTH
+
+data RuntimeError = RuntimeError
+  { runtimeErrorMsg :: String
+  , runtimeErrorStackTrace :: Maybe String
+  , runtimeErrorRecoverable :: Bool
+  , runtimeErrorRequestId :: Maybe String
+  } deriving (Show)
+
+data HandlerError = HandlerError
+  { handlerErrorMsg :: String
+  , handlerErrorStackTrace :: Maybe String
+  } deriving (Show)
+
+unrecoverable :: RuntimeError
+unrecoverable = RuntimeError
+  { runtimeErrorMsg         = ""
+  , runtimeErrorStackTrace  = Nothing
+  , runtimeErrorRecoverable = False
+  , runtimeErrorRequestId   = Nothing
+  }
+
+newRuntimeError :: String -> IO RuntimeError
+newRuntimeError runtimeErrorMsg = do
+  shouldTrace <- lookupEnv "HASKELL_BACKTRACE"
+  let runtimeError = RuntimeError
+        { runtimeErrorMsg         = runtimeErrorMsg
+        , runtimeErrorStackTrace  = Nothing
+        , runtimeErrorRecoverable = True
+        , runtimeErrorRequestId   = Nothing
+        }
+  case shouldTrace of
+    Just "1" -> do
+      let tr = $__LOCATION__
+      return $ runtimeError { runtimeErrorStackTrace = Just tr }
+
+    _ -> return runtimeError

src/AWS/Lambda/Error.hs

@@ -0,0 +1,112 @@
+module AWS.Lambda.Runtime where
+
+import           Relude
+
+import           Data.Aeson
+
+import qualified AWS.Lambda.Context            as Context
+import qualified AWS.Lambda.Env                as Env
+import qualified AWS.Lambda.Error              as Error
+
+defaultMaxRetries :: Int
+defaultMaxRetries = 3
+
+type Handler e o = e -> Context.Context -> IO (Either Error.HandlerError o)
+
+start :: (FromJSON e, ToJSON o) => Handler e o -> Option runtime -> IO ()
+start f = startWithConfig f Env.envConfig
+
+startWithConfig
+  :: (FromJSON e, ToJSON o)
+  => Handler e o
+  -> Env.ProviderCapability
+  -> Option runtime
+  -> IO ()
+startWithConfig f Env.ProviderCapability {..} r = do
+  let showError = error . toText . Error.runtimeErrorMsg
+  -- endpoint         <- either showError id <$> getRuntimeApiEndpoint
+  functionSettings <- either showError id <$> getFunctionSettings
+  runtimeClient    <- either showError id <$> newRuntimeClient
+  startWithRuntimeClient f functionSettings runtimeClient
+
+
+startWithRuntimeClient
+  :: (FromJSON e, ToJSON o)
+  => Handler e o
+  -> Env.FunctionSettings
+  -> RuntimeClient
+  -> IO ()
+startWithRuntimeClient = undefined
+
+data Runtime e o = Runtime
+  { runtimeClient :: RuntimeClient
+  , handler :: Handler e o
+  , maxRetries :: Int
+  , settings :: Env.FunctionSettings
+  }
+
+data RuntimeClient
+
+newRuntimeClient :: IO (Either Error.RuntimeError RuntimeClient)
+newRuntimeClient = undefined
+
+eventResponse :: RuntimeClient -> String -> w -> Maybe a
+eventResponse = undefined
+
+eventError :: RuntimeClient -> String -> Maybe a
+eventError = undefined
+
+startRuntime :: Runtime e o -> IO ()
+startRuntime = loopRuntime
+ where
+  loopRuntime Runtime {..} = do
+    (event, ctx) <- getNextEvent 0 Nothing
+    let requestId                 = Context.awsRequestId ctx
+    let err :: Error.RuntimeError = undefined
+    res <- event ctx
+    case res of
+      Just (response :: Int) -> do
+        let responseBytes = encode response
+        case eventResponse runtimeClient requestId responseBytes of
+          Just _ ->
+            putTextLn
+              $  "Response for"
+              <> show requestId
+              <> "accepted by Runtime API"
+          Nothing -> do
+            putTextLn
+              $  "Could not send response for "
+              <> show requestId
+              <> " to Runtime API: "
+              <> show err
+            unless
+              (Error.runtimeErrorRecoverable err)
+              (putTextLn
+                "Error is not recoverable, sending fail_init signal and panicking."
+              )
+
+      Nothing -> do
+        putTextLn
+          $  "Handler returned an error for "
+          <> show requestId
+          <> ": "
+          <> show err
+          <> ""
+
+        case eventError runtimeClient requestId of
+          Just _  -> putTextLn "Error accepted by Runtime API"
+          Nothing -> do
+            putTextLn
+              $  "Could not send response for "
+              <> show requestId
+              <> " to Runtime API: "
+              <> show err
+
+            unless
+              (Error.runtimeErrorRecoverable err)
+              (putTextLn
+                "Error is not recoverable, sending fail_init signal and panicking."
+              )
+
+getNextEvent :: Int -> Maybe error -> IO (e, Context.Context)
+getNextEvent = undefined

src/AWS/Lambda/Runtime.hs

@@ -0,0 +1,9 @@
+module Lib
+  ( someFunc
+  )
+where
+
+import           Relude
+
+someFunc :: IO ()
+someFunc = putTextLn "someFunc"

src/Lib.hs

@@ -0,0 +1,64 @@
+# 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
+#
+# 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-12.13
+
+# 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:
+  - fmt-0.6.1.1
+# 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.9"
+#
+# 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

stack.yaml

@@ -0,0 +1,3 @@
+import Relude
+main :: IO ()
+main = putTextLn "Test suite not yet implemented"