#FileGetter
import ssh
import re
#Function to delete specified regix matches in a string
def replace(value,reg):
  return re.sub(reg, r'', value)
server = ssh.Connection(host='10.30.160.81', username='report_viewer_smtbusa', password='4S9xSfVFvzsyC1CXHWpR')
relations=open(".\Relationships",'r')
for relation in relations:
    relation=replace(relation,r'\n')
    temp=relation.split(":")
    creatorName=temp[0]
    folderName=temp[1]
    command='find /store/reporting/reports/{0}/reports* | sort -n -r | grep -P ".+{1}.+\.csv"| head -1'.format(creatorName, folderName)
    result = server.execute(command)[0].decode("utf-8")
    result=replace(result,r'\n')
    print(temp[2])
    server.get(result,f"./Reports/"+temp[2]+".csv")

#LogGenerator
from docx import Document
from docx.shared import Inches
import matplotlib.pyplot as plt
from LogGrapghGen import newest,barCreate, FolderFiles,replace
import os
from pathlib import Path
import re
import datetime

def BarGrapghCreator(filepath,description, document,heading=1):
    rows=[]
    label=0
    value=10
    fd=open(filepath,'r')
    for row in fd:
        row=row[:-1]
        rowArray=[]
        if row[0] == '"':
            rowArray=row.split('","')
            rowArray[0]=rowArray[0][1:]
            rowArray[-1]=rowArray[-1][:-1]
        else:
            rowArray=row.split(',')
        rows.append(rowArray)
    if len(rows)<2:
        return False
    p = Path(filepath)
    imagePath,catagory,large=barCreate(rows,p.name[:-4],label,value)
    document.add_heading(catagory, level=heading)
    if large:
        document.add_picture(imagePath, width=Inches(7), height=Inches(5))
    else:
        document.add_picture(imagePath, width=Inches(7), height=Inches(3.5))
    document.add_paragraph(description)
    return True

def TableCreator(filepath,description, document,heading=1):
    rows=[]
    fd=open(filepath,'r')
    for row in fd:
        row=row[:-1]
        rowArray=[]
        if row[0] == '"':
            rowArray=row.split('","')
            rowArray[0]=rowArray[0][1:]
            rowArray[-1]=rowArray[-1][:-1]
        else:
            rowArray=row.split(',')
        rows.append(rowArray)
    p = Path(filepath)
    if len(rows)<2:
        return False
    document.add_heading(p.name[:-4], level=heading)
    table = document.add_table(1, len(rows[0]))
    table.style = 'GridTable4-Accent1'
    heading_cells = table.rows[0].cells
    for index in range(len(rows[0])):
        heading_cells[index].text=rows[0][index]
    for arrayrow in rows[1:]:
        cells = table.add_row().cells
        for index in range(len(arrayrow)):
            cells[index].text=arrayrow[index]
    document.add_paragraph(description)
    return True

def USBTableCreator(filepath,description, document,heading=1):
    rows=[]
    names=[]
    fd=open(filepath,'r', encoding="utf8")
    rows.append(["Username", "PC Name", "Time"])
    for row in fd:
        row=row[:-1]
        rowArray=[]
        rowArray=row.split(',')
        if "USB Device" in rowArray[15]:
            if rowArray[21] in names:
                continue
            rows.append([rowArray[21],rowArray[4],rowArray[11]])
            names.append(rowArray[21])
    p = Path(filepath)
    if len(rows)<2:
        return False
    document.add_heading(p.name[:-4], level=heading)
    table = document.add_table(1, len(rows[0]))
    table.style = 'GridTable4-Accent1'
    heading_cells = table.rows[0].cells
    for index in range(len(rows[0])):
        heading_cells[index].text=rows[0][index]
    for arrayrow in rows[1:]:
        cells = table.add_row().cells
        for index in range(len(arrayrow)):
            cells[index].text=arrayrow[index]
    document.add_paragraph(description)
    return True

