/* 
 * Copyright (C) 2006, 2007  Dennis Hunziker, Ueli Kistler
 * Copyright (C) 2007  Reto Schuettel, Robin Stocker
 *
 * IFS Institute for Software, HSR Rapperswil, Switzerland
 * 
 */

package org.python.pydev.refactoring.ast.visitors.context;

import java.util.ArrayList;
import java.util.List;

import org.python.pydev.parser.jython.SimpleNode;
import org.python.pydev.parser.jython.ast.Assign;
import org.python.pydev.parser.jython.ast.Attribute;
import org.python.pydev.parser.jython.ast.ClassDef;
import org.python.pydev.parser.jython.ast.FunctionDef;
import org.python.pydev.parser.jython.ast.Import;
import org.python.pydev.parser.jython.ast.ImportFrom;
import org.python.pydev.parser.jython.ast.Module;
import org.python.pydev.parser.jython.ast.Name;
import org.python.pydev.refactoring.ast.adapters.AbstractScopeNode;
import org.python.pydev.refactoring.ast.adapters.ModuleAdapter;
import org.python.pydev.refactoring.ast.adapters.SimpleAdapter;

public class ScopeAssignedVisitor extends AbstractContextVisitor<SimpleAdapter> {

    private List<String> globalVars;

    public ScopeAssignedVisitor(ModuleAdapter module, AbstractScopeNode<?> parent) {
        super(module, parent);
        globalVars = new ArrayList<String>();
    }

    @Override
    protected void registerInContext(SimpleNode node) {
        String varName = nodeHelper.getName(node);
        if (!(globalVars.contains(varName))) {
            globalVars.add(varName);
            super.registerInContext(node);
        }

    }

    @Override
    public Object visitImport(Import node) throws Exception {
        return null;
    }

    @Override
    public Object visitImportFrom(ImportFrom node) throws Exception {
        return null;
    }

    @Override
    public Object visitName(Name node) throws Exception {
        if (nodeHelper.isAssign(stack.peek())) {
            registerInContext(node);
        }
        return null;
    }

    @Override
    public Object visitAttribute(Attribute node) throws Exception {
        // ignore attribute (must)
        return null;
    }

    @Override
    public Object visitAssign(Assign node) throws Exception {
        before(node);
        visit(node.targets);
        after();
        return null;
    }

    @Override
    public Object visitModule(Module node) throws Exception {
        visit(node.body);
        return null;
    }

    @Override
    public Object visitFunctionDef(FunctionDef node) throws Exception {
        return null;
    }

    @Override
    public Object visitClassDef(ClassDef node) throws Exception {
        return null;
    }

    @Override
    protected SimpleAdapter createAdapter(AbstractScopeNode<?> parent, SimpleNode node) {
        return new SimpleAdapter(moduleAdapter, parent, node, moduleAdapter.getAdapterPrefs());
    }

}