aboutsummaryrefslogtreecommitdiffstats
path: root/theme-compiler/src/com/vaadin/sass/internal/visitor/ExtendNodeHandler.java
blob: c05f8f6d385b38c27e1d296c07c15bc8b04675e1 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
/*
 * Copyright 2011 Vaadin Ltd.
 * 
 * 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.vaadin.sass.internal.visitor;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.Map.Entry;

import com.vaadin.sass.internal.ScssStylesheet;
import com.vaadin.sass.internal.tree.BlockNode;
import com.vaadin.sass.internal.tree.ExtendNode;
import com.vaadin.sass.internal.tree.Node;

public class ExtendNodeHandler {
    private static Map<String, List<ArrayList<String>>> extendsMap = new HashMap<String, List<ArrayList<String>>>();

    public static void traverse(ExtendNode node) throws Exception {
        buildExtendsMap(node);
        modifyTree(ScssStylesheet.get());
    }

    private static void modifyTree(Node node) throws Exception {
        for (Node child : node.getChildren()) {
            if (child instanceof BlockNode) {
                BlockNode blockNode = (BlockNode) child;
                String selectorString = blockNode.getSelectors();
                if (extendsMap.get(selectorString) != null) {
                    for (ArrayList<String> sList : extendsMap
                            .get(selectorString)) {
                        ArrayList<String> clone = (ArrayList<String>) sList
                                .clone();
                        addAdditionalSelectorListToBlockNode(blockNode, clone,
                                null);
                    }
                } else {
                    for (Entry<String, List<ArrayList<String>>> entry : extendsMap
                            .entrySet()) {
                        if (selectorString.contains(entry.getKey())) {
                            for (ArrayList<String> sList : entry.getValue()) {
                                ArrayList<String> clone = (ArrayList<String>) sList
                                        .clone();
                                addAdditionalSelectorListToBlockNode(blockNode,
                                        clone, entry.getKey());
                            }
                        }
                    }
                }
            }
        }

    }

    private static void buildExtendsMap(ExtendNode node) {
        String extendedString = node.getListAsString();
        if (extendsMap.get(extendedString) == null) {
            extendsMap.put(extendedString, new ArrayList<ArrayList<String>>());
        }
        extendsMap.get(extendedString).add(
                ((BlockNode) node.getParentNode()).getSelectorList());
    }

    private static void addAdditionalSelectorListToBlockNode(
            BlockNode blockNode, ArrayList<String> list, String selectorString) {
        if (list != null) {
            for (int i = 0; i < list.size(); i++) {
                if (selectorString == null) {
                    blockNode.getSelectorList().add(list.get(i));
                } else {
                    ArrayList<String> newTags = new ArrayList<String>();
                    for (final String existing : blockNode.getSelectorList()) {
                        if (existing.contains(selectorString)) {
                            newTags.add(existing.replace(selectorString,
                                    list.get(i)));
                        }
                    }
                    blockNode.getSelectorList().addAll(newTags);
                }
            }
        }
    }
}