import paramiko
import sys

sys.stdout.reconfigure(encoding='utf-8')
key_path = r'C:\Users\Snarf\.ssh\id_ed25519_snarfnet'

c = paramiko.SSHClient()
c.set_missing_host_key_policy(paramiko.AutoAddPolicy())
c.connect("140.44.4.75", username="root", key_filename=key_path, timeout=10)

token = "eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJpc3MiOiIzZWUyNTE1MDk0NGU0ZjRjYmJkMGM1MWI5MThmY2EzYiIsImlhdCI6MTc0NjgxMDM3NSwiZXhwIjoyMDYyMTcwMzc1fQ.Le6M7WL63nxtLTxccFacQR5GVUY5C-a_zGtmXT_uaz4"

cmd = f"""
docker exec $(docker ps -q -f name=heimdall) curl -k -i -X POST https://snarfhome.snarfnet.net/api/template \
  -H "Authorization: Bearer {token}" \
  -H "Content-Type: application/json" \
  -d '{{"template": "{{{{ states.light | count}}}}"}}'
"""

stdin, stdout, stderr = c.exec_command(cmd)
print("Heimdall curl test to snarfhome.snarfnet.net:\n", stdout.read().decode())

cmd2 = f"""
docker exec $(docker ps -q -f name=heimdall) curl -k -i -X POST http://14.4.4.20:8123/api/template \
  -H "Authorization: Bearer {token}" \
  -H "Content-Type: application/json" \
  -d '{{"template": "{{{{ states.light | count}}}}"}}'
"""
stdin, stdout, stderr = c.exec_command(cmd2)
print("Heimdall curl test to 14.4.4.20:8123:\n", stdout.read().decode())

c.close()