package fr.monlouyan.bakefile;

import java.io.IOException;
import java.nio.file.Files;
import java.nio.file.Paths;
import java.util.*;
import java.util.regex.Matcher;
import java.util.regex.Pattern;
import java.util.stream.Collectors;

/**
 * Parseur de fichier Bakefile.
 * Cette classe est responsable de l'analyse syntaxique du fichier Bakefile
 * pour extraire les règles de build, les dépendances et les commandes associées.
 * 
 * @author Moncef STITI, Yanis HAMOUDI, Louay DARDOURI
 * @version 1.0
 */
public class BakefileParser {
	
	/**
     * Nom du fichier Bakefile à parser (donc Bakefile...).
     */
    private String filename;

	/**
     * Regex pour détecter les targets et leurs dépendances.
     * Format : "nom1 nom2 nom3 : dépendance1 dépendance2"
     * La nouvelle regex gère plusieurs cibles séparées par des espaces
     */
    private static final Pattern TARGET_PATTERN = Pattern.compile("^((?:\"[^\"]*\"|[A-Za-z0-9_.\\/\\-\\$\\(\\)\\{\\}~]+)(?:\\s+(?:\"[^\"]*\"|[A-Za-z0-9_.\\/\\-\\$\\(\\)\\{\\}~]+))*)\\s*:\\s*([^#]*?)\\s*(?:#.*)?$");

	/**
     * Regex pour détecter les lignes de commande associées à une target.
     * Format : "    gcc -o program program.c" (ligne indentée)
     */
    private static final Pattern COMMAND_PATTERN = Pattern.compile("^\\t(.+)$");

	/**
     * Regex pour détecter les définitions de variables.
     * Format : "FLAGS = -ansi -pedantic"
     */
    private static final Pattern VARIABLE_PATTERN = Pattern.compile("^(\\w+)\\s*=\\s*([^#]*?)\\s*(?:#.*)?$");

	/**
     * Regex pour détecter les déclarations .PHONY
     * Format : ".PHONY: clean all"
     */
    private static final Pattern PHONY_PATTERN = Pattern.compile("^\\.PHONY:\\s*([^#]*?)\\s*(?:#.*)?$");

	/**
	 * Regex pour détecter les lignes de continuation.
	 * Format : "    gcc -o program program.c \"
	 */
    private static final Pattern CONTINUATION_PATTERN = Pattern.compile("^(.*)\\\\\\s*$");

	/**
	 * Regex pour détecter les références de variables.
	 * Format : "${VAR}" ou "$(VAR)"
	 */
    public static final Pattern VARIABLE_REFERENCE = Pattern.compile("\\$\\{(\\w+)\\}|\\$\\((\\w+)\\)");
    
	/**
     * Première cible trouvée dans le fichier Bakefile.
     */
    private static String firstTarget;

	/**
     * Stocke les variables définies dans le Bakefile.
     */
    private static Map<String, String> variables = new HashMap<>();
    
	/**
	 * Constructeur de la classe BakefileParser.
	 * @param filename Nom du fichier Bakefile à parser
	 */
    public BakefileParser(String filename) {
        this.filename = filename;
        firstTarget = null;
    }

