/*
 * Copyright 2022 Google LLC
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package com.google.android.libraries.mobiledatadownload.file.integration.downloader;

import android.content.SharedPreferences;
import android.net.Uri;
import com.google.android.downloader.DownloadMetadata;
import com.google.common.base.Optional;
import com.google.common.util.concurrent.ListenableFuture;
import com.google.common.util.concurrent.ListeningExecutorService;
import java.io.IOException;

/** DownloadMetadataStore based on SharedPreferences. */
public final class SharedPreferencesDownloadMetadata implements DownloadMetadataStore {

  private final Object lock = new Object();

  private final SharedPreferences sharedPrefs;
  private final ListeningExecutorService backgroundExecutor;

  public SharedPreferencesDownloadMetadata(
      SharedPreferences sharedPrefs, ListeningExecutorService backgroundExecutor) {
    this.sharedPrefs = sharedPrefs;
    this.backgroundExecutor = backgroundExecutor;
  }

  /** Data fields for each URI persisted in SharedPreferences. */
  private enum Key {
    CONTENT_TAG("ct"),
    LAST_MODIFIED_TIME_SECS("lmts");

    final String sharedPrefsSuffix;

    Key(String sharedPrefsSuffix) {
      this.sharedPrefsSuffix = sharedPrefsSuffix;
    }
  }

  private static final String SPLIT_CHAR = "|";

  /** Returns the SharedPreferences key used to store the given data field for {@code uri}. */
  private static String getKey(Uri uri, Key key) {
    return uri + SPLIT_CHAR + key.sharedPrefsSuffix;
  }

  @Override
  public ListenableFuture<Optional<DownloadMetadata>> read(Uri uri) {
    return backgroundExecutor.submit(
        () -> {
          synchronized (lock) {
            // Checking for CONTENT_TAG is sufficient since we will always store both the content
            // tag and last modified timestamp in the same commit.
            if (!sharedPrefs.contains(getKey(uri, Key.CONTENT_TAG))) {
              return Optional.absent();
            }

            String contentTag = sharedPrefs.getString(getKey(uri, Key.CONTENT_TAG), "");
            long lastModifiedTimeSeconds =
                sharedPrefs.getLong(getKey(uri, Key.LAST_MODIFIED_TIME_SECS), 0);
            return Optional.of(DownloadMetadata.create(contentTag, lastModifiedTimeSeconds));
          }
        });
  }

  @Override
  public ListenableFuture<Void> upsert(Uri uri, DownloadMetadata metadata) {
    return backgroundExecutor.submit(
        () -> {
          synchronized (lock) {
            SharedPreferences.Editor editor = sharedPrefs.edit();
            editor.putString(getKey(uri, Key.CONTENT_TAG), metadata.getContentTag());
            editor.putLong(
                getKey(uri, Key.LAST_MODIFIED_TIME_SECS), metadata.getLastModifiedTimeSeconds());
            commitOrThrow(editor);

            return null;
          }
        });
  }

  @Override
  public ListenableFuture<Void> delete(Uri uri) {
    return backgroundExecutor.submit(
        () -> {
          synchronized (lock) {
            SharedPreferences.Editor editor = sharedPrefs.edit();
            editor.remove(getKey(uri, Key.CONTENT_TAG));
            editor.remove(getKey(uri, Key.LAST_MODIFIED_TIME_SECS));
            commitOrThrow(editor);

            return null;
          }
        });
  }

  /** Calls {@code editor.commit()} and throws IOException if the commit failed. */
  private static void commitOrThrow(SharedPreferences.Editor editor) throws IOException {
    if (!editor.commit()) {
      throw new IOException("Failed to commit");
    }
  }
}