package peergos.shared.login;

import peergos.shared.crypto.asymmetric.*;
import peergos.shared.user.*;

import java.util.*;
import java.util.concurrent.*;

public class OfflineAccountStore implements Account {

    private final Account target;
    private final LoginCache local;

    public OfflineAccountStore(Account target, LoginCache local) {
        this.target = target;
        this.local = local;
    }

    @Override
    public CompletableFuture<Boolean> setLoginData(LoginData login, byte[] auth) {
        return target.setLoginData(login, auth).thenApply(r -> {
            if (r)
                local.setLoginData(login);
            return r;
        });
    }

    @Override
    public CompletableFuture<UserStaticData> getLoginData(String username, PublicSigningKey authorisedReader, byte[] auth) {
        return target.getLoginData(username, authorisedReader, auth)
                .thenApply(entryPoints -> {
                    local.setLoginData(new LoginData(username, entryPoints, authorisedReader, Optional.empty()));
                    return entryPoints;
                }).exceptionally(t -> local.getEntryData(username, authorisedReader).join());
    }
}

package peergos.shared.login;

import peergos.shared.crypto.asymmetric.*;
import peergos.shared.user.*;
import peergos.shared.util.*;

import java.util.*;
import java.util.concurrent.*;

public class OfflineAccountStore implements Account {

    private final Account target;
    private final LoginCache local;

    public OfflineAccountStore(Account target, LoginCache local) {
        this.target = target;
        this.local = local;
    }

    @Override
    public CompletableFuture<Boolean> setLoginData(LoginData login, byte[] auth) {
        return target.setLoginData(login, auth).thenApply(r -> {
            if (r)
                local.setLoginData(login);
            return r;
        });
    }

    @Override
    public CompletableFuture<UserStaticData> getLoginData(String username, PublicSigningKey authorisedReader, byte[] auth) {
        return Futures.asyncExceptionally(() -> target.getLoginData(username, authorisedReader, auth)
                        .thenApply(entryPoints -> {
                            local.setLoginData(new LoginData(username, entryPoints, authorisedReader, Optional.empty()));
                            return entryPoints;
                        }),
                t -> local.getEntryData(username, authorisedReader));
    }
}

Remove join in offline account cache