/*
 * Copyright 2013 Urs Wolfer
 *
 * 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.urswolfer.intellij.plugin.gerrit.ui.diff;

import com.google.gerrit.extensions.api.changes.DraftInput;
import com.google.gerrit.extensions.common.ChangeInfo;
import com.google.gerrit.extensions.common.Comment;
import com.google.gerrit.extensions.common.CommentInfo;
import com.intellij.openapi.actionSystem.AnAction;
import com.intellij.openapi.actionSystem.AnActionEvent;
import com.intellij.openapi.actionSystem.PlatformDataKeys;
import com.intellij.openapi.editor.Editor;
import com.intellij.openapi.editor.markup.RangeHighlighter;
import com.intellij.openapi.project.DumbAware;
import com.intellij.openapi.project.Project;
import com.intellij.openapi.ui.popup.JBPopup;
import com.intellij.openapi.ui.popup.JBPopupAdapter;
import com.intellij.openapi.ui.popup.LightweightWindowEvent;
import com.intellij.util.Consumer;
import com.urswolfer.intellij.plugin.gerrit.rest.GerritUtil;

import javax.swing.*;

/**
 * @author Urs Wolfer
 *
 * Some parts based on code from:
 * https://github.com/ktisha/Crucible4IDEA
 */
@SuppressWarnings("ComponentNotRegistered") // added with code
public class AddCommentAction extends AnAction implements DumbAware {

    private final Editor editor;
    private final CommentsDiffTool commentsDiffTool;
    private final GerritUtil gerritUtil;
    private final ChangeInfo changeInfo;
    private final String revisionId;
    private final String filePath;
    private final CommentBalloonBuilder commentBalloonBuilder;
    private final Comment.Side commentSide;
    private final Comment commentToEdit;
    private final RangeHighlighter lineHighlighter;
    private final RangeHighlighter rangeHighlighter;
    private final Comment replyToComment;

    public AddCommentAction(String label,
                            Icon icon,
                            CommentsDiffTool commentsDiffTool,
                            GerritUtil gerritUtil,
                            Editor editor,
                            CommentBalloonBuilder commentBalloonBuilder,
                            ChangeInfo changeInfo,
                            String revisionId,
                            String filePath,
                            Comment.Side commentSide,
                            Comment commentToEdit,
                            RangeHighlighter lineHighlighter,
                            RangeHighlighter rangeHighlighter,
                            Comment replyToComment) {
        super(label, null, icon);

        this.commentsDiffTool = commentsDiffTool;
        this.gerritUtil = gerritUtil;
        this.changeInfo = changeInfo;
        this.revisionId = revisionId;
        this.filePath = filePath;
        this.editor = editor;
        this.commentBalloonBuilder = commentBalloonBuilder;
        this.commentSide = commentSide;
        this.commentToEdit = commentToEdit;
        this.lineHighlighter = lineHighlighter;
        this.rangeHighlighter = rangeHighlighter;
        this.replyToComment = replyToComment;
    }

    public void actionPerformed(AnActionEvent e) {
        final Project project = e.getData(PlatformDataKeys.PROJECT);
        if (project == null) return;
        addVersionedComment(project);
    }

    private void addVersionedComment(final Project project) {
        if (editor == null || filePath == null) return;

        final CommentForm commentForm = new CommentForm(project, editor, filePath, commentSide, commentToEdit);
        final JBPopup balloon = commentBalloonBuilder.getNewCommentBalloon(commentForm, "Comment");
        balloon.addListener(new JBPopupAdapter() {
            @Override
            public void onClosed(LightweightWindowEvent event) {
                DraftInput comment = commentForm.getComment();
                if (comment != null) {
                    handleComment(comment, project);
                }
            }
        });
        commentForm.setBalloon(balloon);
        balloon.showInBestPositionFor(editor);
        commentForm.requestFocus();
    }

    private void handleComment(final DraftInput comment, final Project project) {
        if (commentToEdit != null) {
            comment.id = commentToEdit.id;
        }

        if (replyToComment != null) {
            comment.inReplyTo = replyToComment.id;
        }

        gerritUtil.saveDraftComment(changeInfo._number, revisionId, comment, project,
                new Consumer<CommentInfo>() {
                    @Override
                    public void consume(CommentInfo commentInfo) {
                        if (commentToEdit != null) {
                            commentsDiffTool.removeComment(editor, lineHighlighter, rangeHighlighter);
                        }
                        commentsDiffTool.addComment(editor, changeInfo, revisionId, project, commentInfo);
                    }
                });
    }
}

@@ -4838,24 +4838,170 @@
 Comment.id;%0A
+            comment.side = replyToComment.side;%0A            comment.line = replyToComment.line;%0A            comment.range = replyToComment.range;%0A
         %7D%0A%0A 

set correct line, range and side in reply-to-comments