com.google.jstestdriver.html.HtmlDocNodeFactory.java Source code

Introduction

Here is the source code for com.google.jstestdriver.html.HtmlDocNodeFactory.java

Source

/*
 * Copyright 2009 Google Inc.
 * 
 * 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.google.jstestdriver.html;

import com.google.common.collect.Lists;
import com.google.jstestdriver.html.HtmlDocParser.NodeFactory;
import com.google.jstestdriver.token.BufferedTokenStream;
import com.google.jstestdriver.token.ConcreteToken;
import com.google.jstestdriver.token.Node;
import com.google.jstestdriver.token.Nodes;
import com.google.jstestdriver.token.Token;

import java.util.List;

public class HtmlDocNodeFactory implements NodeFactory {
    private static final Token DOC_START = ConcreteToken.from("/*:DOC");
    private static final Token EQUALS = ConcreteToken.from("=");
    private static final Token END_COMMENT = ConcreteToken.from("*/");

    private final CreateNodeStrategy createStrategy;

    public HtmlDocNodeFactory(CreateNodeStrategy createStrategy) {
        this.createStrategy = createStrategy;
    }

    public void create(BufferedTokenStream stream, Nodes nodes) {
        Token id = null;
        List<Token> html = Lists.newLinkedList();

        if (!DOC_START.equals(stream.read())) {
            stream.reset();
            return;
        }

        id = stream.read();

        if (!EQUALS.equals(stream.read())) {
            stream.reset();
            return;
        }

        while (stream.available()) {
            Token token = stream.read();
            if (END_COMMENT.equals(token)) {
                break;
            }
            html.add(token);
        }

        stream.mark();
        nodes.add(createStrategy.create(id, html));
    }

    public interface CreateNodeStrategy {
        public Node create(Token id, List<Token> html);
    }
}