src/main/java/org/zwobble/couscous/interpreter/values/StringInterpreterValue.java


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

package org.zwobble.couscous.interpreter.values;

import java.util.Optional;

import org.zwobble.couscous.interpreter.NoSuchField;
import org.zwobble.couscous.values.IntegerValue;
import org.zwobble.couscous.values.PrimitiveValue;
import org.zwobble.couscous.values.PrimitiveValues;
import org.zwobble.couscous.values.StringValue;

import static java.util.Arrays.asList;

import lombok.Value;
import lombok.val;

@Value
public class StringInterpreterValue implements InterpreterValue {
    public static final ConcreteType TYPE = ConcreteType.builder(StringInterpreterValue.class, StringValue.REF)
        .method("length", asList(),
            (environment, arguments) ->
                new IntegerInterpreterValue(arguments.getReceiver().value.length()))
        
        .method("substring", asList(IntegerValue.REF, IntegerValue.REF),
            (environment, arguments) -> {
                val startIndex = (IntegerInterpreterValue)arguments.get(0);
                val endIndex = (IntegerInterpreterValue)arguments.get(1);
                return new StringInterpreterValue(arguments.getReceiver().value.substring(startIndex.getValue(), endIndex.getValue()));
            })
        
        .build();
    
    String value;

    @Override
    public ConcreteType getType() {
        return TYPE;
    }

    @Override
    public Optional<PrimitiveValue> toPrimitiveValue() {
        return Optional.of(PrimitiveValues.value(value));
    }

    @Override
    public InterpreterValue getField(String fieldName) {
        throw new NoSuchField(fieldName);
    }
}