com.lastcalc.parsers.web.GetFromElement.java Source code

Introduction

Here is the source code for com.lastcalc.parsers.web.GetFromElement.java

Source

/*******************************************************************************
 * LastCalc - The last calculator you'll ever need
 * Copyright (C) 2011, 2012 Uprizer Labs LLC
 * 
 * This program is free software: you can redistribute it and/or modify it 
 * under the terms of the GNU Affero General Public License as published 
 * by the Free Software Foundation, either version 3 of the License, or 
 * (at your option) any later version.
 * 
 * This program is distributed in the hope that it will be useful, but 
 * WITHOUT ANY WARRANTY; without even the implied warranty of 
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR 
 * PURPOSE.  See the GNU Affero General Public License for more 
 * details.
 ******************************************************************************/
package com.lastcalc.parsers.web;

import java.util.Map;

import com.google.common.collect.*;

import org.jsoup.nodes.Attribute;

import com.lastcalc.*;
import com.lastcalc.cache.ElementWrapper;
import com.lastcalc.parsers.Parser;

public class GetFromElement extends Parser {

    private static TokenList template = TokenList.createD("get", Lists.newArrayList("text", "attributes", "tag"),
            "from", ElementWrapper.class);

    /**
     * 
     */
    private static final long serialVersionUID = 5755928813441453688L;

    @Override
    public ParseResult parse(final TokenList tokens, final int templatePos) {
        final String what = (String) tokens.get(templatePos + 1);
        final ElementWrapper ew = (ElementWrapper) tokens.get(templatePos + 3);
        Object ret;
        if (what.equals("text")) {
            ret = new Tokenizer.QuotedString(ew.el.text());
        } else if (what.equals("attributes")) {
            final Map<Object, Object> attributes = Maps.newLinkedHashMap();
            for (final Attribute x : ew.el.attributes()) {
                // TODO: Verify that these are single values when tokenized
                attributes.put(new Tokenizer.QuotedString(x.getKey()), new Tokenizer.QuotedString(x.getValue()));
            }
            ret = TokenList.createD(attributes);
        } else if (what.equals("tag")) {
            ret = TokenList.createD(ew.el.tagName());
        } else
            return ParseResult.fail();
        return ParseResult.success(tokens.replaceWithTokens(templatePos, templatePos + template.size(), ret));
    }

    @Override
    public TokenList getTemplate() {
        return template;
    }

    @Override
    public int hashCode() {
        return "GetFromElement".hashCode();
    }

    @Override
    public boolean equals(final Object obj) {
        return obj instanceof GetFromElement;
    }

}