co.arcs.groove.basking.task.GetSongsToSyncTask Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of basking Show documentation

Java + command-line tool for synchronising Grooveshark libraries.

There is a newer version: 1.7.2

package co.arcs.groove.basking.task;

import com.google.common.collect.ImmutableSet;
import com.google.common.collect.Sets;
import com.google.common.collect.Sets.SetView;
import com.google.common.eventbus.EventBus;

import java.util.Set;

import co.arcs.groove.basking.event.Events.GetSongsToSyncFinishedEvent;
import co.arcs.groove.basking.event.Events.GetSongsToSyncProgressChangedEvent;
import co.arcs.groove.basking.event.Events.GetSongsToSyncStartedEvent;
import co.arcs.groove.thresher.Client;
import co.arcs.groove.thresher.Song;
import co.arcs.groove.thresher.User;

public class GetSongsToSyncTask implements Task> {

    private final EventBus bus;
    private final Client client;
    private final String username;
    private final String password;

    public GetSongsToSyncTask(EventBus bus, Client client, String username, String password) {
        this.bus = bus;
        this.client = client;
        this.username = username;
        this.password = password;
    }

    @Override
    public Set call() throws Exception {

        bus.post(new GetSongsToSyncStartedEvent(this));
        bus.post(new GetSongsToSyncProgressChangedEvent(this, 0, 3));

        User user = client.login(username, password);

        bus.post(new GetSongsToSyncProgressChangedEvent(this, 1, 3));

        // The library.get() response contains favorited songs that do not have
        // the 'favorited' property set. To work around this, favorites are
        // removed from the library set and replaced with instances from the
        // favorites set. The result is that all songs within the resulting set
        // are 'collected', and some are 'favorited'.
        ImmutableSet library = ImmutableSet.copyOf(user.library().get());

        bus.post(new GetSongsToSyncProgressChangedEvent(this, 2, 3));

        ImmutableSet favorites = ImmutableSet.copyOf(user.favorites().get());

        SetView nonFavorites = Sets.difference(library, favorites);
        SetView all = Sets.union(nonFavorites, favorites);

        bus.post(new GetSongsToSyncProgressChangedEvent(this, 3, 3));

        bus.post(new GetSongsToSyncFinishedEvent(this, all.size()));

        return Sets.newHashSet(all);
    }
}