# Copyright (c) 2017 Adafruit Industries Author: Tony DiCola & James DeVito
#
# Permission is hereby granted, free of charge, to any person obtaining a copy
# of this software and associated documentation files (the "Software"), to deal
# in the Software without restriction, including without limitation the rights
# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
# copies of the Software, and to permit persons to whom the Software is
# furnished to do so, subject to the following conditions:
#
# The above copyright notice and this permission notice shall be included in
# all copies or substantial portions of the Software.
#
# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
# THE SOFTWARE.
import time

import Adafruit_GPIO.SPI as SPI
import Adafruit_SSD1306

from PIL import Image
from PIL import ImageDraw
from PIL import ImageFont

import subprocess

# Raspberry Pi pin configuration:
RST = None     # on the PiOLED this pin isnt used
# Note the following are only used with SPI:
DC = 23
SPI_PORT = 0
SPI_DEVICE = 0

# Beaglebone Black pin configuration:
# RST = 'P9_12'
# Note the following are only used with SPI:
# DC = 'P9_15'
# SPI_PORT = 1
# SPI_DEVICE = 0

# 128x32 display with hardware I2C:
#disp = Adafruit_SSD1306.SSD1306_128_32(rst=RST)

# 128x64 display with hardware I2C:
disp = Adafruit_SSD1306.SSD1306_128_64(rst=RST)

# Note you can change the I2C address by passing an i2c_address parameter like:
# disp = Adafruit_SSD1306.SSD1306_128_64(rst=RST, i2c_address=0x3C)

# Alternatively you can specify an explicit I2C bus number, for example
# with the 128x32 display you would use:
# disp = Adafruit_SSD1306.SSD1306_128_32(rst=RST, i2c_bus=2)

# 128x32 display with hardware SPI:
# disp = Adafruit_SSD1306.SSD1306_128_32(rst=RST, dc=DC, spi=SPI.SpiDev(SPI_PORT, SPI_DEVICE, max_speed_hz=8000000))

# 128x64 display with hardware SPI:
# disp = Adafruit_SSD1306.SSD1306_128_64(rst=RST, dc=DC, spi=SPI.SpiDev(SPI_PORT, SPI_DEVICE, max_speed_hz=8000000))

# Alternatively you can specify a software SPI implementation by providing
# digital GPIO pin numbers for all the required display pins.  For example
# on a Raspberry Pi with the 128x32 display you might use:
# disp = Adafruit_SSD1306.SSD1306_128_32(rst=RST, dc=DC, sclk=18, din=25, cs=22)

# Initialize library.
disp.begin()

# Clear display.
disp.clear()
disp.display()

# Create blank image for drawing.
# Make sure to create image with mode '1' for 1-bit color.
width = disp.width
height = disp.height
image = Image.new('1', (width, height))

# Get drawing object to draw on image.
draw = ImageDraw.Draw(image)

# Draw a black filled box to clear the image.
draw.rectangle((0,0,width,height), outline=0, fill=0)

# Draw some shapes.
# First define some constants to allow easy resizing of shapes.
padding = -2
top = padding
bottom = height-padding
# Move left to right keeping track of the current x position for drawing shapes.
x = 0


# Load default font.
font = ImageFont.load_default()

# Misaki Font, awesome 8x8 pixel Japanese font, can be downloaded from the following URL.
# $ wget http://www.geocities.jp/littlimi/arc/misaki/misaki_ttf_2015-04-10.zip
#font = ImageFont.truetype('/home/pi/font/misakifont/misaki_gothic.ttf', 8, encoding='unic')

# IPA True Type  font
fontS = ImageFont.truetype("fonts-japanese-gothic.ttf", 12)
fontL = ImageFont.truetype("fonts-japanese-gothic.ttf", 14)

# Alternatively load a TTF font.  Make sure the .ttf font file is in the same directory as the python script!
# Some other nice fonts to try: http://www.dafont.com/bitmap.php
# font = ImageFont.truetype('Minecraftia.ttf', 8)

