util/dataforms.lua

   1 -- Prosody IM
   2 -- Copyright (C) 2008-2009 Matthew Wild
   3 -- Copyright (C) 2008-2009 Waqas Hussain
   4 --
   5 -- This project is MIT/X11 licensed. Please see the
   6 -- COPYING file in the source package for more information.
   7 --
   8
   9 local setmetatable = setmetatable;
  10 local pairs, ipairs = pairs, ipairs;
  11 local tostring, type = tostring, type;
  12 local t_concat = table.concat;
  13 local st = require "util.stanza";
  14
  15 module "dataforms"
  16
  17 local xmlns_forms = 'jabber:x:data';
  18
  19 local form_t = {};
  20 local form_mt = { __index = form_t };
  21
  22 function new(layout)
  23         return setmetatable(layout, form_mt);
  24 end
  25
  26 function form_t.form(layout, data)
  27         local form = st.stanza("x", { xmlns = xmlns_forms, type = "form" });
  28         if layout.title then
  29                 form:tag("title"):text(layout.title):up();
  30         end
  31         if layout.instructions then
  32                 form:tag("instructions"):text(layout.instructions):up();
  33         end
  34         for n, field in ipairs(layout) do
  35                 local field_type = field.type or "text-single";
  36                 -- Add field tag
  37                 form:tag("field", { type = field_type, var = field.name, label = field.label });
  38
  39                 local value = (data and data[field.name]) or field.value;
  40
  41                 if value then
  42                         -- Add value, depending on type
  43                         if field_type == "hidden" then
  44                                 if type(value) == "table" then
  45                                         -- Assume an XML snippet
  46                                         form:tag("value")
  47                                                 :add_child(value)
  48                                                 :up();
  49                                 else
  50                                         form:tag("value"):text(tostring(value)):up();
  51                                 end
  52                         elseif field_type == "boolean" then
  53                                 form:tag("value"):text((value and "1") or "0"):up();
  54                         elseif field_type == "fixed" then
  55
  56                         elseif field_type == "jid-multi" then
  57                                 for _, jid in ipairs(value) do
  58                                         form:tag("value"):text(jid):up();
  59                                 end
  60                         elseif field_type == "jid-single" then
  61                                 form:tag("value"):text(value):up();
  62                         elseif field_type == "text-single" or field_type == "text-private" then
  63                                 form:tag("value"):text(value):up();
  64                         elseif field_type == "text-multi" then
  65                                 -- Split into multiple <value> tags, one for each line
  66                                 for line in value:gmatch("([^\r\n]+)\r?\n*") do
  67                                         form:tag("value"):text(line):up();
  68                                 end
  69                         elseif field_type == "list-single" then
  70                                 for _, val in ipairs(value) do
  71                                         if type(val) == "table" then
  72                                                 form:tag("option", { label = val.label }):tag("value"):text(val.value):up():up();
  73                                         else
  74                                                 form:tag("option", { label= val }):tag("value"):text(tostring(val)):up():up();
  75                                         end
  76                                 end
  77                         end
  78                 end
  79
  80                 if field.required then
  81                         form:tag("required"):up();
  82                 end
  83
  84                 -- Jump back up to list of fields
  85                 form:up();
  86         end
  87         return form;
  88 end
  89
  90 local field_readers = {};
  91
  92 function form_t.data(layout, stanza)
  93         local data = {};
  94
  95         for field_tag in stanza:childtags() do
  96                 local field_type;
  97                 for n, field in ipairs(layout) do
  98                         if field.name == field_tag.attr.var then
  99                                 field_type = field.type;
 100                                 break;
 101                         end
 102                 end
 103
 104                 local reader = field_readers[field_type];
 105                 if reader then
 106                         data[field_tag.attr.var] = reader(field_tag);
 107                 end
 108
 109         end
 110         return data;
 111 end
 112
 113 field_readers["text-single"] =
 114         function (field_tag)
 115                 local value = field_tag:child_with_name("value");
 116                 if value then
 117                         return value[1];
 118                 end
 119         end
 120
 121 field_readers["text-private"] =
 122         field_readers["text-single"];
 123
 124 field_readers["jid-single"] =
 125         field_readers["text-single"];
 126
 127 field_readers["jid-multi"] =
 128         function (field_tag)
 129                 local result = {};
 130                 for value_tag in field_tag:childtags() do
 131                         if value_tag.name == "value" then
 132                                 result[#result+1] = value_tag[1];
 133                         end
 134                 end
 135                 return result;
 136         end
 137
 138 field_readers["text-multi"] =
 139         function (field_tag)
 140                 local result = {};
 141                 for value_tag in field_tag:childtags() do
 142                         if value_tag.name == "value" then
 143                                 result[#result+1] = value_tag[1];
 144                         end
 145                 end
 146                 return t_concat(result, "\n");
 147         end
 148
 149 field_readers["list-single"] =
 150         field_readers["text-single"];
 151
 152 field_readers["boolean"] =
 153         function (field_tag)
 154                 local value = field_tag:child_with_name("value");
 155                 if value then
 156                         if value[1] == "1" or value[1] == "true" then
 157                                 return true;
 158                         else
 159                                 return false;
 160                         end
 161                 end
 162         end
 163
 164 field_readers["hidden"] =
 165         function (field_tag)
 166                 local value = field_tag:child_with_name("value");
 167                 if value then
 168                         return value[1];
 169                 end
 170         end
 171
 172 return _M;
 173
 174
 175 --[=[
 176
 177 Layout:
 178 {
 179
 180         title = "MUC Configuration",
 181         instructions = [[Use this form to configure options for this MUC room.]],
 182
 183         { name = "FORM_TYPE", type = "hidden", required = true };
 184         { name = "field-name", type = "field-type", required = false };
 185 }
 186
 187
 188 --]=]