package file_system;

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

public class Directory {
    private List<File> fileList;
    private List<Directory> directoryList;
    private int dSize;

    public Directory() {
        fileList = new ArrayList<>();
        directoryList = new ArrayList<>();
    }

    public Directory (File file){
        fileList = new ArrayList<>();
        fileList.add(file);
        dSize += file.getSize();
    }

    public Directory(File file1, File file2) {
        fileList = new ArrayList<>();
        fileList.add(file1);
        dSize += file1.getSize();
        fileList.add(file2);
        dSize += file2.getSize();
    }

    public Directory(Directory directory, File file) {
        directoryList = new ArrayList<>();
        directoryList.add(directory);
        dSize += directory.getSize();
        fileList = new ArrayList<>();
        fileList.add(file);
        dSize += file.getSize();
    }

    public int getSize(){
        return dSize;
    }

    public Directory add(File file) {
        fileList.add(file);
        dSize += file.getSize();
        return this;
    }

    public Directory add(Directory directory) {
        if( directory != this ){
            directoryList.add(directory);
        }else{
            throw new IllegalArgumentException("You want to add self to itself: " + directory);
        }
        dSize += directory.getSize();
        return this;
    }
}