#include <libhbsdcontrol.h>
#include <unistd.h>
#include "include/feature.h"
static VALUE get_rb_eError(void);

/**
 * BSD::Control::Feature#set!
 **/
VALUE
feature_set(VALUE self, VALUE rb_path, VALUE rb_state)
{
  Check_Type(rb_path, T_STRING);
  Check_Type(rb_state, T_FIXNUM);
  if (getuid() != 0) {
    rb_raise(get_rb_eError(), "This operation requires root privileges.");
  } else if (rb_funcall(rb_cFile, rb_intern("exist?"), 1, rb_path) == Qfalse) {
    rb_raise(get_rb_eError(), "The given path does not exist.");
  } else {
    int r;
    VALUE rb_name;
    rb_name = rb_funcall(self, rb_intern("name"), 0);
    Check_Type(rb_name, T_STRING);
    r = hbsdcontrol_set_feature_state(
      RSTRING_PTR(rb_path),
      RSTRING_PTR(rb_name),
      NUM2INT(rb_state)
      );
    if (r == 0) {
      return (Qtrue);
    } else {
      rb_raise(get_rb_eError(), "hbsdcontrol_set_feature_state failed");
    }
  }
}


static VALUE
get_rb_eError(void)
{
  VALUE rb_mBSD = rb_const_get(rb_cObject, rb_intern("BSD")),
    rb_mControl = rb_const_get(rb_mBSD, rb_intern("Control")),
    rb_eError = rb_const_get(rb_mControl, rb_intern("Error"));
  return (rb_eError);
}