hamlit 1.7.2 → 2.0.1

data/bin/ruby

@@ -0,0 +1,3 @@
+#!/bin/bash
+
+bundle exec ruby -Ilib:test -rtest_helper $@

data/bin/setup

@@ -0,0 +1,7 @@
+#!/bin/bash
+set -euo pipefail
+IFS=$'\n\t'
+
+bundle install
+
+# Do any other automated setup that you need to do here

data/bin/stackprof

@@ -0,0 +1,27 @@
+#!/usr/bin/env ruby
+
+require 'bundler/setup'
+require 'hamlit'
+require 'stackprof'
+
+def open_flamegraph(report)
+  temp = `mktemp /tmp/stackflame-XXXXXXXX`.strip
+  data_path = "#{temp}.js"
+  system("mv #{temp} #{data_path}")
+
+  File.open(data_path, 'w') do |f|
+    report.print_flamegraph(f)
+  end
+
+  viewer_path = File.join(`bundle show stackprof`.strip, 'lib/stackprof/flamegraph/viewer.html')
+  url = "file://#{viewer_path}?data=#{data_path}"
+  system(%Q[osascript -e 'open location "#{url}"'])
+end
+
+haml = File.read(ARGV.first)
+StackProf.start(mode: :wall, interval: 1, raw: false)
+Hamlit::Engine.new.call(haml)
+StackProf.stop
+
+report = StackProf::Report.new(StackProf.results)
+report.print_text(false)

data/{test → bin/test}

@@ -1,14 +1,14 @@
 #!/bin/bash
 
-set -e
-trap 'rm .ruby-version' 0
-
 VERSIONS=(
   2.0.0
-  2.1.6
-  2.2.2
+  2.1.7
+  2.2.3
 )
 
