org.jetbrains.kotlin.wizards.FileCreationOp.java Source code

Introduction

Here is the source code for org.jetbrains.kotlin.wizards.FileCreationOp.java

Source

/*******************************************************************************
 * Copyright 2000-2014 JetBrains s.r.o.
 *
 * 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 org.jetbrains.kotlin.wizards;

import static org.eclipse.ui.ide.undo.WorkspaceUndoUtil.getUIInfoAdapter;

import java.io.ByteArrayInputStream;
import java.io.File;
import java.lang.reflect.InvocationTargetException;

import org.eclipse.core.commands.ExecutionException;
import org.eclipse.core.resources.IFile;
import org.eclipse.core.resources.IProject;
import org.eclipse.core.runtime.CoreException;
import org.eclipse.core.runtime.IPath;
import org.eclipse.core.runtime.IProgressMonitor;
import org.eclipse.jdt.core.IPackageFragment;
import org.eclipse.jdt.core.IPackageFragmentRoot;
import org.eclipse.jface.operation.IRunnableWithProgress;
import org.eclipse.swt.widgets.Shell;
import org.eclipse.ui.PlatformUI;
import org.eclipse.ui.ide.undo.CreateFileOperation;
import org.jetbrains.kotlin.core.log.KotlinLogger;
import org.jetbrains.kotlin.idea.JetFileType;

public class FileCreationOp implements IRunnableWithProgress {

    private final static String EXT = "." + JetFileType.INSTANCE.getDefaultExtension();

    private final IPackageFragmentRoot sourceDir;
    private final IPackageFragment packageFragment;
    private final String unitName;
    private final String contents;
    private final Shell shell;

    private IFile result;

    public FileCreationOp(IPackageFragmentRoot sourceDir, IPackageFragment packageFragment, String unitName,
            String contents, Shell shell) {
        this.sourceDir = sourceDir;
        this.packageFragment = packageFragment;
        this.contents = contents;
        this.shell = shell;
        this.unitName = unitName;
    }

    public IFile getResult() {
        return result;
    }

    @Override
    public void run(IProgressMonitor monitor) throws InvocationTargetException {
        result = makeFile(packageFragment, sourceDir, unitName);
        try {
            if (!result.exists()) {
                CreateFileOperation op = new CreateFileOperation(result, null, null, "Create Kotlin Source File");
                PlatformUI.getWorkbench().getOperationSupport().getOperationHistory().execute(op, monitor,
                        getUIInfoAdapter(shell));
                result.appendContents(new ByteArrayInputStream(contents.getBytes()), false, false, monitor);
            } else {
                result = null;
            }
        } catch (ExecutionException e) {
            KotlinLogger.logAndThrow(e);
        } catch (CoreException e) {
            KotlinLogger.logAndThrow(e);
        }
    }

    static String getCompilationUnitName(String name) {
        if (name.endsWith(EXT)) {
            return name;
        }

        return name + EXT;
    }

    static String getSimpleUnitName(String name) {
        if (name.endsWith(EXT)) {
            return name.substring(0, name.length() - EXT.length());
        }

        return name;
    }

    static String getExtensionRegexp() {
        return "(\\" + EXT + ")?";
    }

    public static IFile makeFile(IPackageFragment packageFragment, IPackageFragmentRoot sourceDir,
            String unitName) {
        IPath path = packageFragment.getPath().append(getCompilationUnitName(unitName));
        IProject project = sourceDir.getJavaProject().getProject();

        return project.getFile(path.makeRelativeTo(project.getFullPath()));
    }

    static boolean fileExists(IFile file) {
        return new File(file.getRawLocation().toOSString()).exists();
    }
}