while True:

    # Draw a black filled box to clear the image.
    draw.rectangle((0,0,width,height), outline=0, fill=0)

    # Shell scripts for system monitoring from here : https://unix.stackexchange.com/questions/119126/command-to-display-memory-usage-disk-usage-and-cpu-load

    #cmd = "hostname -I | cut -d\' \' -f1"
    #IP = subprocess.check_output(cmd, shell = True )

    cmd = "top -bn1 | grep load | awk '{printf \"%.2f%%\", $(NF-2)}'"
    CPU = subprocess.check_output(cmd, shell = True )

    #cmd = "free -m | awk 'NR==2{printf \"Mem: %s/%sMB %.2f%%\", $3,$2,$3*100/$2 }'"
    #MemUsage = subprocess.check_output(cmd, shell = True )

    #cmd = "df -h | awk '$NF==\"/\"{printf \"Disk: %d/%dGB %s\", $3,$2,$5}'"
    #Disk = subprocess.check_output(cmd, shell = True )

    #cmd = "tail -n100 /opt/NoraGateway/log/NoraGateway.log | awk 'match($0, /REF:.*\//) {print substr($0, RSTART, RLENGTH-1)}' | tail -n1"
    #LinkREF = subprocess.check_output(cmd, shell = True )
    #cmd = "grep \"Reflector Link Established\" /opt/NoraGateway/log/NoraGateway.status | awk 'match($0, /REF:.*\//) {print substr($0, RSTART, RLENGTH-1)}' | head -n1"
    #LinkREF = subprocess.check_output(cmd, shell = True )

    #cmd = "tail -n100 /opt/NoraGateway/log/NoraGateway.log | awk 'match($0, /RPT:.*/) {print substr($0, RSTART, RLENGTH)}' | tail -n1"
    #LinkRPT = subprocess.check_output(cmd, shell = True )
    #cmd = "grep \"Reflector Link Established\" /opt/NoraGateway/log/NoraGateway.status | awk 'match($0, /RPT:.*/) {print substr($0, RSTART, RLENGTH)}' | head -n1"
    #LinkRPT = subprocess.check_output(cmd, shell = True )

    #cmd = "date | awk '{print substr($0, RSTART, 17)}'"
    #NowDate = subprocess.check_output(cmd, shell = True )

    cmd = "grep \"NoraGateway v\" /tmp/NoraGateway.status | awk '{print substr($3, RSTART, 13)}'"
    NoraVer = subprocess.check_output(cmd, shell = True )

    #cmd = "head -n17 /tmp/NoraGateway.status | tail -n1 | awk 'split($0, a, \",\") {print substr(a[1], RSTART, 10)}'"
    cmd = "grep \"JapanTrust,Internal\" /tmp/NoraGateway.status | tail -n1 | awk 'split($0, a, \",\") {print substr(a[1], RSTART, 10)}'"
    RPT1 = subprocess.check_output(cmd, shell = True )

    #cmd = "head -n17 /tmp/NoraGateway.status | tail -n1 | awk 'split($0, b, \",\") {print substr(b[2], RSTART, 10)}'"
    cmd = "grep \"JapanTrust,Internal\" /tmp/NoraGateway.status | tail -n1 | awk 'split($0, b, \",\") {print substr(b[2], RSTART, 10)}'"
    RPT2 = subprocess.check_output(cmd, shell = True )

    cmd = "grep INCOMING /tmp/NoraGateway.status | awk 'split($0, a, \",\") {print substr(a[6], RSTART, 9)}' | head -n1"
    inMY1 = subprocess.check_output(cmd, shell = True )

    cmd = "grep INCOMING /tmp/NoraGateway.status | awk 'split($0, a, \",\") {print substr(a[3], RSTART, 9)}' | head -n1"
    inMY2 = subprocess.check_output(cmd, shell = True )

    cmd = "grep OUTGOING /tmp/NoraGateway.status | awk 'split($0, b, \",\") {print substr(b[6], RSTART, 9)}' | head -n1"
    outMY1 =  subprocess.check_output(cmd, shell = True )

    cmd = "grep OUTGOING /tmp/NoraGateway.status | awk 'split($0, b, \",\") {print substr(b[3], RSTART, 9)}' | head -n1"
    outMY2 =  subprocess.check_output(cmd, shell = True )

    # Write two lines of text.

    #                         "123456789012345678901"
    draw.text((x, top),       str(NoraVer),  font=fontS, fill=255)
    draw.text((96, top),      str(CPU),  font=fontS, fill=255)
    draw.text((x, top+13),    str(inMY1),                    font=fontL, fill=255)
    draw.text((57, top+13),   ">>"   + str(inMY2),            font=fontL, fill=255)
    draw.text((x, top+29),    str(outMY1),                   font=fontL, fill=255)
    draw.text((57, top+29),   ">>" + str(outMY2),             font=fontL, fill=255)
    draw.text((x, top+44),    "     RPT1: "  + str(RPT1),     font=fontS, fill=255)
    draw.text((x, top+56),    " REF/RPT2: "  + str(RPT2),     font=fontS, fill=255)
    #draw.text((x, top+56),    CCC,     font=font12, fill=255)

    # Display image.
    disp.image(image)
    disp.display()
    #time.sleep(.1)
    time.sleep(5)