import requests
import sys


def waybackurls(host, with_subs):
    if with_subs:
        url = 'http://web.archive.org/cdx/search/cdx?url=*.%s/*&output=json&fl=original' % host
    else:
        url = 'http://web.archive.org/cdx/search/cdx?url=%s/*&output=json&fl=original' % host
    r = requests.get(url)
    results = r.json()
    urls = set()
    for i in results:
        urls.add(i[0])
    return urls


if __name__ == '__main__':
    argc = len(sys.argv)
    if argc < 2:
        print('Usage:\n\tpython3 waybackurls.py <url> <include_subdomains:optional>')
        sys.exit()

    host = sys.argv[1]
    with_subs = False
    if argc > 3:
        with_subs = True

    urls = waybackurls(host, with_subs)
    if urls:
        filename = '%s-waybackurls.txt' % host
        with open(filename, 'w') as f:
            f.write('\n'.join(urls))
        print('[*] Saved results to %s' % filename)
    else:
        print('[-] Found nothing')