QFQ

import subprocess

import argparse

import os

import sys

import socket

def list_folders_and_files(path, max_depth):

    try:

        if os.path.exists(os.path.join(path, 'fileadmin')) and os.path.isdir(os.path.join(path, 'fileadmin')):

            items = []

            for root, dirs, files in os.walk(os.path.join(path, 'fileadmin')):

                depth = root[len(os.path.join(path, 'fileadmin')) + len(os.path.sep):].count(os.path.sep)

                if depth <= max_depth:

                    for dir in dirs:

                        if depth + 1 <= max_depth:  # Check the next depth level

                            items.append(os.path.join(root, dir) + '/')

                    for file in files:

                        if depth <= max_depth:  # Check the current depth level

                            items.append(os.path.join(root, file))

            for idx, item in enumerate(items, start=1):

                print(f"{idx}. {item}")

            return items

        else:

            print("Error: 'fileadmin' directory not found")

    except OSError:

        print("Error: Invalid path")

    return []

def get_user_selection(items):

    selection = input("Select paths with enter comma-separated list of numbers: ")

    selected_indices = [int(idx) for idx in selection.split(",") if idx.isdigit() and 1 <= int(idx) <= len(items)]

    selected_paths = [items[idx - 1] for idx in selected_indices]

    return selected_paths

def rsync_and_check_sync(selected_paths, project_dir_name, target_hostname):

    syncable_paths = []

    for idx, path in enumerate(selected_paths, start=1):

        rsync_command = [

            'rsync',

            '-av',

            '--dry-run',

            os.path.join(path),

            f'{target_hostname}:{path}'

        ]

        result = subprocess.run(rsync_command, stdout=subprocess.PIPE, stderr=subprocess.PIPE)

        stdout = result.stdout.decode('utf-8')

        stderr = result.stderr.decode('utf-8')

        if "to-chk" in stderr:

            print(f"{idx}. {path} {target_hostname}:{path} >> To sync")

            print(stdout)

            syncable_paths.append((idx, rsync_command))

        else:

            print(f"{idx}. {path} {target_hostname}:{path} >> Nothing to sync")

    return syncable_paths

def execute_selected_syncs(syncable_paths):

    input_str = input("Enter comma-separated list of numbers to execute: ")

    selected_indices = [int(idx) for idx in input_str.split(",") if idx.isdigit()]

    for idx, rsync_command in syncable_paths:

        if idx in selected_indices:

            print("Rsync command to execute:")

            print(" ".join(rsync_command))

            confirm = input("Do you want to execute this sync? (y/N): ").strip().lower()

            if confirm == "y":

                rsync_command.remove('--dry-run')

                subprocess.run(rsync_command)

            else:

                print("Sync operation cancelled.")

if __name__ == "__main__":

    if len(sys.argv) < 2:

        print("Usage: python3 script.py project_dir_name")

        sys.exit(1)

    project_dir_name = sys.argv[1]

    #max_depth = int(sys.argv[2])

    max_depth = 1

    target_hostname = 'w22a'

    path = os.path.join('/var/www/html/', project_dir_name)

    hostname = socket.gethostname()

    print("Welcome to the migration live guide\n---------------------------------")

    print("Project: " + path + "\nSource Host: " + hostname + "\nTarget Host: " + target_hostname)

    print("--------------------------------------------------------\nStep 1: List of all files and folders from fileadmin (Level 1)")

    # Step one: list all folders and files in fileadmin

    items = list_folders_and_files(path, max_depth)

    print("--------------------------------------------------------\nStep 2: Executing rsync with dry-run")

    # Step two: allow user to input selection

    if items:

        selected_paths = get_user_selection(items)

        print("Selected paths:")

        for path in selected_paths:

            print(path)

        # Step four: perform rsync and check synchronization

        syncable_paths = rsync_and_check_sync(selected_paths, project_dir_name, target_hostname)

        print("--------------------------------------------------------\nStep 3: Executing rsync final")

        if len(syncable_paths) == 0:

            print("Nothing to sync.")

        else:

            execute_selected_syncs(syncable_paths) 

    print("--------------------------------------------------------\nStep 4: Starting table data transfer")

    # Step four in seperate script

    subprocess.run(['python3', 'table_data_transfer.py'])