// Copyright (C) 2016 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.googlesource.gerrit.plugins.verifystatus.client;

import com.google.gerrit.client.GerritUiExtensionPoint;
import com.google.gerrit.client.info.ChangeInfo;
import com.google.gerrit.client.info.ChangeInfo.RevisionInfo;
import com.google.gerrit.client.rpc.NativeMap;
import com.google.gerrit.plugin.client.Plugin;
import com.google.gerrit.plugin.client.extension.Panel;
import com.google.gerrit.plugin.client.rpc.RestApi;
import com.google.gwt.http.client.URL;
import com.google.gwt.user.client.rpc.AsyncCallback;
import com.google.gwt.user.client.ui.Anchor;
import com.google.gwt.user.client.ui.FlowPanel;
import com.google.gwt.user.client.ui.Grid;
import com.google.gwt.user.client.ui.HorizontalPanel;
import com.google.gwt.user.client.ui.Image;
import com.google.gwt.user.client.ui.InlineHyperlink;
import com.google.gwt.user.client.ui.InlineLabel;

/** Extension for change screen that displays a status below the label info. */
public class JobsPanel extends FlowPanel {
  private final ConfigInfo info;

  static class Factory implements Panel.EntryPoint {
    private final ConfigInfo info;

    public Factory(ConfigInfo info) {
      this.info = info;
    }

    @Override
    public void onLoad(Panel panel) {
      RevisionInfo rev = panel.getObject(GerritUiExtensionPoint.Key.REVISION_INFO).cast();
      if (rev.isEdit()) {
        return;
      }

      panel.setWidget(new JobsPanel(panel, info));
    }
  }

  JobsPanel(Panel panel, ConfigInfo info) {
    this.info = info;
    final ChangeInfo change = panel.getObject(GerritUiExtensionPoint.Key.CHANGE_INFO).cast();
    String decodedChangeId = URL.decodePathSegment(change.id());
    final RevisionInfo rev = panel.getObject(GerritUiExtensionPoint.Key.REVISION_INFO).cast();
    new RestApi("changes")
        .id(decodedChangeId)
        .view("revisions")
        .id(rev.id())
        .view(Plugin.get().getPluginName(), "verifications")
        .addParameter("sort", info.sortJobsPanel())
        .addParameter("filter", "CURRENT")
        .get(
            new AsyncCallback<NativeMap<VerificationInfo>>() {
              @Override
              public void onSuccess(NativeMap<VerificationInfo> result) {
                if (!result.isEmpty()) {
                  final String patchsetId = change._number() + "/" + rev.id();
                  display(patchsetId, result);
                }
              }

              @Override
              public void onFailure(Throwable caught) {
                // never invoked
              }
            });
  }

  private void display(String patchsetId, NativeMap<VerificationInfo> jobs) {
    int row = 0;
    int column = 1;
    Grid grid = new Grid(row, column);
    for (String key : jobs.keySet()) {
      HorizontalPanel p = new HorizontalPanel();
      short vote = jobs.get(key).value();
      if (vote > 0) {
        p.add(
            new Image(
                ((vote == 2) && info.enableInProgressStatus())
                    ? VerifyStatusPlugin.RESOURCES.progress()
                    : VerifyStatusPlugin.RESOURCES.greenCheck()));
      } else if (vote < 0) {
        p.add(new Image(VerifyStatusPlugin.RESOURCES.redNot()));
      } else if (vote == 0) {
        p.add(new Image(VerifyStatusPlugin.RESOURCES.warning()));
      }
      Anchor anchor = new Anchor(jobs.get(key).name(), jobs.get(key).url());
      anchor.setTitle("view logs");
      p.add(anchor);
      InlineLabel label = new InlineLabel(" (" + jobs.get(key).duration() + ")");
      label.setTitle("duration");
      p.add(label);
      if (jobs.get(key).rerun()) {
        Image img = new Image(VerifyStatusPlugin.RESOURCES.rerun());
        img.setTitle("re-run");
        p.add(img);
      }
      if (jobs.get(key).abstain()) {
        Image img = new Image(VerifyStatusPlugin.RESOURCES.info());
        img.setTitle("non voting");
        p.add(img);
      }
      grid.insertRow(row);
      grid.setWidget(row, 0, p);
      row++;
    }
    HorizontalPanel p = new HorizontalPanel();
    InlineHyperlink all =
        new InlineHyperlink(
            "Show All Reports", "/x/" + Plugin.get().getName() + "/jobs/" + patchsetId);
    p.add(all);
    grid.insertRow(row);
    grid.setWidget(row, 0, p);
    add(grid);
  }
}
