summaryrefslogtreecommitdiffstats
path: root/tags/cpp-1.0-incubating-M2-final/sca/tools/scagen/src/org/apache/tuscany/sca/cpp/tools/common/DirectoryTree.java
blob: 36d0af18b3b2a8114e31c0059a4372336eeb2db6 (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
98
99
100
101
102
103
104
105
/**
 *
 *  Licensed to the Apache Software Foundation (ASF) under one
 *  or more contributor license agreements.  See the NOTICE file
 *  distributed with this work for additional information
 *  regarding copyright ownership.  The ASF licenses this file
 *  to you 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.
 */

/* @version $Rev$ $Date$ */

/*
 * Branched from the original class that was also contributed to the
 * org.apache.axis.tools.common package.
 */
package org.apache.tuscany.sca.cpp.tools.common;

import java.io.File;
import java.util.Set;
import java.util.StringTokenizer;

public class DirectoryTree {
    private FileActor actor;

    private Set extensions;

    public DirectoryTree(FileActor actor, Set extensions) {
        this.actor = actor;
        this.extensions = extensions;
    }

    /**
     * Starts adding trace into the given file. If the given file is a directory
     * then this the starting directory and all code beneath and in this
     * directory will be given trace.
     * 
     * @param source -
     *            either the starting directory or one file to add trace to.
     */
    public void walkTree(File source, File target, int depth) throws Exception {
        depth++;
        boolean noTarget = (null == target);

        if (!source.canRead())
            Utils.rude("Cannot read from source directory " + source);
        if (!noTarget && !target.canWrite())
            Utils.rude("Cannot write to target directory " + target);

        if (source.isDirectory()) {
            File[] filesInDirectory = source.listFiles();
            for (int i = 0; i < filesInDirectory.length; i++) {
                File file = filesInDirectory[i];
                String name = file.getName();
                int dot = name.lastIndexOf('.');
                String ext = null;
                if (-1 != dot)
                    ext = name.substring(dot + 1);

                if (file.isDirectory()) {
                    File newTarget = null;
                    if (!noTarget) {
                        StringTokenizer st = new StringTokenizer(
                                file.getPath(), "\\/");
                        String newdir = null;
                        while (st.hasMoreTokens())
                            newdir = st.nextToken();
                        String targetName = maybeAppendSeparator(target
                                .toString());
                        newTarget = new File(targetName + newdir);
                        if (!newTarget.mkdir())
                            Utils.rude("Failed to create target directory "
                                    + newTarget);
                    }

                    // recurse
                    walkTree(file, newTarget, depth);
                } else if (file.isFile()
                        && (extensions == null || (!file.isHidden() && extensions
                                .contains(ext)))) {
                    // this is a file and we need to add trace into it !
                    actor.actOnFile(file, target, depth);
                }
            }
        } else {
            actor.actOnFile(source, target, depth);
        }
    }

    public static String maybeAppendSeparator(String name) {
        if (!name.endsWith("/") && !name.endsWith("\\"))
            name += "/";
        return name;
    }
}