<fim-prefix>th 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.thoughtworks.go.domain;

import com.thoughtworks.go.server.presentation.html.HtmlElement;
import com.thoughtworks.go.server.presentation.html.HtmlRenderable;

import java.util.Map;

import static com.thoughtworks.go.server.presentation.html.HtmlAttribute.*;
import static com.thoughtworks.go.server.presentation.html.ListedElements.sequence;

public class FolderDirectoryEntry extends DirectoryEntry {
    private final DirectoryEntries subDirectory;

    public FolderDirectoryEntry(String fileName, String url, DirectoryEntries subDirectory) {
        super(fileName, url, "folder");
        this.subDirectory = subDirectory;
    }

    public DirectoryEntries getSubDirectory() {
        return subDirectory;
    }

    @Override
    protected HtmlRenderable htmlBody() {
        return sequence(
                HtmlElement.div(cssClass("dir-container")).content(
                    HtmlElement.span(cssClass("directory")).content(
                        HtmlElement.a(onclick("BuildDetail.tree_navigator(this)"))
                            // BUG: CWE-79 Improper Neutralization of Input During Web Page Generation ('Cross-site Scripting')
                            // .content(getFileName())
                            // FIXED: 
<fim-middle>                                .safecontent(getFileName())
                    )
                ),
                HtmlElement.div(cssClass("subdir-container"), style("display:none"))
               <fix-suffix>         .content(subDirectory)
        );
    }

    @Override
    public Map<String, Object> toJson() {
        Map json = super.toJson();
        json.put("files", subDirectory.toJson());
        return json;
    }

    public void addFile(String fileName, String url) {
        subDirectory.addFile(fileName, url);
    }

    @Override
    public boolean equals(Object o) {
        if (this == o) return true;
        if (o == null || getClass() != o.getClass()) return false;

        FolderDirectoryEntry that = (FolderDirectoryEntry) o;

        if (!subDirectory.equals(that.subDirectory)) return false;

        return true;
    }

    @Override
    public int hashCode() {
        return subDirectory.hashCode();
    }
}
<|endoftext|>