#!/usr/bin/bash

set -e

# -------------------------------------------------------------------
# Logging setup
# -------------------------------------------------------------------
SCRIPT_DIR="$(cd "$(dirname "$0")" && pwd)"
log() {
    echo -e "$(date '+%Y-%m-%d %H:%M:%S')  $1" | tee -a "$LOG_FILE"
}

# -------------------------------------------------------------------
# Operator input validation
# -------------------------------------------------------------------
ALLOWED_OPERATORS=( ATT TMO FWV CNX NXT SHACC 3UK EE VODAFONE VMO2 )

if [ $# -ne 1 ]; then
    echo "Usage: $0 <operator>"
    exit 1
fi

OPERATOR=$1

# Convert to lowercase
OPERATOR="${OPERATOR,,}"

if ! printf '%s\n' "${ALLOWED_OPERATORS[@],,}" | grep -qx "$OPERATOR"; then
  echo "ERROR: Invalid operator '$1'"
  echo "Allowed operators: ${ALLOWED_OPERATORS[*]}"
  exit 1
fi

LOG_FILE="$SCRIPT_DIR/${OPERATOR}_NR_PMFileToS3_$(date '+%Y%m%d_%H%M%S').log"

# -------------------------------------------------------------------
# Lock file to prevent overlapping cron runs
# -------------------------------------------------------------------
LOCKFILE="/tmp/SyncNROMFileToS3_${OPERATOR}.lock"
exec 200>$LOCKFILE
flock -n 200 || {
    echo "$(date '+%Y-%m-%d %H:%M:%S')  Another instance is running. Exiting." >> "$LOG_FILE"
    exit 1
}


# -------------------------------------------------------------------
# Input / configuration
# -------------------------------------------------------------------
BASE_DIR="/opt/device/P20_logs/"
PM_FOLDER="PERFORMANCE"
MINUTES=15

ACCOUNT_ID=$(/usr/local/bin/aws sts get-caller-identity --query "Account" --output text)
REGION="ap-southeast-1"
###########################################

BUCKET="${ACCOUNT_ID}-${REGION}-onecell-files"

now="$(date +"%T")"
log "Script started at $now"
echo -e "\nOperator: $OPERATOR" | tee -a $LOG_FILE
echo "Account ID: $ACCOUNT_ID" | tee -a $LOG_FILE
echo "Region:     $REGION"     | tee -a $LOG_FILE
echo -e "S3 bucket:  $BUCKET\n"   | tee -a $LOG_FILE

# -------------------------------------------------------------------
# Iterate per device
# -------------------------------------------------------------------
for DEVICE_PATH in "$BASE_DIR/$OPERATOR"/*; do

    if [ -d "$DEVICE_PATH" ]; then
		DEVICE_ID="$(basename "$DEVICE_PATH")"
	else
		log "Skipping path (not a directory or does not exist): $DEVICE_PATH"
		continue
	fi


    DEVICE_ID="$(basename "$DEVICE_PATH")"
    PM_PATH="$DEVICE_PATH/$PM_FOLDER"

    if [ ! -d "$PM_PATH" ]; then
		log "-------------------------------------------"
        log "Skipping device : $DEVICE_ID (no $PM_FOLDER folder)"
        continue
    fi

	log "-------------------------------------------"
    log "Scanning PM files under device: $DEVICE_ID"

    FILES=$(find "$PM_PATH" \
        -maxdepth 1 \
        -type f \
        -mmin -"$MINUTES" \
        2>/dev/null || true)

    if [ -z "$FILES" ]; then
        log "No PM files found for device $DEVICE_ID in last $MINUTES minutes"
        continue
    fi

    # -------------------------------------------------------------------
    # Process each file for this device
    # -------------------------------------------------------------------
    for FILE in $FILES; do

        BASENAME=$(basename "$FILE")

        DATE=$(echo "$BASENAME" | grep -oE '[0-9]{8}' | head -n1)
        if [ -z "$DATE" ]; then
            log "Skipping file (no date found): $FILE"
            continue
        fi

        S3_OPERATOR="$OPERATOR"
        case "$OPERATOR" in
          3uk|ee|vodafone|vmo2)
          log "operator is $OPERATOR"
          S3_OPERATOR="fwv/$OPERATOR"
          ;;
        esac

        DEST="s3://$BUCKET/onecell/$S3_OPERATOR/NR/PM/${DEVICE_ID}/$DATE/"
        #log "Copying: $FILE → $DEST"
		#
        #if /usr/local/bin/aws s3 cp "$FILE" "$DEST"; then
        #    log "File copied successfully: $FILE"
		#	 log "Deleting $FILE"
        #    if rm -f "$FILE"; then
        #        log "File deleted: $FILE"
        #    else
        #        log "WARNING: Unable to delete file: $FILE"
		#		 log "Skipping deletion and continuing..."
        #    fi
        #else
        #    log "ERROR: Failed to copy file to S3: $FILE"
        #fi

		log "Moving: $FILE → $DEST\n"
		/usr/local/bin/aws s3 mv "$FILE" "$DEST"

		if [ $? -eq 0 ]; then
			log "File Moved successfully: $FILE"
		else
			log "ERROR: File Upload Failed: $FILE"
		fi

        log "-------------------------------------------"
    done
done
log "\n"
log "-------------------------------------------"
log "All eligible NR PM files processed successfully."
log "Script Execution Completed"