/*
 * Copyright (C) 2016-2022 ActionTech.
 * License: http://www.gnu.org/licenses/gpl.html GPL version 2 or higher.
 */

package com.actiontech.dble.plan.common.item.function.strfunc;

import com.actiontech.dble.plan.common.item.Item;
import com.actiontech.dble.plan.common.item.function.ItemFunc;

import java.util.List;


public class ItemFuncQuote extends ItemStrFunc {

    public ItemFuncQuote(List<Item> args, int charsetIndex) {
        super(args, charsetIndex);
    }

    @Override
    public final String funcName() {
        return "quote";
    }

    @Override
    public String valStr() {
        String old = args.get(0).valStr();
        if (args.get(0).isNull()) {
            this.nullValue = true;
            return EMPTY;
        }
        StringBuilder newSb = new StringBuilder();
        newSb.append("'");
        for (char c : old.toCharArray()) {
            switch (c) {
                case 0:
                case '\032':
                case '\'':
                case '\\':
                    newSb.append('\\').append(c);
                    break;
                default:
                    newSb.append(c);
                    break;
            }
        }
        newSb.append("'");
        return newSb.toString();
    }

    @Override
    public ItemFunc nativeConstruct(List<Item> realArgs) {
        return new ItemFuncQuote(realArgs, charsetIndex);
    }
}
