81 lines
2.5 KiB
Python
81 lines
2.5 KiB
Python
#!/usr/bin/env python3
|
|
|
|
import json
|
|
import sys
|
|
import os
|
|
import urllib.request
|
|
from argparse import ArgumentParser
|
|
|
|
def main():
|
|
parser = ArgumentParser()
|
|
parser.add_argument('--list', action='store_true')
|
|
parser.add_argument('--host')
|
|
args = parser.parse_args()
|
|
|
|
if args.host:
|
|
print('{}')
|
|
return
|
|
|
|
token = os.environ.get('LINODE_API_TOKEN')
|
|
if not token:
|
|
print('{"_meta": {"hostvars": {}}}')
|
|
return
|
|
|
|
try:
|
|
req = urllib.request.Request('https://api.linode.com/v4/linode/instances')
|
|
req.add_header('Authorization', f'Bearer {token}')
|
|
|
|
with urllib.request.urlopen(req, timeout=30) as response:
|
|
data = json.loads(response.read().decode('utf-8'))
|
|
|
|
inventory = {"_meta": {"hostvars": {}}, "all": {"children": []}}
|
|
groups = {}
|
|
|
|
for instance in data.get('data', []):
|
|
if instance['status'] != 'running':
|
|
continue
|
|
|
|
hostname = instance['label']
|
|
ip = instance['ipv4'][0] if instance.get('ipv4') else None
|
|
if not ip:
|
|
continue
|
|
|
|
# Host vars
|
|
tags = instance.get('tags', [])
|
|
inventory['_meta']['hostvars'][hostname] = {
|
|
'ansible_host': ip,
|
|
'ansible_user': 'phlux',
|
|
'linode_region': instance['region'],
|
|
'linode_type': instance['type'],
|
|
'linode_tags': tags,
|
|
'is_k3s': 'k3s' in tags,
|
|
'is_control_plane': 'control-plane' in tags,
|
|
'is_worker_node': 'worker-node' in tags
|
|
}
|
|
|
|
# Create groups
|
|
for tag in tags:
|
|
group = f"tag_{tag.replace('-', '_')}"
|
|
if group not in groups:
|
|
groups[group] = []
|
|
groups[group].append(hostname)
|
|
|
|
region_group = f"region_{instance['region'].replace('-', '_')}"
|
|
if region_group not in groups:
|
|
groups[region_group] = []
|
|
groups[region_group].append(hostname)
|
|
|
|
# Add groups to inventory
|
|
for group, hosts in groups.items():
|
|
inventory[group] = {"hosts": hosts}
|
|
inventory['all']['children'].append(group)
|
|
|
|
print(json.dumps(inventory, indent=2))
|
|
|
|
except Exception as e:
|
|
print(f'{{"error": "{e}"}}', file=sys.stderr)
|
|
print('{"_meta": {"hostvars": {}}}')
|
|
|
|
if __name__ == '__main__':
|
|
main()
|