/* Copyright (c) 2020 Contrast Security, Inc.  See
 * https://www.contrastsecurity.com/enduser-terms-0317a for more details. */

#include "cs__assess_string_interpolation26.h"
#include "../cs__common/cs__common.h"
#include <ruby.h>

static VALUE rb_str_concat_literals_hook(size_t num, VALUE *strary) {
    VALUE result = rb_str_concat_literals_original(num, strary);
    VALUE rb_params = rb_ary_new_from_values((int)num, strary);
    rb_funcall(string_propagator, track_interpolation, 2, rb_params, result);
    return result;
}

static int install_hooks() {
    rb_str_concat_literals_original = rb_str_concat_literals;
    patch_via_funchook(&rb_str_concat_literals_original,
                       &rb_str_concat_literals_hook);

    return 0;
}

void Init_cs__assess_string_interpolation26(void) {
    string_propagator =
        rb_define_class_under(core_assess, "StringPropagator", rb_cObject);
    track_interpolation = rb_intern("track_interpolation");
    install_hooks();
}