/*
 This file is part of Subsonic.

 Subsonic is free software: you can redistribute it and/or modify
 it under the terms of the GNU General Public License as published by
 the Free Software Foundation, either version 3 of the License, or
 (at your option) any later version.

 Subsonic is distributed in the hope that it will be useful,
 but WITHOUT ANY WARRANTY; without even the implied warranty of
 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 GNU General Public License for more details.

 You should have received a copy of the GNU General Public License
 along with Subsonic.  If not, see <http://www.gnu.org/licenses/>.

 Copyright 2009 (C) Sindre Mehus
 */
package net.nullsum.audinaut.service.parser;

import android.content.Context;

import net.nullsum.audinaut.domain.MusicDirectory;

import org.xmlpull.v1.XmlPullParser;

import java.io.InputStream;
import java.util.HashMap;
import java.util.Map;

import static net.nullsum.audinaut.domain.MusicDirectory.Entry;

/**
 * @author Sindre Mehus
 */
public class MusicDirectoryParser extends MusicDirectoryEntryParser {

    public MusicDirectoryParser(Context context, int instance) {
        super(context, instance);
    }

    public MusicDirectory parse(String artist, InputStream inputStream) throws Exception {
        init(inputStream);

        MusicDirectory dir = new MusicDirectory();
        int eventType;
        boolean isArtist = false;
        Map<String, Entry> titleMap = new HashMap<>();
        do {
            eventType = nextParseEvent();
            if (eventType == XmlPullParser.START_TAG) {
                String name = getElementName();
                if ("child".equals(name) || "song".equals(name)) {
                    Entry entry = parseEntry(artist);

                    // Only check for songs
                    if (!entry.isDirectory()) {
                        // Check if duplicates
                        String disc = (entry.getDiscNumber() != null) ? Integer.toString(entry.getDiscNumber()) : "";
                        String track = (entry.getTrack() != null) ? Integer.toString(entry.getTrack()) : "";
                        String duplicateId = disc + "-" + track + "-" + entry.getTitle();

                        Entry duplicate = titleMap.get(duplicateId);
                        if (duplicate != null) {
                            // Check if the first already has been rebased or not
                            if (duplicate.getTitle().equals(entry.getTitle())) {
                                duplicate.rebaseTitleOffPath();
                            }

                            // Rebase if this is the second instance of this title found
                            entry.rebaseTitleOffPath();
                        } else {
                            titleMap.put(duplicateId, entry);
                        }
                    }

                    dir.addChild(entry);
                } else if ("directory".equals(name) || "artist".equals(name) || ("album".equals(name) && !isArtist)) {
                    dir.setName(get("name"));
                    dir.setId(get("id"));
                    dir.setParent(get("artistId"));
                    isArtist = true;
                } else if ("album".equals(name)) {
                    Entry entry = parseEntry(artist);
                    entry.setDirectory(true);
                    dir.addChild(entry);
                } else if ("error".equals(name)) {
                    handleError();
                }
            }
        } while (eventType != XmlPullParser.END_DOCUMENT);

        validate();

        return dir;
    }
}