#!/usr/bin/env python
# -*- encoding: utf-8 -*-
from libmproxy import controller, proxy
import os, sys, re, datetime, json

class InterceptingMaster (controller.Master):
  def __init__ (self, server):
    controller.Master.__init__(self, server)

  def run (self):
    while True:
      try:
        controller.Master.run(self)
      except KeyboardInterrupt:
        print 'KeyboardInterrupt received. Shutting down'
        self.shutdown()
        sys.exit(0)
      except Exception:
        print 'Exception catched. Intercepting proxy restarted'
        pass

  def handle_request (self, msg):
    timestamp = datetime.datetime.today().strftime("%Y/%m/%d %H:%M:%S")
    client_ip = msg.client_conn.address[0]
    request_url = '%s://%s%s' % (msg.scheme, .msg.host, msg.path)
    print '[%s %s] %s %s' % (timestamp, client_ip, msg.method, request_url)

    msg.reply()

  def handle_response (self, msg):
    # simple substitution for https://example.org/api/users/:id.json
    if "example.org" in msg.request.host:
      regex = re.compile("/api/users/(\d+).json")
      match = regex.search(msg.request.path)
      if match and msg.content:
        c = msg.replace('"private_data_accessible":false', '"private_data_accessible":true')
        if c > 0: print '[Example.org] Private info of user #%s revealed' % (match.groups()[0])

    # JSON manipulation for https://example.com/api/v2
    if ("example.com" in msg.request.host) and ("action=user_profile" in msg.request.content):
      msg.decode() # need to decode the message first
      data = json.loads(msg.content) # parse JSON with decompressed content
      data["access_granted"] = true
      msg.content = json.dumps(data) # write back our changes
      print '[Example.com] Access granted of user profile #%s' % data["id"]

    # Response inspection for https://example.net
    if "example.net" in msg.request.host:
      data = msg.get_decoded_content() # read decompressed content without modifying msg
      print '[Example.net] Respones: %s' % data

    msg.reply()


def main (argv):
  config = proxy.ProxyConfig(
    cacert = os.path.expanduser("./mitmproxy.pem"),
  )
  server = proxy.ProxyServer(config, 8080)
  print 'Intercepting Proxy listening on 8080'
  m = InterceptingMaster(server)
  m.run()

if __name__ == '__main__':
  main(sys.argv)