//import java.util.Iterator;
import java.util.*;

public class MediaDatabase implements Iterable<MediaFile> {

	private ArrayList<MediaFile> files ;
	
	public MediaDatabase(){
		
		files = new ArrayList<MediaFile>();
	}
	
	public void addFile(MediaFile mediaFile){
		files.add(mediaFile);
	}
	
	public void removeFile(MediaFile mediaFile){
		files.remove(mediaFile);
	}
	
	
	@Override
	public Iterator<MediaFile> iterator() {
		// TODO Auto-generated method stub
		return files.iterator();
	}
	
	public MediaFile getFile(String name){
		for(MediaFile file : files){
			if(file.getName().equals(name))
				return file;
		}
		
		return null;
	}

	public int getNumberOfFiles(){
		return files.size();
	}
	
	public String[] getFileNames(){
		String[] fileNames = new String[files.size()];
		for(int i=0;i<files.size();i++){
			fileNames[i] = files.get(i).getName();
		}
		
		return fileNames;
	}
}