TextProducerInternal.java

/*
 * Copyright (c) 2013. Codearte
 */
package eu.codearte.jfairy.producer.text;

import static com.google.common.collect.Lists.*;
import static java.util.Arrays.asList;
import static org.apache.commons.lang3.StringUtils.*;

import java.util.List;
import javax.inject.Inject;

import com.google.common.base.Joiner;
import com.google.common.base.Splitter;
import eu.codearte.jfairy.data.DataMaster;
import eu.codearte.jfairy.producer.BaseProducer;

class TextProducerInternal {

    private static final String LOREM_IPSUM = "loremIpsum";

    private static final String TEXT = "text";

    private static final String ALPHABET = "alphabet";

    private static final int WORD_COUNT_PRECISION_IN_SENTENCE = 6;

    private final BaseProducer baseProducer;

    private final String loremIpsum;

    private final String text;

    private final List<String> words;

    private final String alphabet;

    private final int maxAlphabetIndex;

    private final List<String> latinWords;

    @Inject
    public TextProducerInternal(DataMaster dataMaster, BaseProducer baseProducer) {
        this.baseProducer = baseProducer;
        loremIpsum = dataMaster.getString(LOREM_IPSUM);
        text = dataMaster.getString(TEXT);
        words = asList(split(text, ' '));
        latinWords = asList(split(loremIpsum, ' '));
        alphabet = dataMaster.getString(ALPHABET);
        maxAlphabetIndex = alphabet.length() - 1;
    }

    public String loremIpsum() {
        return loremIpsum;
    }

    public String rawWords(List<String> words, int count, int precision) {
        List<String> result = readRawWords(words, count, precision);
        return TextUtils.joinWithSpace(result);
    }

    public String cleanLatinWords(int count) {
        return cleanWords(latinWords, count);
    }

    public String cleanWords(int count) {
        return cleanWords(words, count);
    }

    private String cleanWords(List<String> words, int count) {
        List<String> result = newArrayList();
        for (String part : readRawWords(words, count, 0)) {
            result.add(uncapitalize(replaceChars(part, "., ", "")));
        }
        return TextUtils.joinWithSpace(result);
    }

    public String randomString(int charsCount) {
        StringBuilder sb = new StringBuilder(charsCount);
        for (int i = 0; i < charsCount; i++) {
            sb.append(alphabet.charAt(baseProducer.randomInt(maxAlphabetIndex)));
        }
        return sb.toString();
    }

    private List<String> readRawWords(List<String> words, int count, int precision) {
        return baseProducer.randomElements(words, baseProducer.randomBetween(count, count + precision));
    }

    public String text() {
        return text;
    }

    public String sentence(int wordCount) {
        return sentence(words, wordCount);
    }

    public String latinSentence(int wordCount) {
        return sentence(latinWords, wordCount);
    }

    private String sentence(List<String> words, int wordCount) {
        String randomWords = rawWords(words, wordCount, WORD_COUNT_PRECISION_IN_SENTENCE);
        List<String> sentences = newArrayList();
        for (String sentence : Splitter.on(". ").split(randomWords)) {
            sentences.add(capitalize(sentence));
        }
        String sentence = capitalize(Joiner.on(". ").join(sentences));
        sentence = removeEnd(sentence, ",");
        if (!endsWith(sentence, ".")) {
            sentence += ".";
        }
        return sentence;
    }
}