#!/bin/bash
export PATH=/bin:/sbin:/usr/bin:/usr/sbin:$PATH

BIN=$(basename $0)
DEV=$1
UUID=$2
LOG_FILE="/var/log/usb_crypt.log"

UCRYPT_PATH="/opt/ucrypt"

function adddate {
    while IFS= read -r line; do
        echo "$(date +"%Y-%m-%d %H:%m:%S"): $line"
    done
}

if [[ $# -ne 2 ]]; then 
	echo "Usage: $BIN <device> <UUID> " | adddate >>$LOG_FILE 2>&1
	echo "FAILED" >"$UCRYPT_PATH/usb_crypt_shell"
	exit 2
fi

if [[ ! -d "$UCRYPT_PATH" ]]; then 
	mkdir -p $UCRYPT_PATH
fi

ucryptsetup setFlag $DEV
if [[ $? -eq 0 ]]; then
	log_ucrypt $DEV add 1 
	echo "Confirm format ucrypt $DEV success" | adddate >>$LOG_FILE 2>&1
	echo "SUCCESS" >"$UCRYPT_PATH/usb_crypt_shell"
else
	echo "Confirm format ucrypt $DEV failed" | adddate >>$LOG_FILE 2>&1
	echo "FAILED" >"$UCRYPT_PATH/usb_crypt_shell"
	exit 4
fi

