|
|
|
|
@ -1,142 +1,30 @@
|
|
|
|
|
import subprocess
|
|
|
|
|
import re
|
|
|
|
|
import ngrok
|
|
|
|
|
import pyqrcode
|
|
|
|
|
import time
|
|
|
|
|
from ..utils.print_markdown import print_markdown
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def create_tunnel(
|
|
|
|
|
tunnel_method="ngrok", server_host="localhost", server_port=10001, qr=False, domain=None
|
|
|
|
|
):
|
|
|
|
|
print_markdown("Exposing server to the internet...")
|
|
|
|
|
|
|
|
|
|
server_url = ""
|
|
|
|
|
if tunnel_method == "bore":
|
|
|
|
|
try:
|
|
|
|
|
output = subprocess.check_output("command -v bore", shell=True)
|
|
|
|
|
except subprocess.CalledProcessError:
|
|
|
|
|
print(
|
|
|
|
|
"The bore-cli command is not available. Please run 'cargo install bore-cli'."
|
|
|
|
|
)
|
|
|
|
|
print("For more information, see https://github.com/ekzhang/bore")
|
|
|
|
|
exit(1)
|
|
|
|
|
|
|
|
|
|
time.sleep(6)
|
|
|
|
|
# output = subprocess.check_output(f'bore local {server_port} --to bore.pub', shell=True)
|
|
|
|
|
process = subprocess.Popen(
|
|
|
|
|
f"bore local {server_port} --to bore.pub",
|
|
|
|
|
shell=True,
|
|
|
|
|
stdout=subprocess.PIPE,
|
|
|
|
|
stderr=subprocess.STDOUT,
|
|
|
|
|
universal_newlines=True,
|
|
|
|
|
)
|
|
|
|
|
|
|
|
|
|
while True:
|
|
|
|
|
line = process.stdout.readline()
|
|
|
|
|
print(line)
|
|
|
|
|
if not line:
|
|
|
|
|
break
|
|
|
|
|
if "listening at bore.pub:" in line:
|
|
|
|
|
remote_port = re.search("bore.pub:([0-9]*)", line).group(1)
|
|
|
|
|
server_url = f"bore.pub:{remote_port}"
|
|
|
|
|
print_markdown(
|
|
|
|
|
f"Your server is being hosted at the following URL: bore.pub:{remote_port}"
|
|
|
|
|
)
|
|
|
|
|
break
|
|
|
|
|
|
|
|
|
|
elif tunnel_method == "localtunnel":
|
|
|
|
|
if subprocess.call("command -v lt", shell=True):
|
|
|
|
|
print("The 'lt' command is not available.")
|
|
|
|
|
print(
|
|
|
|
|
"Please ensure you have Node.js installed, then run 'npm install -g localtunnel'."
|
|
|
|
|
)
|
|
|
|
|
print(
|
|
|
|
|
"For more information, see https://github.com/localtunnel/localtunnel"
|
|
|
|
|
)
|
|
|
|
|
exit(1)
|
|
|
|
|
else:
|
|
|
|
|
process = subprocess.Popen(
|
|
|
|
|
f"npx localtunnel --port {server_port}",
|
|
|
|
|
shell=True,
|
|
|
|
|
stdout=subprocess.PIPE,
|
|
|
|
|
stderr=subprocess.STDOUT,
|
|
|
|
|
universal_newlines=True,
|
|
|
|
|
)
|
|
|
|
|
server_host="localhost", server_port=10001, qr=False, domain=None
|
|
|
|
|
):
|
|
|
|
|
"""
|
|
|
|
|
To use most of ngrok’s features, you’ll need an authtoken. To obtain one, sign up for free at ngrok.com and
|
|
|
|
|
retrieve it from the authtoken page in your ngrok dashboard.
|
|
|
|
|
|
|
|
|
|
found_url = False
|
|
|
|
|
url_pattern = re.compile(r"your url is: https://[a-zA-Z0-9.-]+")
|
|
|
|
|
https://dashboard.ngrok.com/get-started/your-authtoken
|
|
|
|
|
|
|
|
|
|
while True:
|
|
|
|
|
line = process.stdout.readline()
|
|
|
|
|
if not line:
|
|
|
|
|
break # Break out of the loop if no more output
|
|
|
|
|
match = url_pattern.search(line)
|
|
|
|
|
if match:
|
|
|
|
|
found_url = True
|
|
|
|
|
remote_url = match.group(0).replace("your url is: ", "")
|
|
|
|
|
server_url = remote_url
|
|
|
|
|
print(
|
|
|
|
|
f"\nYour server is being hosted at the following URL: {remote_url}"
|
|
|
|
|
)
|
|
|
|
|
break # Exit the loop once the URL is found
|
|
|
|
|
|
|
|
|
|
if not found_url:
|
|
|
|
|
print(
|
|
|
|
|
"Failed to extract the localtunnel URL. Please check localtunnel's output for details."
|
|
|
|
|
)
|
|
|
|
|
|
|
|
|
|
elif tunnel_method == "ngrok":
|
|
|
|
|
# Check if ngrok is installed
|
|
|
|
|
is_installed = (
|
|
|
|
|
subprocess.check_output("command -v ngrok", shell=True).decode().strip()
|
|
|
|
|
)
|
|
|
|
|
if not is_installed:
|
|
|
|
|
print("The ngrok command is not available.")
|
|
|
|
|
print(
|
|
|
|
|
"Please install ngrok using the instructions at https://ngrok.com/docs/getting-started/"
|
|
|
|
|
)
|
|
|
|
|
exit(1)
|
|
|
|
|
|
|
|
|
|
# If ngrok is installed, start it on the specified port
|
|
|
|
|
# process = subprocess.Popen(f'ngrok http {server_port} --log=stdout', shell=True, stdout=subprocess.PIPE)
|
|
|
|
|
|
|
|
|
|
if domain:
|
|
|
|
|
domain = f"--domain={domain}"
|
|
|
|
|
else:
|
|
|
|
|
domain = ""
|
|
|
|
|
process = subprocess.Popen(
|
|
|
|
|
f"ngrok http {server_port} --scheme http,https {domain} --log=stdout",
|
|
|
|
|
shell=True,
|
|
|
|
|
stdout=subprocess.PIPE,
|
|
|
|
|
)
|
|
|
|
|
|
|
|
|
|
# Initially, no URL is found
|
|
|
|
|
found_url = False
|
|
|
|
|
# Regular expression to match the ngrok URL
|
|
|
|
|
url_pattern = re.compile(r"https://[a-zA-Z0-9-]+\.ngrok(-free)?\.app")
|
|
|
|
|
You can set it as `NGROK_AUTHTOKEN` in your environment variables
|
|
|
|
|
"""
|
|
|
|
|
print_markdown("Exposing server to the internet...")
|
|
|
|
|
|
|
|
|
|
# Read the output line by line
|
|
|
|
|
while True:
|
|
|
|
|
line = process.stdout.readline().decode("utf-8")
|
|
|
|
|
if not line:
|
|
|
|
|
break # Break out of the loop if no more output
|
|
|
|
|
match = url_pattern.search(line)
|
|
|
|
|
if match:
|
|
|
|
|
found_url = True
|
|
|
|
|
remote_url = match.group(0)
|
|
|
|
|
server_url = remote_url
|
|
|
|
|
print(
|
|
|
|
|
f"\nYour server is being hosted at the following URL: {remote_url}"
|
|
|
|
|
)
|
|
|
|
|
break # Exit the loop once the URL is found
|
|
|
|
|
if domain:
|
|
|
|
|
listener = ngrok.forward(f"{server_host}:{server_port}", authtoken_from_env=True, domain=domain)
|
|
|
|
|
else:
|
|
|
|
|
listener = ngrok.forward(f"{server_host}:{server_port}", authtoken_from_env=True)
|
|
|
|
|
|
|
|
|
|
if not found_url:
|
|
|
|
|
print(
|
|
|
|
|
"Failed to extract the ngrok tunnel URL. Please check ngrok's output for details."
|
|
|
|
|
)
|
|
|
|
|
listener_url = listener.url()
|
|
|
|
|
|
|
|
|
|
if server_url and qr:
|
|
|
|
|
text = pyqrcode.create(remote_url)
|
|
|
|
|
print(f"Ingress established at: {listener_url}");
|
|
|
|
|
if listener_url and qr:
|
|
|
|
|
text = pyqrcode.create(listener_url)
|
|
|
|
|
print(text.terminal(quiet_zone=1))
|
|
|
|
|
|
|
|
|
|
return server_url
|
|
|
|
|
return listener_url
|
|
|
|
|
|
Ben Xu
5 months ago