import 'package:d4rt/d4rt.dart';
import 'package:collection/collection.dart';
import 'package:d4rt_formulas/d4rt_formulas.dart';

class Multimap<K, V> extends DelegatingMap<K, List<V>> {
  final Map<K, List<V>> _map;

  Multimap(super.map) : _map = map;

  factory Multimap.create() {
    return Multimap({});
  }

  @override
  List<V>? operator [](Object? key) {
    if (_map.containsKey(key)) {
      return super[key];
    }
    final List<V> newList = [];
    super[key as K] = newList;
    return super[key];
  }
}

class Corpus{
  final Multimap<String, Formula> _tags = Multimap.create();
  final Map<String, Formula> _allFormulas = {};

  void loadFormulas(List<Formula> formulas, {bool replaceOnDuplicates = false, bool checkUnits = true}) {
    for (final formula in formulas) {
      if (!replaceOnDuplicates && _allFormulas.containsKey(formula.name)) {
        throw ArgumentError("Duplicate formula:$formula");
      }

      if( checkUnits ){
        for( final inputVar in formula.input + [formula.output] ){
          if( inputVar.unit != null && !_allUnits.containsKey(inputVar.unit) ){
            throw ArgumentError( "Unit not found: ${inputVar.unit}");
          }
        }
      }

      _allFormulas[formula.name] = formula;
      for( final tag in formula.tags ){
        _tags[tag]?.add(formula);
      }
    }
  }

  List<Formula> getTagFormulas(String tag){
    if( _tags[tag] == null ){
      return [];
    }
    return _tags[tag]?.toList(growable:false) as List<Formula>;
  }

  List<Formula> getFormulas(){
    return _allFormulas.values.toList(growable:false);
  }

  final Multimap<String, String> _baseToUnits = Multimap.create();
  final Map<String, UnitSpec> _allUnits = {};

  void loadUnits(List<UnitSpec> units, [bool replaceOnDuplicates = false]) {
    for (final unit in units) {
      if (!replaceOnDuplicates && _allUnits.containsKey(unit.name)) {
        throw ArgumentError("Duplicate unit:$unit");
      }
      _allUnits[unit.name] = unit;
      _baseToUnits[unit.baseUnit]?.add(unit.name);
    }
  }

  List<String> unitsOfSameMagnitude(String? unit){
    if( unit == null ){
      return ["unitless"];
    }
    final base = getUnit(unit).baseUnit;
    return _baseToUnits[base] as List<String>;
  }


  UnitSpec getUnit(String unit) {
    if (!_allUnits.containsKey(unit)) {
      print(  _allUnits.keys.join(",") );
      throw ArgumentError("Unit not found:$unit");
    }
    return _allUnits.get(unit);
  }

  String _converterFromCodeStringAsExpression(Number x, String codeString) {
    final buffer = StringBuffer();
    buffer.writeln("final x = ${x};");
    buffer.writeln("main(){return $codeString;}");
    final code = buffer.toString();
    return code;
  }

  String _converterFromCodeStringAsStatement(Number x, String codeString) {
    final buffer = StringBuffer();
    buffer.writeln("final x = ${x};");
    buffer.writeln("main(){ $codeString; return x; }");
    final code = buffer.toString();
    return code;
  }

  Number _convertToBase(Number x, String fromUnit) {
    final unit = getUnit(fromUnit);

    if (unit.factorFromUnitToBase != null) {
      return x * (unit.factorFromUnitToBase as Number);
    }

    if (unit.codeFromUnitToBase == null) {
      throw ArgumentError("Unit has no codeFromUnitToBase: $unit");
    }

    final ret = _convertUsingCode(x, unit.codeFromUnitToBase as String);
    return ret;
  }

  Number _convertFromBase(Number x, String toUnit) {
    final unit = getUnit(toUnit);

    if (unit.factorFromUnitToBase != null) {
      return x / (unit.factorFromUnitToBase as Number);
    }

    if (unit.codeFromBaseToUnit == null) {
      throw ArgumentError("Unit has no codeFromBaseToUnit: $unit");
    }

    final ret = _convertUsingCode(x, unit.codeFromBaseToUnit as String);
    return ret;
  }


  Number _convertUsingCode(Number x, String code ){
    late String completeSourceExpression;
    late String completeSourceStatement;
    try {
      completeSourceExpression = _converterFromCodeStringAsExpression(x, code);
      final ret = _evaluate(completeSourceExpression);
      return ret;
    }
    catch(e1, stack1){
      try{
        completeSourceStatement = _converterFromCodeStringAsStatement(x, code);
        final ret = _evaluate(completeSourceStatement);
        return ret;
      }
      catch( e2, stack2 ){
        print(completeSourceExpression);
        print(e1);
        print(stack1);
        print(completeSourceStatement);
        print(e2);
        print(stack2);
        throw FormulaEvaluationException( "Evaluation as statement and expression failed" );
      }
    }
  }

  static Number _evaluate(String code, [D4rt? interpreter]) {
    final d4rtInterpreter = interpreter ?? FormulaEvaluator.createDefaultInterpreter();
    FormulaEvaluator.prepareInterpreter(d4rtInterpreter);
    final completeCode = "${FormulaEvaluator.d4rtImports}\n$code";
    final result = d4rtInterpreter.execute(source: completeCode);
    return result.toDouble();
  }


  Number convert(Number x, String fromUnit, String toUnit) {
    final xBase = _convertToBase(x, fromUnit);
    final xTo = _convertFromBase(xBase, toUnit);

    //print( "convert: x:${x}${fromUnit} xTo:${xTo}${toUnit}");
    return xTo;
  }

  Iterable<UnitSpec> allUnits() => _allUnits.values;


}