#!/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_LogFileToS3_$(date '+%Y%m%d_%H%M%S').log"

# -------------------------------------------------------------------
# Lock file to prevent overlapping cron runs
# ------------------------------------------------------------------- 
LOCKFILE="/tmp/SyncNRLogFileToS3_${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/"
MINUTES=15

# Option 1: auto-detect device IDs
AUTO_DETECT_DEVICES=true

# Option 2: manually configure device IDs
MANUAL_DEVICE_IDS=("csblrlabsvtcc9ec132-1132" "csblrlabsvtcc9ec132-1132-test")


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


# -------------------------------------------------------------------
# Determine device IDs
# -------------------------------------------------------------------
DEVICE_IDS=()
if [ "$AUTO_DETECT_DEVICES" = true ]; then
    log "Auto-detecting device IDs under $BASE_DIR/$OPERATOR"
    while IFS= read -r dir; do
        DEVICE_IDS+=("$(basename "$dir")")
    done < <(find "$BASE_DIR/$OPERATOR" -mindepth 1 -maxdepth 1 -type d 2>/dev/null)
else
    DEVICE_IDS=("${MANUAL_DEVICE_IDS[@]}")
    log "Using manually configured device IDs: ${DEVICE_IDS[*]}"
fi

# -------------------------------------------------------------------
# Iterate over devices
# -------------------------------------------------------------------
for DEV in "${DEVICE_IDS[@]}"; do
    FILES_PATH="$BASE_DIR/$OPERATOR/$DEV/"

    if [ ! -d "$FILES_PATH" ]; then
        log "WARNING: files directory does not exist: $FILES_PATH\n"
        continue
    fi
	
	log "\n"
	log "-------------------------------------------"
    log "Scanning device directory: $FILES_PATH\n"

    # Fetch ONLY files directly under files/ (no subfolders)
    FILES=$(find "$FILES_PATH" -maxdepth 1 -type f -mmin -"$MINUTES" 2>/dev/null || true)

    if [ -z "$FILES" ]; then
        log "No files found in $FILES_PATH updated in last $MINUTES minutes"
        continue
    fi

    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/logs/${DEV}/$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
    done
	log "-------------------------------------------"
done

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