from discord_webhook import DiscordWebhook, DiscordEmbed
from mextract import mextract

webhookURL = "https://discordapp.com/api/webhooks/740630755259973713/CNhxPBJ4C6UZ-m1DVXUQ54KeJQD3I9Di0sAwldmyjw-s7dMPUWplq7rhGXegpWGtKSve"
maxCharPerMessage = 1994
rescheduled = ""

with open ("/etc/bacula/scripts/webhooks/RoboticsMail.md", "r") as myfile:
    cleaned_stdout=myfile.read()

if "Rescheduled" in cleaned_stdout: # If the job was rescheduled
    cleaned_stdout = cleaned_stdout.split("Rescheduled", 1)[-1] # Remove the failed data
    rescheduled = "\nJob was rescheduled at least once."

endtime = mextract(cleaned_stdout, "End time:")
elapsedtime = mextract(cleaned_stdout, "Elapsed time:")
byteswritten = mextract(cleaned_stdout, "SD Bytes Written:")
usedvolumes = mextract(cleaned_stdout, "Volume name\(s\):")

if "Backup OK" in cleaned_stdout and "Backup OK -- with warnings" not in cleaned_stdout:
    truncated_stdout = ["""Job Backup-Robotics completed with status Backup OK, omitting full report.
All data from 1721's google drive was successfully written to tape on """ + endtime + """.
Backup took """ + elapsedtime + " And used " + byteswritten + " bytes."]
    if len(usedvolumes) > 1:
        truncated_stdout[0] = truncated_stdout[0] + ("\nWrote data to volume(s) " + usedvolumes + rescheduled)
else:
    # Discord limits each message to 2000 chars, if the message is longer than that, truncate it
    truncated_stdout = [cleaned_stdout[i:i+maxCharPerMessage] for i in range(0, len(cleaned_stdout), maxCharPerMessage)] 

print (truncated_stdout) # Print the truncated message


for message_part in truncated_stdout: # For every message part in the truncated output, run this loop
    message_part = "```" + message_part + "```" # Encapsulate the messsage
    print(message_part)
    webhook = DiscordWebhook(url=webhookURL, content=message_part) # Attach the message       
    response = webhook.execute() # Hit send