#!/bin/bash

LOGFILE=/var/log/kern.log
MESSAGE='usb.*1\-1\.XXX.*reset' #トリガーとなるメッセージ。XXXにUSBポート番号を。
USBDEVCMD=`which lsusb`
AWKCMD=`which awk`
GREPCMD=`which grep`
CATCMD=`which cat`
SUDOCMD=`which sudo`
TAILCMD=`which tail`
FINDCMD=`which find`
RESETCMD=`which usbdevicereset.sh`
USBADDR='XXXX:XXXX'  #USBデバイスのID
USBTTYDEV='/dev/ttyUSB0' #期待するデバイスファイル
CHECKFILE=/tmp/${0}.check

#check environment
if [ "x$SUDOCMD" == "x" \
    -o "x$CATCMD" == "x" \
    -o "x$GREPCMD" == "x" \
    -o "x$USBDEVCMD" == "x" \
    -o "x$RESETCMD" == "x" \
    -o "x$AWKCMD" == "x" \
    -o "x$TAILCMD" == "x" \
    -o "x$FINDCMD" == "x" ];
then
    echo "some comand[s] was/were not found"
    exit 1
fi

if [ ! -f $LOGFILE ];
then
    echo "log file was not found"
    echo "log file is [${LOGFILE}]"
    exit 1
fi

#functions
checkaddress () {
    $USBDEVCMD |\
        $GREPCMD $USBADDR >/dev/null 2>&1 && \
        echo -n "OK" || echo -n "!"
}

watchcheckfile () {
    if [ ! -f $CHECKFILE ];
    then
        echo -n "OK"
    fi
    $FINDCMD `dirname $CHECKFILE` -type f -mtime +1 -print 2>/dev/null |\
        $GREPCMD `basename $CHECKFILE` >/dev/null 2>&1 && \
        echo -n "OK" || echo -n "!"
}

watchreset () {
    $CATCMD $LOGFILE |\
        $GREPCMD "${MESSAGE}" >/dev/null 2>&1  && \
        echo -n "OK" || echo -n "!"
}

usbreset () {
    $RESETCMD $USBADDR |\
        $GREPCMD "Reset successful" >/dev/null 2>&1 && \
        $CATCMD $LOGFILE |\
        $TAIL -1 |\
        $GREPCMD $USBTTYDEV >/dev/null 2>&1 && \
        echo -n "OK" || echo -n "!"
}

#main

#check usb address
ret=`checkaddress`
if [ "x$ret" == "x!" -o "x$ret" == "x" ];
then
    echo "usb device was not found"
    exit 1
fi

#watch checkfile
ret=`watchcheckfile`
if [ "x$ret" == "x!" -o "x$ret" == "x" ];
then
    exit 0
fi

#watch usbdevice reset
ret=`watchreset`
if [ "x$ret" == "x!" -o "x$ret" == "x" ];
then
    exit 0
fi

#reset usbdevice
ret=`usbreset`
if [ "x$ret" == "x!" -o "x$ret" == "x" ];
then
    ret=`usbreset`
    if [ "x$ret" == "x!" -o "x$ret" == "x" ];
    then
        echo "FAIL!!!"
        exit 1
    else
        touch $CHECKFILE
        exit 0
    fi
fi