	/**
	 * Gérer les lignes de continuation.
	 * @param lines Liste des lignes du fichier Bakefile
	 * @param startIndex Index de la première ligne de continuation
	 * @return Tableau contenant la commande combinée, les lignes brutes et le nombre de lignes traitées
	 */
    private Object[] handleContinuationLines(List<String> lines, int startIndex) {
        StringBuilder combinedLine = new StringBuilder();
        List<String> rawLines = new ArrayList<>();
        int i = startIndex;
        
        // Ajouter la première ligne avec son backslash
        String firstLine = lines.get(i);
        rawLines.add(firstLine); // Garder la ligne telle quelle
        
        Matcher contMatcher = CONTINUATION_PATTERN.matcher(firstLine);
        if (contMatcher.matches()) {
            combinedLine.append(contMatcher.group(1).trim()).append(" ");
            i++;
        }
        
        // Traiter les lignes suivantes
        while (i < lines.size()) {
            String line = lines.get(i);
            rawLines.add(line); // Garder la ligne telle quelle
            
            contMatcher = CONTINUATION_PATTERN.matcher(line);
            if (contMatcher.matches()) {
                // Ajouter sans le backslash à la commande combinée
                combinedLine.append(contMatcher.group(1).trim()).append(" ");
                i++;
            } else {
                // Dernière ligne de la séquence (sans backslash)
                combinedLine.append(line.trim());
                i++;
                break;
            }
        }
        
        if (BakeCLI.isDebug()) {
            System.out.println("Debug: Combined " + (i - startIndex) + " lines into: [" + 
                              combinedLine.toString().trim() + "]");
            System.out.println("Debug: Raw lines preserved: " + rawLines.size());
        }
        
        // Retourner la commande combinée, les lignes brutes et le nombre de lignes
        return new Object[] { 
            combinedLine.toString().trim(),  // Commande combinée pour exécution
            rawLines,                        // Lignes brutes pour affichage
            i - startIndex                   // Nombre de lignes traitées
        };
    }

	/**
	 * Remplacer les variables dans une chaîne.
	 * @param input Chaîne à traiter
	 * @return Chaîne avec les variables remplacées
	 */
	private String replaceVariables(String input) {
		if (input == null) return null;
		
		String result = input;
		Set<String> processedVars = new HashSet<>();
		boolean changed;
		
		do {
			changed = false;
			Matcher matcher = VARIABLE_REFERENCE.matcher(result);
			StringBuffer sb = new StringBuffer();
			
			while (matcher.find()) {
				String varName = matcher.group(1) != null ? matcher.group(1) : matcher.group(2);
				
				// Vérifier si la variable existe
				String replacement = "";
				if (variables.containsKey(varName)) {
					String varValue = variables.get(varName);
					// Ne pas remplacer récursivement ici pour éviter les boucles infinies
					replacement = varValue;
				} else if (BakeCLI.isDebug()) {
					System.out.println("Debug: Variable '" + varName + "' not defined, replacing with empty string");
				}
				
				matcher.appendReplacement(sb, Matcher.quoteReplacement(replacement));
				changed = true;
				processedVars.add(varName);
			}
			matcher.appendTail(sb);
			result = sb.toString();
			
			// Si aucun changement n'a été fait dans ce passage, arrêter
			if (!changed) {
				break;
			}
			
			// Réinitialiser processedVars pour le prochain passage si nécessaire
			processedVars.clear();
			
		} while (changed);
		
		return result.trim();
	}

	/**
	 * Découper les dépendances en une liste de chaînes.
	 * @param depStr Chaîne de dépendances
	 * @return Liste de dépendances
	 */
	private List<String> splitDependencies(String depStr) {
		if (depStr == null || depStr.trim().isEmpty()) {
			return new ArrayList<>();
		}
	
		String cleanedStr = depStr.replaceAll("\\\\\\s*", " ");
		List<String> deps = new ArrayList<>();
		StringBuilder currentDep = new StringBuilder();
		boolean inQuotes = false;
		
		for (int i = 0; i < cleanedStr.length(); i++) {
			char c = cleanedStr.charAt(i);
			
			if (c == '"') {
				inQuotes = !inQuotes;
				currentDep.append(c);
			} else if (c == ' ' && !inQuotes) {
				// Espace en dehors des guillemets, c'est un séparateur
				if (currentDep.length() > 0) {
					String dep = currentDep.toString().trim();
					if (!dep.isEmpty()) {
						deps.add(dep);
					}
					currentDep = new StringBuilder();
				}
			} else {
				currentDep.append(c);
			}
		}
		
		// Ajouter la dernière dépendance s'il existe
		if (currentDep.length() > 0) {
			String dep = currentDep.toString().trim();
			if (!dep.isEmpty()) {
				deps.add(dep);
			}
		}
		
		if (BakeCLI.isDebug()) {
			System.out.println("Debug: Split dependencies: " + deps);
		}
		
		// Appliquer replaceVariables à chaque dépendance
		return deps.stream()
				  .map(this::replaceVariables)
				  .collect(Collectors.toList());
	}

