#!/usr/bin/env python3
# -*-coding:UTF-8 -*

import os
import sys
import configparser
from TorSplashCrawler import TorSplashCrawler

tor_browser_agent = 'Mozilla/5.0 (Windows NT 6.1; rv:24.0) Gecko/20100101 Firefox/24.0'
default_crawler_options = {'html': 1, 'har': 1, 'png': 1, 'closespider_pagecount': 50}

if __name__ == '__main__':

    if len(sys.argv) != 7:
        print('usage:', 'tor_crawler.py', 'splash_url', 'type', 'url', 'domain', 'paste', 'super_father')
        exit(1)

    configfile = os.path.join(os.environ['AIL_BIN'], 'packages/config.cfg')
    if not os.path.exists(configfile):
        raise Exception('Unable to find the configuration file. \
                        Did you set environment variables? \
                        Or activate the virtualenv.')

    cfg = configparser.ConfigParser()
    cfg.read(configfile)

    splash_url = sys.argv[1]
    type = sys.argv[2]

    url = sys.argv[3]
    domain = sys.argv[4]
    paste = sys.argv[5]
    super_father = sys.argv[6]
    
    if crawler_options is None:
        crawler_options = default_crawler_options

    crawler_options['depth_limit'] = cfg.getint("Crawler", "crawler_depth_limit")
    crawler_options['user_agent'] = tor_browser_agent

    crawler = TorSplashCrawler(splash_url, crawler_options)
    crawler.crawl(type, crawler_options, url, domain, paste, super_father)