com.hp.alm.ali.idea.action.attachment.AttachmentUpdateTask.java Source code

Introduction

Here is the source code for com.hp.alm.ali.idea.action.attachment.AttachmentUpdateTask.java

Source

/*
 * Copyright 2013 Hewlett-Packard Development Company, L.P
 *
 * 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.hp.alm.ali.idea.action.attachment;

import com.hp.alm.ali.idea.entity.EntityRef;
import com.hp.alm.ali.idea.services.AttachmentService;
import com.hp.alm.ali.idea.progress.CanceledException;
import com.hp.alm.ali.idea.progress.IndicatingInputStream;
import com.intellij.openapi.progress.PerformInBackgroundOption;
import com.intellij.openapi.progress.ProgressIndicator;
import com.intellij.openapi.progress.Task;
import com.intellij.openapi.project.Project;
import com.intellij.openapi.ui.Messages;
import com.intellij.util.ui.UIUtil;

import java.io.File;
import java.io.FileNotFoundException;

public class AttachmentUpdateTask extends Task.Backgroundable {

    private File file;
    private String filename;
    private String name;
    private String description;
    private EntityRef parent;
    private AttachmentService attachmentService;

    public AttachmentUpdateTask(Project project, File file, String filename, String name, String description,
            EntityRef parent) {
        super(project, "Update attachment: " + filename, true, PerformInBackgroundOption.DEAF);

        this.file = file;
        this.filename = filename;
        this.name = name;
        this.description = description;
        this.parent = parent;

        this.attachmentService = project.getComponent(AttachmentService.class);
    }

    public void run(ProgressIndicator indicator) {
        if (file != null) {
            IndicatingInputStream fis;
            try {
                fis = new IndicatingInputStream(file, indicator);
            } catch (FileNotFoundException e) {
                UIUtil.invokeLaterIfNeeded(new Runnable() {
                    @Override
                    public void run() {
                        Messages.showErrorDialog("File not found.", "Failed to upload attachment");
                    }
                });
                return;
            }
            try {
                if (!attachmentService.updateAttachmentContent(filename, parent, fis, file.length(), false)) {
                    return;
                }
            } catch (CanceledException e) {
                return;
            }
        }
        if (description != null && !indicator.isCanceled()) {
            attachmentService.updateAttachmentProperty(filename, parent, "description", description, false);
        }
        if (name != null) {
            attachmentService.updateAttachmentProperty(filename, parent, "name", name, false);
        }
    }
}