def main():
    document = Document('LOG-MANAGEMENT-REPORT-DAILY-TEMPLETE.docx')
    #document.add_heading('Siem Report', 0)
    current_section = document.sections[-1]
    current_section.left_margin = Inches(1)
    current_section.right_margin = Inches(1)
    #p = document.add_paragraph('')
    folderPath=".\Reports"
    folderpath = os.path.abspath(folderPath)
    categoriesFile=open(".\Categories",'r')
    for cat in categoriesFile:
        tabs=len(re.findall(r'\t', cat))
        cat=replace(cat,r'\t')
        cat=replace(cat,r'\n')
        export=False
        for filepath in filter(lambda x: cat in x, FolderFiles(folderpath)):
            print(cat)
            export=True
            if filepath == "":
                continue
            description=""
            if "Top 10" in cat:
                if not BarGrapghCreator(filepath,description,document,tabs+2):
                    print("Invalid Input for "+filepath+" will skip genenration of graph")
                    continue
            elif "USB" in cat:
                if not USBTableCreator(filepath,description,document,tabs+2):
                    print("There is no data in "+filepath+" will skip genenration of table")
                    continue
            else:
                if not TableCreator(filepath,description,document,tabs+2):
                    print("There is no data in "+filepath+" will skip genenration of table")
                    continue
        if not export and tabs==0:
            document.add_heading(cat, level=1+1)
    filename='Siem Report '+datetime.datetime.today().strftime('%Y-%m-%d')+'.docx'
    document.save(filename)
    print(filename)


if __name__ == '__main__':
    main()

#LogGraphGen
import os
import matplotlib.pyplot as plt; plt.rcdefaults()
import numpy as np
import matplotlib.pyplot as plt
from itertools import cycle, islice
import re
import operator


def replace(value,reg):
  return re.sub(reg, r'', value)

def newest(path):
    files = os.listdir(path)
    paths = []
    for basename in files:
        if basename.endswith(".csv"):
            paths = [os.path.join(path, basename)]
    if len(paths) == 0:
        return ""
    else:
        return max(paths, key=os.path.getmtime)

def FolderFiles(path):
    files = os.listdir(path)
    paths = []
    for basename in files:
        if basename.endswith(".csv"):
            paths.append(os.path.join(path, basename))
    if len(paths) == 0:
        return ""
    else:
        return paths

def label_bar(ax, bars, text_format, is_inside=True, **kwargs):
    """
    Attach a text label to each bar displaying its y value
    """
    max_y_value = max(bar.get_height() for bar in bars)
    if is_inside:
        distance = max_y_value * 0.05
    else:
        distance = max_y_value * 0.01
    for bar in bars:
        text = text_format.format(bar.get_height())
        text_x = bar.get_x() + bar.get_width() / 2
        if is_inside:
            text_y = bar.get_height() - distance
        else:
            text_y = bar.get_height() + distance
        ax.text(text_x, text_y, text, ha='center', va='bottom', **kwargs)

def barCreate(rows, category, labels, data):
    x_labels=[]
    values=[]
    x_dic = {}
    big=False
    for index in range(1,len(rows)):
        x_label=rows[index][labels]
        if len(x_label)>30:
            big=True
        value=rows[index][data].split(",")
        valuenum="".join(value)
        if x_label in x_dic.keys():
            x_dic[rows[index][labels]]=x_dic[rows[index][labels]] + int(valuenum)
        else:
            x_dic[rows[index][labels]]=int(valuenum)
    sortedDic=sorted(x_dic.items(), key=operator.itemgetter(1),reverse=True)
    for item in sortedDic:
        x_labels.append(item[0])
        values.append(item[1])
    objects = tuple(x_labels)
    y_pos = np.arange(len(objects))
    if big:
        plt.figure(figsize=(15, 11))
    else:
        plt.figure(figsize=(11, 5.5))
    my_colors = [(x/10.0, x/20.0, 0.75) for x in range(len(y_pos))]
    cmap = plt.cm.hsv
    bars=plt.bar(y_pos, values, align='center', alpha=0.5, color=cmap(np.linspace(0, 1, len(objects)+1)), label=objects)
    plt.xticks(y_pos, objects, rotation='vertical')
    try:
        plt.tight_layout()
    except:
        pass
    plt.ylabel(rows[0][data])
    plt.title(category)
    value_format = "{}"
    label_bar(plt, bars, value_format, is_inside=False, fontweight="bold")
    filepath = ".\images\\"+category+".png"
    if big:
        if "URL" in category:
            pass
        else:
            plt.legend(bars, objects)
    plt.savefig(filepath,dpi=300, format='png', bbox_inches='tight')
    #plt.show()
    plt.close()
    return filepath,category,big