	/**
	 * Découper les cibles en une liste de chaînes.
	 * @param targetStr Chaîne de cibles
	 * @return Liste de cibles
	 */
	private List<String> splitTargets(String targetStr) {
		if (targetStr == null || targetStr.trim().isEmpty()) {
			return new ArrayList<>();
		}
		
		List<String> targets = new ArrayList<>();
		StringBuilder currentTarget = new StringBuilder();
		boolean inQuotes = false;
		
		for (int i = 0; i < targetStr.length(); i++) {
			char c = targetStr.charAt(i);
			
			if (c == '"') {
				inQuotes = !inQuotes;
				currentTarget.append(c);
			} else if (c == ' ' && !inQuotes) {
				// Espace en dehors des guillemets, c'est un séparateur
				if (currentTarget.length() > 0) {
					String target = currentTarget.toString().trim();
					if (!target.isEmpty()) {
						targets.add(target);
					}
					currentTarget = new StringBuilder();
				}
			} else {
				currentTarget.append(c);
			}
		}
		
		// Ajouter le dernier target s'il existe
		if (currentTarget.length() > 0) {
			String target = currentTarget.toString().trim();
			if (!target.isEmpty()) {
				targets.add(target);
			}
		}
		
		if (BakeCLI.isDebug()) {
			System.out.println("Debug: Split targets: " + targets);
		}
		
		// Appliquer replaceVariables à chaque cible
		return targets.stream()
					 .map(this::replaceVariables)
					 .collect(Collectors.toList());
	}

