localstack-terraform-samples/apigatewayv2-ws-http/main.tf at master · localstack-samples/localstack-terraform-samples · GitHub

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
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
resource "random_pet" "random" {
  length    = 2
  separator = "-"
}


resource "aws_iam_role" "execution_role" {
  name               = "api-gateway-role"
  assume_role_policy = <<EOF
{
    "Version": "2012-10-17",
    "Statement": [
        {
        "Sid": "",
        "Effect": "Allow",
        "Principal": {
            "Service": "apigateway.amazonaws.com"
        },
        "Action": "sts:AssumeRole"
        }
    ]
}
EOF
}

resource "aws_apigatewayv2_api" "websockets" {
  name                       = random_pet.random.id
  protocol_type              = "WEBSOCKET"
  route_selection_expression = "$request.body.action"
}

resource "aws_apigatewayv2_route" "connect_route" {
  api_id    = aws_apigatewayv2_api.websockets.id
  route_key = "$connect"
  target    = "integrations/${aws_apigatewayv2_integration.http.id}"
}

resource "aws_apigatewayv2_route" "disconnect_route" {
  api_id    = aws_apigatewayv2_api.websockets.id
  route_key = "$disconnect"
  target    = "integrations/${aws_apigatewayv2_integration.http.id}"
}

resource "aws_apigatewayv2_route" "default_route" {
  api_id    = aws_apigatewayv2_api.websockets.id
  route_key = "$default"
  target    = "integrations/${aws_apigatewayv2_integration.http.id}"
}

resource "aws_apigatewayv2_integration" "http" {
  api_id                        = aws_apigatewayv2_api.websockets.id
  credentials_arn               = aws_iam_role.execution_role.arn
  payload_format_version        = "1.0"
  integration_type              = "HTTP"
  integration_method            = "POST"
  integration_uri               = "https://httpbin.org/anything/echo"
  template_selection_expression = "default"
  request_templates = {
    "default" = file("${path.module}/template.json")
  }
}

resource "aws_apigatewayv2_integration_response" "integration_response" {
  api_id                        = aws_apigatewayv2_api.websockets.id
  integration_id                = aws_apigatewayv2_integration.http.id
  template_selection_expression = "$integration.response.statuscode"
  integration_response_key      = "/200/"
  response_templates = {
    "200" = <<EOF
    $input.body
    EOF
  }
}

resource "aws_apigatewayv2_route_response" "connect_response" {
  api_id             = aws_apigatewayv2_api.websockets.id
  route_id           = aws_apigatewayv2_route.connect_route.id
  route_response_key = "$default"
}

resource "aws_apigatewayv2_route_response" "disconnect_response" {
  api_id             = aws_apigatewayv2_api.websockets.id
  route_id           = aws_apigatewayv2_route.disconnect_route.id
  route_response_key = "$default"
}

resource "aws_apigatewayv2_route_response" "default_response" {
  api_id             = aws_apigatewayv2_api.websockets.id
  route_id           = aws_apigatewayv2_route.default_route.id
  route_response_key = "$default"
}

resource "aws_apigatewayv2_stage" "stage" {
  api_id      = aws_apigatewayv2_api.websockets.id
  name        = "dev"
  auto_deploy = true

  access_log_settings {
    destination_arn = aws_cloudwatch_log_group.log_group.arn
    format = jsonencode({
      requestId               = "$context.requestId"
      sourceIp                = "$context.identity.sourceIp"
      requestTime             = "$context.requestTime"
      message                 = "$context.messageId"
      status                  = "$context.status"
      responseLength          = "$context.responseLength"
      integrationErrorMessage = "$context.integrationErrorMessage"
      connectionId            = "$context.connectionId"
    })
  }

  default_route_settings {
    data_trace_enabled     = true
    logging_level          = "INFO"
    throttling_burst_limit = 100
    throttling_rate_limit  = 100
  }
}

resource "aws_apigatewayv2_deployment" "deployment" {
  api_id = aws_apigatewayv2_api.websockets.id
  depends_on = [
    aws_apigatewayv2_integration_response.integration_response,
    aws_apigatewayv2_route_response.default_response,
  ]
}

resource "aws_cloudwatch_log_group" "log_group" {
  name              = random_pet.random.id
  retention_in_days = 1
}

resource "aws_cloudwatch_log_stream" "log_stream" {
  name           = random_pet.random.id
  log_group_name = aws_cloudwatch_log_group.log_group.name

  depends_on = [
    aws_cloudwatch_log_group.log_group
  ]
}

output "ws_url" {
  value = aws_apigatewayv2_api.websockets.api_endpoint
}