2014-02-14 15:18:10 -08:00
|
|
|
# vim: set sw=4 et:
|
|
|
|
|
|
2014-02-13 01:00:39 -05:00
|
|
|
from json import dumps, load
|
|
|
|
|
from time import sleep
|
2014-03-05 23:19:09 -05:00
|
|
|
from itertools import chain
|
2014-02-13 01:00:39 -05:00
|
|
|
import os, re
|
2014-02-14 15:18:10 -08:00
|
|
|
import logging
|
2014-02-13 01:00:39 -05:00
|
|
|
|
2014-03-06 18:25:35 -08:00
|
|
|
logger = logging.getLogger('behaviors')
|
2014-03-05 23:19:09 -05:00
|
|
|
behaviors_directory = os.path.sep.join(__file__.split(os.path.sep)[:-1] + ['behaviors.d'])
|
2014-03-07 11:29:43 -08:00
|
|
|
behavior_files = chain(*[[os.path.join(dir, file) for file in files if file.endswith('.js')] for dir, dirs, files in os.walk(behaviors_directory)])
|
2014-03-05 23:19:09 -05:00
|
|
|
behaviors = []
|
|
|
|
|
for file_name in behavior_files:
|
2014-03-06 18:25:35 -08:00
|
|
|
logger.debug("reading behavior file {}".format(file_name))
|
2014-03-05 23:19:09 -05:00
|
|
|
lines = open(file_name).readlines()
|
|
|
|
|
pattern, script = lines[0][2:].strip(), ''.join(lines[1:])
|
2014-03-06 18:25:35 -08:00
|
|
|
behaviors.append({'url_regex': pattern, 'script': script, 'file': file_name})
|
|
|
|
|
logger.info("will run behaviors from {} to urls matching {}".format(file_name, pattern))
|
2014-03-05 23:19:09 -05:00
|
|
|
|
2014-02-14 15:18:10 -08:00
|
|
|
def execute(url, websock, command_id):
|
2014-03-05 23:19:09 -05:00
|
|
|
for behavior in behaviors:
|
2014-03-06 18:25:35 -08:00
|
|
|
if re.match(behavior['url_regex'], url):
|
2014-03-05 23:19:09 -05:00
|
|
|
msg = dumps(dict(method="Runtime.evaluate", params={"expression": behavior['script']}, id=next(command_id)))
|
|
|
|
|
logger.debug('sending message to {}: {}'.format(websock, msg))
|
|
|
|
|
websock.send(msg)
|