	/**
	 * Analyser le fichier Bakefile pour extraire les règles de build.
	 * @return Liste des règles extraites
	 */
	public List<Rule> parse() {
		List<Rule> rules = new ArrayList<>();
		Set<String> phonyTargets = new HashSet<>();

		if (!Files.exists(Paths.get(filename))) {
			System.out.println("*** No targets specified and no makefile found.  Stop.");
			System.exit(2);
		}       

		try {
			List<String> lines = Files.readAllLines(Paths.get(filename));
			List<String> currentTargets = null;
			List<String> dependencies = new ArrayList<>();
			List<String> commands = new ArrayList<>();
			List<List<String>> displayCommands = new ArrayList<>();
			
			// Variable pour suivre si la ligne précédente était une continuation
			boolean previousLineContinues = false;
			
			for (int i = 0; i < lines.size(); i++) {
				String line = lines.get(i).replace("\r", "");
				
				// Ignorer les lignes vides
				if (line.trim().isEmpty()) {
					previousLineContinues = false; // Réinitialiser le flag de continuation
					continue;
				}
				
				// Gérer les erreurs de format (espaces au lieu de tabulations)
				// Mais uniquement si ce n'est pas une continuation de variable
				if (line.matches("^ +.*$") && !previousLineContinues) {
					System.err.println(filename + ":" + (i+1) + ": *** missing separator.  Stop.");
					System.exit(2);
				}
				
				// Vérifier si cette ligne se termine par un backslash (continuation)
				previousLineContinues = line.trim().endsWith("\\");
				
				// Matcher pour les déclarations .PHONY
				Matcher phonyMatcher = PHONY_PATTERN.matcher(line);
				if (phonyMatcher.matches()) {
					String[] phonies = phonyMatcher.group(1).trim().split("\\s+");
					Collections.addAll(phonyTargets, phonies);
					continue;
				}
				
				// Matcher pour les déclarations de variables
				Matcher varMatcher = VARIABLE_PATTERN.matcher(line);
				if (varMatcher.matches()) {
					String varName = varMatcher.group(1);
					String varValue = varMatcher.group(2).trim();
					
					// Vérifier si la ligne se termine par un backslash (continuation)
					if (varValue.endsWith("\\")) {
						StringBuilder fullValue = new StringBuilder(varValue.substring(0, varValue.length() - 1).trim());
						int j = i + 1;
						
						while (j < lines.size()) {
							String nextLine = lines.get(j).trim();
							
							if (nextLine.endsWith("\\")) {
								fullValue.append(" ").append(nextLine.substring(0, nextLine.length() - 1).trim());
								j++;
							} else {
								fullValue.append(" ").append(nextLine);
								i = j; // Mettre à jour l'indice principal pour sauter les lignes traitées
								break;
							}
						}
						
						varValue = fullValue.toString();
					}
					
					// Évaluer les variables référencées dans la valeur
					variables.put(varName, varValue);
					
					if (BakeCLI.isDebug()) {
						System.out.println("Debug: Variable defined: " + varName + " = " + varValue);
					}
					continue;
				}
				
				// Matcher pour les cibles et dépendances
				Matcher targetMatcher = TARGET_PATTERN.matcher(line);
				if (targetMatcher.matches()) {
					// Si nous avions des cibles précédentes, créons les règles correspondantes
					if (currentTargets != null) {
						// Créer une règle pour chaque cible avec les mêmes dépendances et commandes
						for (String target : currentTargets) {
							String resolvedTarget = replaceVariables(target.trim());
							rules.add(new Rule(
								resolvedTarget,
								dependencies,
								commands,
								displayCommands,
								phonyTargets.contains(resolvedTarget)
							));
							
							if (firstTarget == null) {
								firstTarget = resolvedTarget;
							}
						}
					}
					
					// Configuration pour les nouvelles cibles
					String targetStr = targetMatcher.group(1);
					if (BakeCLI.isDebug()) {
						System.out.println("Debug: Raw target(s): " + targetStr);
					}
					currentTargets = splitTargets(targetStr);
					
					if (BakeCLI.isDebug()) {
						System.out.println("Debug: Resolved targets: " + currentTargets);
					}
					
					String depStr = targetMatcher.group(2);
					
					// Vérifier si la chaîne de dépendances se termine par un backslash (continuation)
					if (depStr.trim().endsWith("\\")) {
						StringBuilder fullDeps = new StringBuilder(depStr.substring(0, depStr.trim().length() - 1).trim());
						int j = i + 1;
						
						while (j < lines.size()) {
							String nextLine = lines.get(j).trim();
							
							if (nextLine.endsWith("\\")) {
								fullDeps.append(" ").append(nextLine.substring(0, nextLine.length() - 1).trim());
								j++;
							} else {
								fullDeps.append(" ").append(nextLine);
								i = j; // Mettre à jour l'indice principal pour sauter les lignes traitées
								break;
							}
						}
						
						depStr = fullDeps.toString();
						if (BakeCLI.isDebug()) {
							System.out.println("Debug: Combined dependency line: [" + depStr + "]");
						}
					}
					
					dependencies = splitDependencies(depStr);
					commands = new ArrayList<>();
					displayCommands = new ArrayList<>();
					continue;
				}
				
				// Matcher pour les lignes de commande
				Matcher commandMatcher = COMMAND_PATTERN.matcher(line);
				if (commandMatcher.matches()) {
					String command = commandMatcher.group(1);
					
					// Gérer la continuation de ligne
					if (command.endsWith("\\")) {
						// Traiter la séquence complète de continuation
						Object[] result = handleContinuationLines(lines, i);
						String fullCommand = (String)result[0];
						@SuppressWarnings("unchecked")
						List<String> rawLines = (List<String>)result[1];
						int linesUsed = (Integer)result[2];
						
						// Ajouter la commande complète pour l'exécution
						commands.add(fullCommand);
						
						// Ajouter les lignes brutes pour l'affichage
						displayCommands.add(rawLines);
						
						// Ajuster i pour sauter les lignes traitées (moins 1 car la boucle for incrémente i)
						i += linesUsed - 1;
					} else {
						String executableCommand = command;
						commands.add(executableCommand);
						// Pour l'affichage, préserver le formatage mais remplacer les variables
						String displayLine = line;
						// Ne pas modifier les lignes qui commencent par @ (silencieuses)
						if (command.startsWith("@")) {
							displayLine = line; // Garder le formatage complet pour les commandes silencieuses
						} else {
							// Remplacer les variables dans la partie de la commande uniquement (après la tabulation)
							Matcher cmdMatcher = COMMAND_PATTERN.matcher(line);
							if (cmdMatcher.matches()) {
								String cmdPart = cmdMatcher.group(1);
								displayLine = "\t" + cmdPart;  // <= Garder les références aux variables
							}
						}
						List<String> singleLineDisplay = new ArrayList<>();
						singleLineDisplay.add(displayLine);
						displayCommands.add(singleLineDisplay);
					}
					continue;
				}
			}

			// Traiter les dernières cibles
			if (currentTargets != null) {
				// Créer une règle pour chaque cible avec les mêmes dépendances et commandes
				for (String target : currentTargets) {
					String resolvedTarget = replaceVariables(target.trim());
					rules.add(new Rule(
						resolvedTarget,
						dependencies,
						commands,
						displayCommands,
						phonyTargets.contains(resolvedTarget)
					));
					
					if (firstTarget == null) {
						firstTarget = resolvedTarget;
					}
				}
			}
			
			if (BakeCLI.isDebug()) {
				System.out.println("Debug: First target is: " + firstTarget);
				System.out.println("Debug: Parsed " + rules.size() + " rules.");
				for (Rule rule : rules) {
					System.out.println("Debug: Rule: " + rule.getName());
					System.out.println("Debug:   Commands: " + rule.getCommands().size());
					for (String cmd : rule.getCommands()) {
						System.out.println("Debug:     [" + cmd + "]");
					}
				}
			}

		} catch (IOException e) {
			e.printStackTrace();
		}
		return rules;
	}

