Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Allow custom port specification for replay #761

Merged
merged 4 commits into from
Apr 6, 2022
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
13 changes: 12 additions & 1 deletion ipwb/__main__.py
Original file line number Diff line number Diff line change
Expand Up @@ -58,9 +58,14 @@ def check_args_replay(args):
print(f'Proxying to {args.proxy}')
proxy = args.proxy

port = replay.IPWBREPLAY_PORT
if hasattr(args, 'port') and args.port is not None:
machawk1 marked this conversation as resolved.
Show resolved Hide resolved
print(f'Using custom port {args.port} for replay.')
port = args.port

# TODO: add any other sub-arguments for replay here
if supplied_index_parameter:
replay.start(cdxj_file_path=args.index, proxy=proxy)
replay.start(cdxj_file_path=args.index, proxy=proxy, port=port)
else:
print('ERROR: An index file must be specified if not piping, e.g.,')
print(("> ipwb replay "
Expand Down Expand Up @@ -133,6 +138,12 @@ def check_args(args_in):
help='Proxy URL',
metavar='<host:port>',
nargs='?')
replay_parser.add_argument(
'-p', '--port',
help='Custom Port',
type=int,
default=util.IPWBREPLAY_PORT
)
replay_parser.set_defaults(func=check_args_replay,
onError=replay_parser.print_help)

Expand Down
11 changes: 7 additions & 4 deletions ipwb/replay.py
Original file line number Diff line number Diff line change
Expand Up @@ -1047,12 +1047,15 @@ def get_cdxj_line_binary_search(
return line_found


def start(cdxj_file_path, proxy=None):
def start(cdxj_file_path, proxy=None, port=IPWBREPLAY_PORT):
host_port = ipwb_utils.get_ipwb_replay_config()
app.proxy = proxy

# Retain port for subsequent runs
ipwb_utils.set_ipwb_replay_config(IPWBREPLAY_HOST, port)

if not host_port:
ipwb_utils.set_ipwb_replay_config(IPWBREPLAY_HOST, IPWBREPLAY_PORT)
host_port = (IPWBREPLAY_HOST, port)

# This will throw an exception if daemon is not available.
ipwb_utils.check_daemon_is_alive()
Expand All @@ -1062,9 +1065,9 @@ def start(cdxj_file_path, proxy=None):

try:
print((f'IPWB replay started on '
f'http://{IPWBREPLAY_HOST}:{IPWBREPLAY_PORT}'))
f'http://{host_port[0]}:{host_port[1]}'))

app.run(host='0.0.0.0', port=IPWBREPLAY_PORT)
app.run(host='0.0.0.0', port=host_port[1])
except gaierror:
print('Detected no active Internet connection.')
print('Overriding to use default IP and port configuration.')
Expand Down