Initial commit

.editorconfig

@@ -0,0 +1,9 @@
+root = true
+
+[*.cr]
+charset = utf-8
+end_of_line = lf
+insert_final_newline = true
+indent_style = space
+indent_size = 2
+trim_trailing_whitespace = true

.gitignore

@@ -0,0 +1,9 @@
+/docs/
+/lib/
+/bin/
+/.shards/
+*.dwarf
+
+# Libraries don't need dependency lock
+# Dependencies will be locked in application that uses them
+/shard.lock

LICENSE

@@ -0,0 +1,19 @@
+Copyright 2018 Michael Hart and LambCI contributors
+
+Permission is hereby granted, free of charge, to any person obtaining a copy of
+this software and associated documentation files (the "Software"), to deal in
+the Software without restriction, including without limitation the rights to
+use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies
+of the Software, and to permit persons to whom the Software is furnished to do
+so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

README.md

@@ -0,0 +1,37 @@
+# crambda
+
+TODO: Write a description here
+
+## Installation
+
+1. Add the dependency to your `shard.yml`:
+```yaml
+dependencies:
+  crambda:
+    github: your-github-user/crambda
+```
+2. Run `shards install`
+
+## Usage
+
+```crystal
+require "crambda"
+```
+
+TODO: Write usage instructions here
+
+## Development
+
+TODO: Write development instructions here
+
+## Contributing
+
+1. Fork it (<https://github.com/your-github-user/crambda/fork>)
+2. Create your feature branch (`git checkout -b my-new-feature`)
+3. Commit your changes (`git commit -am 'Add some feature'`)
+4. Push to the branch (`git push origin my-new-feature`)
+5. Create a new Pull Request
+
+## Contributors
+
+- [your-name-here](https://github.com/your-github-user) - creator and maintainer

shard.yml

@@ -0,0 +1,6 @@
+name: crambda
+version: 0.0.1
+
+crystal: 0.27.0
+
+license: MIT

src/crambda.cr

@@ -0,0 +1,64 @@
+require "http/client"
+require "json"
+
+module Crambda
+  VERSION = "0.0.1"
+
+  class LambdaContext
+    getter function_name : String
+    getter function_version : String
+    getter function_memory_size : UInt32
+    getter log_group_name : String
+    getter log_stream_name : String
+    getter aws_request_id : String
+    getter invoked_function_arn : String
+    getter deadline : Time
+    getter identity : JSON::Any
+    getter client_context : JSON::Any
+
+    def initialize(@function_name, @function_version, @function_memory_size,
+                   @log_group_name, @log_stream_name, @aws_request_id, @invoked_function_arn,
+                   @deadline, @identity, @client_context)
+    end
+  end
+
+  def self.start_lambda(&block : (JSON::Any, LambdaContext) -> JSON::Any)
+    function_name = ENV["AWS_LAMBDA_RUNTIME_API"]
+    function_version = ENV["AWS_LAMBDA_FUNCTION_VERSION"]
+    function_memory_size = UInt32.new(ENV["AWS_LAMBDA_FUNCTION_MEMORY_SIZE"])
+    log_group_name = ENV["AWS_LAMBDA_LOG_GROUP_NAME"]
+    log_stream_name = ENV["AWS_LAMBDA_LOG_STREAM_NAME"]
+    host, port = ENV["AWS_LAMBDA_RUNTIME_API"].split(':')
+
+    client = HTTP::Client.new(host, port)
+
+    while true
+      res = client.get("/2018-06-01/runtime/invocation/next")
+      if res.status_code != 200
+        raise "Unexpected response when invoking: #{res.status_code}"
+      end
+
+      ENV["_X_AMZN_TRACE_ID"] = res.headers["Lambda-Runtime-Aws-Request-Id"]? || ""
+
+      context = LambdaContext.new(
+        function_name,
+        function_version,
+        function_memory_size,
+        log_group_name,
+        log_stream_name,
+        res.headers["Lambda-Runtime-Aws-Request-Id"],
+        res.headers["Lambda-Runtime-Invoked-Function-Arn"],
+        Time.unix_ms(Int64.new(res.headers["Lambda-Runtime-Deadline-Ms"])),
+        JSON.parse(res.headers["Lambda-Runtime-Cognito-Identity"]? || "null"),
+        JSON.parse(res.headers["Lambda-Runtime-Client-Context"]? || "null"),
+      )
+
+      result = block.call(JSON.parse(res.body), context)
+
+      res = client.post("/2018-06-01/runtime/invocation/#{context.aws_request_id}/response", body: result.to_json)
+      if res.status_code != 202
+        raise "Unexpected response when responding: #{res.status_code}"
+      end
+    end
+  end
+end