Support querying instances by autoscaling group name (xen0l#212)

README.md

@@ -111,6 +111,11 @@ aws-gate session Name:SSM-test
 aws-gate session SSM-test
 ```
 
+* autoscaling group name (uses tag identifier under the hood)
+```
+aws-gate session asg:dummy-v001
+```
+
 #### SSH ProxyCommand support
 
 AWS SSM Session Manager supports tunneling SSH sessions over it. Moreover, _aws-gate_ supports generating ephemeral SSH

aws_gate/query.py

@@ -68,6 +68,11 @@ def getinstanceidbyinstancename(name, ec2=None):
     return getinstanceidbytag("Name:{}".format(name), ec2=ec2)
 
 
+def getinstancebyautoscalinggroup(name, ec2=None):
+    _, asg_name = name.split(":")
+    return getinstanceidbytag("aws:autoscaling:groupName:{}".format(asg_name), ec2=ec2)
+
+
 def query_instance(name, ec2=None):
     if ec2 is None:
         raise ValueError("EC2 client is not initialized")
@@ -82,6 +87,7 @@ def query_instance(name, ec2=None):
         "private-ip-address": getinstanceidbyprivateipaddress,
         "tag": getinstanceidbytag,
         "name": getinstanceidbyinstancename,
+        "asg": getinstancebyautoscalinggroup,
     }
 
     # If we are provided with instance ID directly, we don't need to contact EC2
@@ -99,6 +105,8 @@ def query_instance(name, ec2=None):
             identifier_type = "dns-name"
         elif name.endswith("compute.internal"):
             identifier_type = "private-dns-name"
+        elif name.startswith("asg:"):
+            identifier_type = "asg"
         elif name.count(":") >= 1:
             identifier_type = "tag"
         else:

tests/unit/test_query.py

@@ -29,6 +29,7 @@ def test_query_aws_api_exception(mocker):
         "Name:dummy-instance",
         "aws:autoscaling:groupName:dummy-v001",
         "dummy-instance",
+        "asg:dummy-v001",
     ],
     ids=[
         "private_ip_address",
@@ -39,6 +40,7 @@ def test_query_aws_api_exception(mocker):
         "tag",
         "tag (multiple colons)",
         "name",
+        "asg",
     ],
 )
 def test_query_instance(name, instance_id, ec2):