package ftbsc.lll.mapper.impl;

import com.google.auto.service.AutoService;
import ftbsc.lll.exceptions.MalformedMappingsException;
import ftbsc.lll.mapper.AbstractMapper;
import ftbsc.lll.mapper.IMapper;
import ftbsc.lll.mapper.MapperProvider;
import ftbsc.lll.mapper.tools.MappingUtils;
import ftbsc.lll.mapper.tools.data.ClassData;

import java.util.ArrayList;
import java.util.List;

/**
 * Special mapper type that actually resolves to an ordered
 * sequence of mappers applied one after the other.
 */
@AutoService(IMapper.class)
public class MultiMapper extends AbstractMapper {
	@Override
	public boolean claim(List<String> lines) {
		return lines.get(0).equals("lll multimapper");
	}

	@Override
	public void populate(List<String> lines, boolean ignoreErrors) throws MalformedMappingsException {
		List<IMapper> mapperList = new ArrayList<>();
		for(int i = 1; i < lines.size(); i++) {
			List<String> data = MapperProvider.fetchFromLocalOrRemote(lines.get(i));
			IMapper mapper = MapperProvider.getMapper(data);
			mapper.populate(data, ignoreErrors);
			mapperList.add(mapper);
		}

		mapperList.get(0).getRawMappings().forEach((name, data) -> {
			ClassData finalData = data;
			for(int i = 1; i < mapperList.size(); i++)
				finalData = mapperList.get(i).getClassData(finalData.nameMapped);

			ClassData sumData = new ClassData(data.name, finalData.nameMapped);

			data.getMethods().forEach((signature, methodData) -> {
				for(int i = 1; i < mapperList.size(); i++) {
					IMapper mapper = mapperList.get(i);
					methodData = mapper.getMethodData(methodData.parentClass.nameMapped, methodData.nameMapped,
						MappingUtils.mapMethodDescriptor(methodData.signature.descriptor, mapper, false));
				}
				sumData.addMethod(signature.name, methodData.nameMapped, signature.descriptor);
			});

			data.getFields().forEach((fieldName, fieldData) -> {
				for(int i = 1; i < mapperList.size(); i++)
					fieldData = mapperList.get(i).getFieldData(fieldData.parentClass.nameMapped, fieldData.nameMapped);
				sumData.addField(fieldName, fieldData.nameMapped);
			});

			this.mappings.put(sumData.name, sumData);
		});
	}

	@Override
	protected AbstractMapper newInstance() {
		return new MultiMapper();
	}
}