/*
* Copyright 2000-2013 JetBrains s.r.o.
* Copyright 2014-2014 AS3Boyan
* Copyright 2014-2014 Elias Ku
*
* 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.intellij.plugins.haxe.util;
import com.intellij.plugins.haxe.lang.psi.HaxeImportStatementRegular;
import com.intellij.plugins.haxe.lang.psi.HaxePackageStatement;
import com.intellij.psi.PsiElement;
import com.intellij.psi.PsiFile;
/**
* @author: Fedor.Korotkov
*/
public class HaxeAddImportHelper {
public static void addImport(String path, PsiFile file) {
int positionIndex = 0;
final PsiElement[] children = file.getChildren();
while ((children[positionIndex] instanceof HaxePackageStatement ||
UsefulPsiTreeUtil.isWhitespaceOrComment(children[positionIndex]) &&
positionIndex < children.length)) {
++positionIndex;
}
assert positionIndex < children.length;
insertImportBefore(path, file, children[positionIndex]);
}
private static void insertImportBefore(String path, PsiFile file, PsiElement child) {
final HaxeImportStatementRegular importStatement =
HaxeElementGenerator.createImportStatementFromPath(file.getProject(), path);
if (importStatement == null) {
return;
}
file.addBefore(importStatement, child);
}
}