+set -e
+trap 'echo "${VERSIONS[2]}" > .ruby-version' 0
+
 function test_with() {
   version=$1
   rbenv local $version
@@ -16,13 +16,9 @@ function test_with() {
     bundle install
   fi
   ruby -v
-  rake spec
+  rake test
 }
 
 for version in ${VERSIONS[@]}; do
   test_with $version
 done
-
-rbenv local ${VERSIONS[2]}
-ruby -v
-rake rails:spec

data/ext/hamlit/extconf.rb

@@ -0,0 +1,14 @@
+require 'mkmf'
+
+houdini_dir = File.expand_path('./houdini', __dir__)
+$INCFLAGS << " -I#{houdini_dir}"
+$CFLAGS   << ' -Wall -Wextra'
+
+$srcs = %w[hamlit.c]
+Dir[File.join(houdini_dir, '*.c')].each do |path|
+  src = File.basename(path)
+  FileUtils.ln_s(path, src, force: true)
+  $srcs << src
+end
+
+create_makefile('hamlit/hamlit')

data/ext/hamlit/hamlit.c

@@ -0,0 +1,512 @@
+#include <ruby.h>
+#include <ruby/encoding.h>
+#include "houdini.h"
+#include "string.h"
+
+VALUE mAttributeBuilder, mObjectRef;
+static ID id_flatten, id_keys, id_parse, id_prepend, id_tr, id_uniq_bang;
+static ID id_data, id_equal, id_hyphen, id_space, id_underscore;
+static ID id_boolean_attributes, id_xhtml;
+
+static VALUE str_data()        { return rb_const_get(mAttributeBuilder, id_data); }
+static VALUE str_equal()       { return rb_const_get(mAttributeBuilder, id_equal); }
+static VALUE str_hyphen()      { return rb_const_get(mAttributeBuilder, id_hyphen); }
+static VALUE str_space()       { return rb_const_get(mAttributeBuilder, id_space); }
+static VALUE str_underscore()  { return rb_const_get(mAttributeBuilder, id_underscore); }
+
+static void
+delete_falsey_values(VALUE values)
+{
+  VALUE value;
+  long i;
+
+  for (i = RARRAY_LEN(values) - 1; 0 <= i; i--) {
+    value = rb_ary_entry(values, i);
+    if (!RTEST(value)) {
+      rb_ary_delete_at(values, i);
+    }
+  }
+}
+
+static int
+str_eq(VALUE str, const char *cstr)
+{
+  return strcmp(RSTRING_PTR(str), cstr) == 0;
+}
+
+static VALUE
+to_s(VALUE value)
+{
+  return rb_convert_type(value, T_STRING, "String", "to_s");
+}
+
+static VALUE
+hyphenate(VALUE str)
+{
+  long i;
+
+  if (OBJ_FROZEN(str)) str = rb_str_dup(str);
+
+  for (i = 0; i < RSTRING_LEN(str); i++) {
+    if (RSTRING_PTR(str)[i] == '_') {
+      rb_str_update(str, i, 1, str_hyphen());
+    }
+  }
+  return str;
+}
+
+static VALUE
+escape_html(VALUE str)
+{
+  gh_buf buf = GH_BUF_INIT;
+
+  Check_Type(str, T_STRING);
+
+  if (houdini_escape_html0(&buf, (const uint8_t *)RSTRING_PTR(str), RSTRING_LEN(str), 0)) {
+    str = rb_enc_str_new(buf.ptr, buf.size, rb_utf8_encoding());
+    gh_buf_free(&buf);
+  }
+
+  return str;
+}
+
+static VALUE
+escape_attribute(VALUE escape_attrs, VALUE str)
+{
+  if (RTEST(escape_attrs)) {
+    return escape_html(str);
+  } else {
+    return str;
+  }
+}
+
+static VALUE
+rb_escape_html(RB_UNUSED_VAR(VALUE self), VALUE value)
+{
+  return escape_html(to_s(value));
+}
+
+static VALUE
+hamlit_build_id(VALUE escape_attrs, VALUE values)
+{
+  VALUE attr_value;
+
+  values = rb_funcall(values, id_flatten, 0);
+  delete_falsey_values(values);
+
+  attr_value = rb_ary_join(values, str_underscore());
+  return escape_attribute(escape_attrs, attr_value);
+}
+
+static VALUE
+hamlit_build_single_class(VALUE escape_attrs, VALUE value)
+{
+  switch (TYPE(value)) {
+    case T_STRING:
+      break;
+    case T_ARRAY:
+      value = rb_funcall(value, id_flatten, 0);
+      delete_falsey_values(value);
+      value = rb_ary_join(value, str_space());
+      break;
+    default:
+      if (RTEST(value)) {
+        value = to_s(value);
+      } else {
+        return rb_str_new_cstr("");
+      }
+      break;
+  }
+  return escape_attribute(escape_attrs, value);
+}
+
+static VALUE
+hamlit_build_multi_class(VALUE escape_attrs, VALUE values)
+{
+  long i, j;
+  VALUE value, buf;
+
+  buf = rb_ary_new2(RARRAY_LEN(values));
+
+  for (i = 0; i < RARRAY_LEN(values); i++) {
+    value = rb_ary_entry(values, i);
+    switch (TYPE(value)) {
+      case T_STRING:
+        rb_ary_concat(buf, rb_str_split(value, " "));
+        break;
+      case T_ARRAY:
+        value = rb_funcall(value, id_flatten, 0);
+        delete_falsey_values(value);
+        for (j = 0; j < RARRAY_LEN(value); j++) {
+          rb_ary_push(buf, to_s(rb_ary_entry(value, j)));
+        }
+        break;
+      default:
+        if (RTEST(value)) {
+          rb_ary_push(buf, to_s(value));
+        }
+        break;
+    }
+  }
+
+  rb_ary_sort_bang(buf);
+  rb_funcall(buf, id_uniq_bang, 0);
+
+  return escape_attribute(escape_attrs, rb_ary_join(buf, str_space()));
+}
+
+static VALUE
+hamlit_build_class(VALUE escape_attrs, VALUE array)
+{
+  if (RARRAY_LEN(array) == 1) {
+    return hamlit_build_single_class(escape_attrs, rb_ary_entry(array, 0));
+  } else {
+    return hamlit_build_multi_class(escape_attrs, array);
+  }
+}
+
+static int
+merge_data_attrs_i(VALUE key, VALUE value, VALUE merged)
+{
+  if (NIL_P(key)) {
+    rb_hash_aset(merged, str_data(), value);
+  } else {
+    key = rb_str_concat(rb_str_new_cstr("data-"), to_s(key));
+    rb_hash_aset(merged, key, value);
+  }
+  return ST_CONTINUE;
+}
+
+static VALUE
+merge_data_attrs(VALUE values)
+{
+  long i;
+  VALUE value, merged = rb_hash_new();
+
+  for (i = 0; i < RARRAY_LEN(values); i++) {
+    value = rb_ary_entry(values, i);
+    switch (TYPE(value)) {
+      case T_HASH:
+        rb_hash_foreach(value, merge_data_attrs_i, merged);
+        break;
+      default:
+        rb_hash_aset(merged, str_data(), value);
+        break;
+    }
+  }
+  return merged;
+}
+
+struct flatten_data_attrs_i2_arg {
+  VALUE flattened;
+  VALUE key;
+};
+
+static int
+flatten_data_attrs_i2(VALUE k, VALUE v, VALUE ptr)
+{
+  VALUE key;
+  struct flatten_data_attrs_i2_arg *arg = (struct flatten_data_attrs_i2_arg *)ptr;
+
+  if (!RTEST(v)) return ST_CONTINUE;
+
+  if (k == Qnil) {
+    rb_hash_aset(arg->flattened, arg->key, v);
+  } else {
+    key = rb_str_dup(arg->key);
+    rb_str_cat(key, "-", 1);
+    rb_str_concat(key, to_s(k));
+
+    rb_hash_aset(arg->flattened, key, v);
+  }
+  return ST_CONTINUE;
+}
+
+static VALUE flatten_data_attrs(VALUE attrs);
+
+static int
+flatten_data_attrs_i(VALUE key, VALUE value, VALUE flattened)
+{
+  struct flatten_data_attrs_i2_arg arg;
+  key = hyphenate(to_s(key));
+
+  switch (TYPE(value)) {
+    case T_HASH:
+      value = flatten_data_attrs(value);
+      arg.key       = key;
+      arg.flattened = flattened;
+      rb_hash_foreach(value, flatten_data_attrs_i2, (VALUE)(&arg));
+      break;
+    default:
+      if (RTEST(value)) rb_hash_aset(flattened, key, value);
+      break;
+  }
+  return ST_CONTINUE;
+}
+
+static VALUE
+flatten_data_attrs(VALUE attrs)
+{
+  VALUE flattened = rb_hash_new();
+  rb_hash_foreach(attrs, flatten_data_attrs_i, flattened);
+
+  return flattened;
+}
+
+static VALUE
+hamlit_build_data(VALUE escape_attrs, VALUE quote, VALUE values)
+{
+  long i;
+  VALUE attrs, buf, keys, key, value;
+
+  attrs = merge_data_attrs(values);
+  attrs = flatten_data_attrs(attrs);
+  keys  = rb_ary_sort_bang(rb_funcall(attrs, id_keys, 0));
+  buf   = rb_str_new("", 0);
+
+  for (i = 0; i < RARRAY_LEN(keys); i++) {
+    key   = rb_ary_entry(keys, i);
+    value = rb_hash_aref(attrs, key);
+
+    switch (value) {
+      case Qtrue:
+        rb_str_concat(buf, str_space());
+        rb_str_concat(buf, key);
+        break;
+      case Qnil:
+        break; // noop
+      case Qfalse:
+        break; // noop
+      default:
+        rb_str_concat(buf, str_space());
+        rb_str_concat(buf, key);
+        rb_str_concat(buf, str_equal());
+        rb_str_concat(buf, quote);
+        rb_str_concat(buf, escape_attribute(escape_attrs, to_s(value)));
+        rb_str_concat(buf, quote);
+        break;
+    }
+  }
+
+  return buf;
+}
+
+static VALUE
+parse_object_ref(VALUE object_ref)
+{
+  return rb_funcall(mObjectRef, id_parse, 1, object_ref);
+}
+
+static int
+merge_all_attrs_i(VALUE key, VALUE value, VALUE merged)
+{
+  VALUE array;
+
+  key = to_s(key);
+  if (str_eq(key, "id") || str_eq(key, "class") || str_eq(key, "data")) {
+    array = rb_hash_aref(merged, key);
+    if (NIL_P(array)) {
+      array = rb_ary_new2(1);
+      rb_hash_aset(merged, key, array);
+    }
+    rb_ary_push(array, value);
+  } else {
+    rb_hash_aset(merged, key, value);
+  }
+  return ST_CONTINUE;
+}
+
+static VALUE
+merge_all_attrs(VALUE hashes)
+{
+  long i;
+  VALUE hash, merged = rb_hash_new();
+
+  for (i = 0; i < RARRAY_LEN(hashes); i++) {
+    hash = rb_ary_entry(hashes, i);
+    rb_hash_foreach(hash, merge_all_attrs_i, merged);
+  }
+  return merged;
+}
+
+int
+is_boolean_attribute(VALUE key)
+{
+  VALUE boolean_attributes;
+  if (str_eq(rb_str_substr(key, 0, 5), "data-")) return 1;
+
+  boolean_attributes = rb_const_get(mAttributeBuilder, id_boolean_attributes);
+  return RTEST(rb_ary_includes(boolean_attributes, key));
+}
+
+void
+hamlit_build_for_id(VALUE escape_attrs, VALUE quote, VALUE buf, VALUE values)
+{
+  rb_str_cat(buf, " id=", 4);
+  rb_str_concat(buf, quote);
+  rb_str_concat(buf, hamlit_build_id(escape_attrs, values));
+  rb_str_concat(buf, quote);
+}
+
+void
+hamlit_build_for_class(VALUE escape_attrs, VALUE quote, VALUE buf, VALUE values)
+{
+  rb_str_cat(buf, " class=", 7);
+  rb_str_concat(buf, quote);
+  rb_str_concat(buf, hamlit_build_class(escape_attrs, values));
+  rb_str_concat(buf, quote);
+}
+
+void
+hamlit_build_for_data(VALUE escape_attrs, VALUE quote, VALUE buf, VALUE values)
+{
+  rb_str_concat(buf, hamlit_build_data(escape_attrs, quote, values));
+}
+
+void
+hamlit_build_for_others(VALUE escape_attrs, VALUE quote, VALUE buf, VALUE key, VALUE value)
+{
+  rb_str_cat(buf, " ", 1);
+  rb_str_concat(buf, key);
+  rb_str_cat(buf, "=", 1);
+  rb_str_concat(buf, quote);
+  rb_str_concat(buf, escape_attribute(escape_attrs, to_s(value)));
+  rb_str_concat(buf, quote);
+}
+
+void
+hamlit_build_for_boolean(VALUE escape_attrs, VALUE quote, VALUE format, VALUE buf, VALUE key, VALUE value)
+{
+  switch (value) {
+    case Qtrue:
+      rb_str_cat(buf, " ", 1);
+      rb_str_concat(buf, key);
+      if ((TYPE(format) == T_SYMBOL || TYPE(format) == T_STRING) && rb_to_id(format) == id_xhtml) {
+        rb_str_cat(buf, "=", 1);
+        rb_str_concat(buf, quote);
+        rb_str_concat(buf, key);
+        rb_str_concat(buf, quote);
+      }
+      break;
+    case Qfalse:
+      break; // noop
+    case Qnil:
+      break; // noop
+    default:
+      hamlit_build_for_others(escape_attrs, quote, buf, key, value);
+      break;
+  }
+}
+
+static VALUE
+hamlit_build(VALUE escape_attrs, VALUE quote, VALUE format, VALUE object_ref, VALUE hashes)
+{
+  long i;
+  VALUE attrs, buf, key, keys, value;
+
+  if (!NIL_P(object_ref)) rb_ary_push(hashes, parse_object_ref(object_ref));
+  attrs = merge_all_attrs(hashes);
+  buf   = rb_str_new("", 0);
+  keys  = rb_ary_sort_bang(rb_funcall(attrs, id_keys, 0));
+
+  for (i = 0; i < RARRAY_LEN(keys); i++) {
+    key   = rb_ary_entry(keys, i);
+    value = rb_hash_aref(attrs, key);
+    if (str_eq(key, "id")) {
+      hamlit_build_for_id(escape_attrs, quote, buf, value);
+    } else if (str_eq(key, "class")) {
+      hamlit_build_for_class(escape_attrs, quote, buf, value);
+    } else if (str_eq(key, "data")) {
+      hamlit_build_for_data(escape_attrs, quote, buf, value);
+    } else if (is_boolean_attribute(key)) {
+      hamlit_build_for_boolean(escape_attrs, quote, format, buf, key, value);
+    } else {
+      hamlit_build_for_others(escape_attrs, quote, buf, key, value);
+    }
+  }
+
+  return buf;
+}
+
+static VALUE
+rb_hamlit_build_id(int argc, VALUE *argv, RB_UNUSED_VAR(VALUE self))
+{
+  VALUE array;
+
+  rb_check_arity(argc, 1, UNLIMITED_ARGUMENTS);
+  rb_scan_args(argc - 1, argv + 1, "*", &array);
+
+  return hamlit_build_id(argv[0], array);
+}
+
+static VALUE
+rb_hamlit_build_class(int argc, VALUE *argv, RB_UNUSED_VAR(VALUE self))
+{
+  VALUE array;
+
+  rb_check_arity(argc, 1, UNLIMITED_ARGUMENTS);
+  rb_scan_args(argc - 1, argv + 1, "*", &array);
+
+  return hamlit_build_class(argv[0], array);
+}
+
+static VALUE
+rb_hamlit_build_data(int argc, VALUE *argv, RB_UNUSED_VAR(VALUE self))
+{
+  VALUE array;
+
+  rb_check_arity(argc, 2, UNLIMITED_ARGUMENTS);
+  rb_scan_args(argc - 2, argv + 2, "*", &array);
+
+  return hamlit_build_data(argv[0], argv[1], array);
+}
+
+static VALUE
+rb_hamlit_build(int argc, VALUE *argv, RB_UNUSED_VAR(VALUE self))
+{
+  VALUE array;
+
+  rb_check_arity(argc, 4, UNLIMITED_ARGUMENTS);
+  rb_scan_args(argc - 4, argv + 4, "*", &array);
+
+  return hamlit_build(argv[0], argv[1], argv[2], argv[3], array);
+}
+
+void
+Init_hamlit(void)
+{
+  VALUE mHamlit, mUtils;
+
+  mHamlit           = rb_define_module("Hamlit");
+  mObjectRef        = rb_define_module_under(mHamlit, "ObjectRef");
+  mUtils            = rb_define_module_under(mHamlit, "Utils");
+  mAttributeBuilder = rb_define_module_under(mHamlit, "AttributeBuilder");
+
+  rb_define_singleton_method(mUtils, "escape_html", rb_escape_html, 1);
+  rb_define_singleton_method(mAttributeBuilder, "build", rb_hamlit_build, -1);
+  rb_define_singleton_method(mAttributeBuilder, "build_id", rb_hamlit_build_id, -1);
+  rb_define_singleton_method(mAttributeBuilder, "build_class", rb_hamlit_build_class, -1);
+  rb_define_singleton_method(mAttributeBuilder, "build_data", rb_hamlit_build_data, -1);
+
+  id_flatten   = rb_intern("flatten");
+  id_keys      = rb_intern("keys");
+  id_parse     = rb_intern("parse");
+  id_prepend   = rb_intern("prepend");
+  id_tr        = rb_intern("tr");
+  id_uniq_bang = rb_intern("uniq!");
+
+  id_data       = rb_intern("DATA");
+  id_equal      = rb_intern("EQUAL");
+  id_hyphen     = rb_intern("HYPHEN");
+  id_space      = rb_intern("SPACE");
+  id_underscore = rb_intern("UNDERSCORE");
+
+  id_boolean_attributes = rb_intern("BOOLEAN_ATTRIBUTES");
+  id_xhtml = rb_intern("xhtml");
+
+  rb_const_set(mAttributeBuilder, id_data,       rb_obj_freeze(rb_str_new_cstr("data")));
+  rb_const_set(mAttributeBuilder, id_equal,      rb_obj_freeze(rb_str_new_cstr("=")));
+  rb_const_set(mAttributeBuilder, id_hyphen,     rb_obj_freeze(rb_str_new_cstr("-")));
+  rb_const_set(mAttributeBuilder, id_space,      rb_obj_freeze(rb_str_new_cstr(" ")));
+  rb_const_set(mAttributeBuilder, id_underscore, rb_obj_freeze(rb_str_new_cstr("_")));
+}