# Wizard Kit: Functions - PNG graph for I/O Benchmark

import base64
import Gnuplot
import math

from functions.common import *

# Functions
def export_png_graph(name, dev):
  """Exports PNG graph using gnuplot, returns file path as str."""
  max_rate = max(TESTS['iobenchmark']['Data'][name]['Read Rates'])
  max_rate /= (1024**2)
  max_rate = max(800, max_rate)
  out_path = '{}/iobenchmark-{}.png'.format(global_vars['LogDir'], name)
  plot_data = '{}/iobenchmark-{}-raw.log'.format(global_vars['LogDir'], name)

  # Adjust Y-axis range to either 1000 or roughly max rate + 200
  ## Round up to the nearest 100 and then add 200
  y_range = (math.ceil(max_rate/100)*100) + 200

  # Run gnuplot commands
  g = Gnuplot.Gnuplot()
  g('reset')
  g('set output "{}"'.format(out_path))
  g('set terminal png large size 660,300 truecolor font "Noto Sans,11"')
  g('set title "I/O Benchmark"')
  g('set yrange [0:{}]'.format(y_range))
  g('set style data lines')
  g('plot "{data}" title "{size} ({tran}) {model} {serial}"'.format(
    data=plot_data,
    size=dev['lsblk'].get('size', '???b'),
    tran=dev['lsblk'].get('tran', '???'),
    model=dev['lsblk'].get('model', 'Unknown Model'),
    serial=dev['lsblk'].get('serial', 'Unknown Serial'),
    ))

  # Cleanup
  g.close()
  del(g)

  return out_path

def upload_to_imgur(image_path):
  """Upload image to Imgur and return image url as str."""
  image_data = None
  image_link = None

  # Bail early
  if not image_path:
    raise GenericError

  # Read image file and convert to base64 then convert to str
  with open(image_path, 'rb') as f:
    image_data = base64.b64encode(f.read()).decode()

  # POST image
  url = "https://api.imgur.com/3/image"
  boundary = '----WebKitFormBoundary7MA4YWxkTrZu0gW'
  payload = ('--{boundary}\r\nContent-Disposition: form-data; '
             'name="image"\r\n\r\n{data}\r\n--{boundary}--')
  payload = payload.format(boundary=boundary, data=image_data)
  headers = {
    'content-type': 'multipart/form-data; boundary={}'.format(boundary),
    'Authorization': 'Client-ID {}'.format(IMGUR_CLIENT_ID),
    }
  response = requests.request("POST", url, data=payload, headers=headers)

  # Return image link
  if response.ok:
    json_data = json.loads(response.text)
    image_link = json_data['data']['link']
  return image_link

def upload_to_nextcloud(image_path, ticket_number, dev_name):
  """Upload image to Nextcloud server and return folder url as str."""
  image_data = None

  # Bail early
  if not image_path:
    raise GenericError

  # Read image file and convert to base64
  with open(image_path, 'rb') as f:
    image_data = f.read()

  # PUT image
  url = '{base_url}/{ticket_number}_iobenchmark_{dev_name}_{date}.png'.format(
    base_url=BENCHMARK_SERVER['Url'],
    ticket_number=ticket_number,
    dev_name=dev_name,
    date=global_vars.get('Date-Time', 'Unknown Date-Time'))
  requests.put(
    url,
    data=image_data,
    headers = {'X-Requested-With': 'XMLHttpRequest'},
    auth = (BENCHMARK_SERVER['User'], BENCHMARK_SERVER['Pass']))

  # Return folder link
  return BENCHMARK_SERVER['Short Url']

if __name__ == '__main__':
  print("This file is not meant to be called directly.")

# vim: sts=2 sw=2 ts=2