	/**
	 * Remplacer les variables dans une chaîne (version statique pour l'évaluation tardive).
	 * @param input Chaîne à traiter
	 * @return Chaîne avec les variables remplacées
	 */
	public static String expandVariables(String input) {
		if (input == null) return null;
		
		String result = input;
		Set<String> processedVars = new HashSet<>();
		boolean changed;
		
		do {
			changed = false;
			Matcher matcher = VARIABLE_REFERENCE.matcher(result);
			StringBuffer sb = new StringBuffer();
			
			while (matcher.find()) {
				String varName = matcher.group(1) != null ? matcher.group(1) : matcher.group(2);
				
				// Vérifier si la variable existe
				String replacement = "";
				if (variables.containsKey(varName)) {
					String varValue = variables.get(varName);
					// Ne pas remplacer récursivement ici pour éviter les boucles infinies
					replacement = varValue;
				} else if (BakeCLI.isDebug()) {
					System.out.println("Debug: Variable '" + varName + "' not defined, replacing with empty string");
				}
				
				matcher.appendReplacement(sb, Matcher.quoteReplacement(replacement));
				changed = true;
				processedVars.add(varName);
			}
			matcher.appendTail(sb);
			result = sb.toString();
			
			// Si aucun changement n'a été fait dans ce passage, arrêter
			if (!changed) {
				break;
			}
			
			// Réinitialiser processedVars pour le prochain passage si nécessaire
			processedVars.clear();
			
		} while (changed);
		
		return result.trim();
	}

	/**
	 * Récupérer la première cible
	 * @return String la première cible
	 */
    public static String getFirstTarget() { 
        return firstTarget; 
    }
}