// Copyright (C) 2014 The Android Open Source Project
//
// 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.gerrit.client.changes;

import com.google.gerrit.client.VoidResult;
import com.google.gerrit.client.editor.EditFileInfo;
import com.google.gerrit.client.rpc.GerritCallback;
import com.google.gerrit.client.rpc.HttpCallback;
import com.google.gerrit.client.rpc.NativeString;
import com.google.gerrit.client.rpc.RestApi;
import com.google.gerrit.reviewdb.client.Patch;
import com.google.gerrit.reviewdb.client.PatchSet;
import com.google.gwt.core.client.JavaScriptObject;
import com.google.gwt.user.client.rpc.AsyncCallback;

/** REST API helpers to remotely edit a change. */
public class ChangeEditApi {
  /** Get file (or commit message) contents. */
  public static void get(PatchSet.Id id, String path,
      HttpCallback<NativeString> cb) {
    RestApi api;
    if (id.get() != 0) {
      // Read from a published revision, when change edit doesn't
      // exist for the caller, or is not currently active.
      api = ChangeApi.revision(id).view("files").id(path).view("content");
    } else if (Patch.COMMIT_MSG.equals(path)) {
      api = editMessage(id.getParentKey().get());
    } else {
      api = editFile(id.getParentKey().get(), path);
    }
    api.get(cb);
  }

  /** Get meta info for change edit. */
  public static void getMeta(PatchSet.Id id, String path,
      AsyncCallback<EditFileInfo> cb) {
    if (id.get() != 0) {
      throw new IllegalStateException("only supported for edits");
    }
    editFile(id.getParentKey().get(), path).view("meta").get(cb);
  }

  /** Put message into a change edit. */
  public static void putMessage(int id, String m, GerritCallback<VoidResult> cb) {
    editMessage(id).put(m, cb);
  }

  /** Put contents into a file or commit message in a change edit. */
  public static void put(int id, String path, String content,
      GerritCallback<VoidResult> cb) {
    if (Patch.COMMIT_MSG.equals(path)) {
      putMessage(id, content, cb);
    } else {
      editFile(id, path).put(content, cb);
    }
  }

  /** Delete a file in the pending edit. */
  public static void delete(int id, String path, AsyncCallback<VoidResult> cb) {
    editFile(id, path).delete(cb);
  }

  /** Rename a file in the pending edit. */
  public static void rename(int id, String path, String newPath,
      AsyncCallback<VoidResult> cb) {
    Input in = Input.create();
    in.oldPath(path);
    in.newPath(newPath);
    ChangeApi.edit(id).post(in, cb);
  }

  /** Restore (undo delete/modify) a file in the pending edit. */
  public static void restore(int id, String path, AsyncCallback<VoidResult> cb) {
    Input in = Input.create();
    in.restorePath(path);
    ChangeApi.edit(id).post(in, cb);
  }

  private static RestApi editMessage(int id) {
    return ChangeApi.change(id).view("edit:message");
  }

  private static RestApi editFile(int id, String path) {
    return ChangeApi.edit(id).id(path);
  }

  private static class Input extends JavaScriptObject {
    static Input create() {
      return createObject().cast();
    }

    final native void restorePath(String p) /*-{ this.restore_path=p }-*/;
    final native void oldPath(String p) /*-{ this.old_path=p }-*/;
    final native void newPath(String p) /*-{ this.new_path=p }-*/;

    protected Input() {
    }
  }
}
