"""
This script enumerates all occurrences of a text, based on a regex.
The following search example returns all soapaction names found in 
    the history, be it in the header or body.

You need to edit the 'regex' variable to whatever suits your need
"""

from org.parosproxy.paros.model import Model;
import re;
from org.parosproxy.paros.view import AbstractFrame;
from org.zaproxy.zap.utils import ZapTextArea;

sessionId = model.getSession();
tbHist = model.getDb().getTableHistory();

""" Change this variable to match the expression you are looking for """
regex='soapactio.+';

def search(msg):
  """ Searching inside the request """
  match = re.search(regex, msg.getRequestHeader().toString(), re.IGNORECASE);
  if(match):
    for header in msg.getRequestHeader().toString().split("\n"):
      match = re.search(regex, header, re.IGNORECASE);
      if(match):
        yield header;
  match = re.search(regex, msg.getRequestBody().toString(), re.IGNORECASE);
  if(match):
    for word in msg.getRequestBody().toString().split():
      match = re.search(regex, word, re.IGNORECASE);
      if(match):
        yield word;
  """ Searching inside the response """
  match = re.search(regex, msg.getResponseHeader().toString(), re.IGNORECASE);
  if(match):
    for header in msg.getResponseHeader().toString().split("\n"):
      match = re.search(regex, header, re.IGNORECASE);
      if(match):
        yield header;
  match = re.search(regex, msg.getResponseBody().toString(), re.IGNORECASE);
  if(match):
    for word in msg.getResponseBody().toString().split():
      match = re.search(regex, word, re.IGNORECASE);
      if(match):
        yield word;


class OutputWindow (AbstractFrame):
    def __init__(self, text):
        self.setAlwaysOnTop(False);
        self.setSize(700, 500);
        ta = ZapTextArea(text);
        self.setContentPane(ta);
        self.setVisible(True);
    

if (tbHist != None):
  final = list();
  print("Searching History...");
  for index in tbHist.getHistoryIds(sessionId.getSessionId()):
    try:
      msg = tbHist.read(index).getHttpMessage();
      results = search(msg);
      for item in results:
        if(item not in final):
          final.append(item);
    except StopIteration:
      pass;
result = '';
for item in final:
  result+=item+"\n";
output = OutputWindow(result);