spinifex/hive/gateway/ec2.go

package gateway

import (
	"fmt"
	"log/slog"

	"github.com/aws/aws-sdk-go/service/ec2"
	"github.com/davecgh/go-spew/spew"
	"github.com/gofiber/fiber/v2"
	"github.com/mulgadc/hive/hive/awsec2query"
	"github.com/mulgadc/hive/hive/awsparser"
)

func (gw *GatewayConfig) EC2_Request(ctx *fiber.Ctx) error {

	fmt.Println("HERE!")

	fmt.Println("Body")
	spew.Dump(ctx.Body())

	fmt.Println("Request POST")
	spew.Dump(ctx.Request())

	// Headers
	fmt.Println("Headers")
	spew.Dump(ctx.GetReqHeaders())

	queryArgs := parseAWSQueryArgs(string(ctx.Body()))

	// Run the action
	switch queryArgs["Action"] {
	case "DescribeInstances":
		return gw.EC2_DescribeInstances(ctx, queryArgs)
	case "RunInstances":
		return gw.EC2_RunInstances(ctx, queryArgs)
	case "CreateKeyPair":
		return gw.EC2_CreateKeyPair(ctx, queryArgs)
	default:
		slog.Warn("EC2 Unsupported Action", "action", queryArgs["Action"])
	}

	return fiber.NewError(fiber.StatusNotImplemented, "Action not implemented")
}

func (gw *GatewayConfig) EC2_DescribeInstances(ctx *fiber.Ctx, args map[string]string) error {

	slog.Info("EC2 DescribeInstances called")

	// Return a dummy response
	response := `<?xml version="1.0" encoding="UTF-8"?>
<DescribeInstancesResponse xmlns="http://ec2.amazonaws.com/doc/2016-11-15/">
  <requestId>2b7ac2f1-9acd-4d73-b0a1-1b2e5607f9ab</requestId>
  <reservationSet>
    <item>
      <reservationId>r-0f6b3c4d5e6789abc</reservationId>
      <ownerId>123456789012</ownerId>
      <groupSet/>
      <instancesSet>
        <item>
          <instanceId>i-0123456789abcdef0</instanceId>
          <imageId>ami-08d4ac5b634553e16</imageId>
          <instanceState>
            <code>16</code>
            <name>running</name>
          </instanceState>
          <privateDnsName>ip-10-0-1-25.ap-southeast-2.compute.internal</privateDnsName>
          <dnsName>ec2-3-26-14-112.ap-southeast-2.compute.amazonaws.com</dnsName>
          <reason/>
          <keyName>my-keypair</keyName>
          <amiLaunchIndex>0</amiLaunchIndex>
          <productCodes/>
          <instanceType>t3.medium</instanceType>
          <launchTime>2025-02-18T04:12:07.000Z</launchTime>
          <placement>
            <availabilityZone>ap-southeast-2a</availabilityZone>
            <groupName/>
            <tenancy>default</tenancy>
          </placement>
          <monitoring>
            <state>disabled</state>
          </monitoring>
          <privateIpAddress>10.0.1.25</privateIpAddress>
          <ipAddress>3.26.14.112</ipAddress>
          <subnetId>subnet-0ab1c2d3e4f567890</subnetId>
          <vpcId>vpc-0123abcd4567efgh8</vpcId>
          <sourceDestCheck>true</sourceDestCheck>
          <groupSet>
            <item>
              <groupId>sg-0abc1234def567890</groupId>
              <groupName>default</groupName>
            </item>
          </groupSet>
          <stateReason/>
          <architecture>x86_64</architecture>
          <rootDeviceType>ebs</rootDeviceType>
          <rootDeviceName>/dev/xvda</rootDeviceName>
          <blockDeviceMapping>
            <item>
              <deviceName>/dev/xvda</deviceName>
              <ebs>
                <volumeId>vol-06f5e4d3c2b1a0f98</volumeId>
                <status>attached</status>
                <attachTime>2025-02-18T04:12:09.000Z</attachTime>
                <deleteOnTermination>true</deleteOnTermination>
              </ebs>
            </item>
          </blockDeviceMapping>
          <virtualizationType>hvm</virtualizationType>
          <tagSet>
            <item>
              <key>Name</key>
              <value>web-01</value>
            </item>
          </tagSet>
          <hypervisor>xen</hypervisor>
          <networkInterfaceSet>
            <item>
              <networkInterfaceId>eni-0a12b34c56d78ef90</networkInterfaceId>
              <subnetId>subnet-0ab1c2d3e4f567890</subnetId>
              <vpcId>vpc-0123abcd4567efgh8</vpcId>
              <description/>
              <ownerId>123456789012</ownerId>
              <status>in-use</status>
              <macAddress>0a:1b:2c:3d:4e:5f</macAddress>
              <privateIpAddress>10.0.1.25</privateIpAddress>
              <privateDnsName>ip-10-0-1-25.ap-southeast-2.compute.internal</privateDnsName>
              <sourceDestCheck>true</sourceDestCheck>
              <groupSet>
                <item>
                  <groupId>sg-0abc1234def567890</groupId>
                  <groupName>default</groupName>
                </item>
              </groupSet>
              <attachment>
                <attachmentId>eni-attach-0abc123def4567890</attachmentId>
                <deviceIndex>0</deviceIndex>
                <status>attached</status>
                <attachTime>2025-02-18T04:12:07.000Z</attachTime>
                <deleteOnTermination>true</deleteOnTermination>
              </attachment>
              <privateIpAddressesSet>
                <item>
                  <privateIpAddress>10.0.1.25</privateIpAddress>
                  <privateDnsName>ip-10-0-1-25.ap-southeast-2.compute.internal</privateDnsName>
                  <primary>true</primary>
                </item>
              </privateIpAddressesSet>
            </item>
          </networkInterfaceSet>
        </item>
      </instancesSet>
    </item>
  </reservationSet>
</DescribeInstancesResponse>
`

	return ctx.SendString(response)
}

func (gw *GatewayConfig) EC2_RunInstances(ctx *fiber.Ctx, args map[string]string) error {

	slog.Info("EC2 RunInstances called")

	// Convert HTTP post args, to the input struct
	var ec2RunInstance = &ec2.RunInstancesInput{}
	awsec2query.QueryParamsToStruct(args, ec2RunInstance)

	// Send the query
	xmlResponse, err := awsparser.EC2_RunInstances(ec2RunInstance)

	if err != nil {
		slog.Error("EC2 RunInstances error", "error", err)
		return fiber.NewError(fiber.StatusInternalServerError, "EC2 RunInstances error")
	}

	// Return the XML response
	ctx.Set("Content-Type", "text/xml")
	return ctx.XML(xmlResponse)

	return nil

}

func (gw *GatewayConfig) EC2_CreateKeyPair(ctx *fiber.Ctx, args map[string]string) error {
	slog.Info("EC2 CreateKeyPair called")

	return nil
}