package drugi;

import jdk.nashorn.api.tree.Tree;

import java.io.IOException;
import java.nio.file.*;
import java.nio.file.attribute.BasicFileAttributes;
import java.util.LinkedHashMap;
import java.util.List;
import java.util.Map;
import java.util.TreeMap;

public class citaj extends SimpleFileVisitor<Path> {
    List<String> extensions;
    Map<Path, TreeMap<String, Long>> sizes;

    public Map<Path, TreeMap<String, Long>> getSizes() {
        return sizes;
    }

    public citaj(List input){
        extensions = input;
        sizes = new LinkedHashMap<>();
    }

    @Override
    public FileVisitResult preVisitDirectory(Path dir, BasicFileAttributes attrs) throws IOException {
        return FileVisitResult.CONTINUE;
    }

    @Override
    public FileVisitResult visitFile(Path file, BasicFileAttributes attrs) throws IOException {
        //ako file zavrsava s nekom od ekstenzija iz extensions onda dodaj sa imenom godine, mjeseca velicinom;
        for(String i : extensions){
            System.out.println(file.toString());
            if(file.toString().endsWith(i)){
                String godina = file.getParent().getParent().getFileName().toString();
                String mjesec = file.getParent().getFileName().toString();
                if(sizes.containsKey(file)){
                    Map tren = sizes.get(file);
                    if(tren.containsKey(i)){
                        tren.replace(i, (Long)tren.get(i) + file.toFile().length());
                    }
                    else{
                        tren.put(i, file.toFile().length());
                    }
                }
                else {
                    TreeMap<String, Long> val = new TreeMap<String, Long>();
                    val.put(i, file.toFile().length());
                    sizes.put(file, val);
                }
            }
        }
        return FileVisitResult.CONTINUE;
    }

    @Override
    public FileVisitResult postVisitDirectory(Path dir, IOException exc) throws IOException {
        return FileVisitResult.CONTINUE;
    }


}