added NGROK_REGION env

This commit is contained in:
vital987 2021-10-30 17:17:40 +05:30
parent 8c892aea01
commit dede11f84b
2 changed files with 18 additions and 4 deletions

View File

@ -24,7 +24,8 @@ ENV VNC_PASS="samplepass" \
#Ngrok Specific
NGROK_BINDTLS=1 \
NGROK_METHOD=tcp \
NGROK_PORT=5900
NGROK_PORT=5900 \
NGROK_REGION="in"
SHELL ["/bin/bash", "-c"]

View File

@ -1,21 +1,34 @@
#!/usr/bin/python3
import os
try:
from pyngrok import ngrok, conf
except:
os.system('pip3 install pyngrok')
from pyngrok import ngrok, conf
try:
AUTH_TOKEN=os.environ['NGROK_AUTH_TOKEN']
except:
print('[!] Ngrok token not found, assign the token to NGROK_AUTH_TOKEN environment variable & try again.\n[-] Terminating...')
print('[-] Ngrok token not found, assign the token to NGROK_AUTH_TOKEN environment variable & try again.\n[-] Terminating...')
exit()
ngrok.set_auth_token(AUTH_TOKEN)
conf.get_default().region = "in"
if "NGROK_REGION" in os.environ:
conf.get_default().region = os.environ['NGROK_REGION']
else:
print("[!] NGROK_REGION unset, defaulting ngrok location to India.")
conf.get_default().region = "in"
try:
primary_tunnel = ngrok.connect(os.environ['NGROK_PORT'], os.environ['NGROK_METHOD'], bind_tls=bool(int(os.getenv('NGROK_BINDTLS', 1))))
except:
print('[!] NGROK_PORT or NGROK_METHOD environment variables has not been set properly, NGROK_METHOD should be http or tcp')
print('[-] NGROK_PORT or NGROK_METHOD environment variables has not been set properly, NGROK_METHOD should be http or tcp')
url=primary_tunnel.public_url
print(f'[>NGROK]: {url}')
input()