r8331: added split(), join() and FileLoad() functions to ejs.
authorAndrew Tridgell <tridge@samba.org>
Mon, 11 Jul 2005 21:52:06 +0000 (21:52 +0000)
committerGerald (Jerry) Carter <jerry@samba.org>
Wed, 10 Oct 2007 18:20:06 +0000 (13:20 -0500)
this gets me most of the way through a ejs provision script
(This used to be commit 0ef8f0282e0490328bbcdc8c6daff3f8fac99087)

source4/scripting/ejs/smbcalls_string.c

index 7817de1b0df4fd8b0a46a629b9aa5490431c8d03..b52a6d2182b4fc78504d3830375597e9299e422b 100644 (file)
@@ -59,6 +59,106 @@ static int ejs_strupper(MprVarHandle eid, int argc, char **argv)
        return 0;
 }
 
+/*
+  usage:
+     list = split(".", "a.foo.bar");
+
+  NOTE: does not take a regular expression, unlink perl split()
+*/
+static int ejs_split(MprVarHandle eid, int argc, char **argv)
+{
+       const char *separator;
+       char *s, *p;
+       struct MprVar ret;
+       int count = 0;
+       TALLOC_CTX *tmp_ctx = talloc_new(mprMemCtx());
+       if (argc != 2) {
+               ejsSetErrorMsg(eid, "split invalid arguments");
+               return -1;
+       }
+       separator = argv[0];
+       s = argv[1];
+
+       ret = mprObject("list");
+
+       while ((p = strstr(s, separator))) {
+               char *s2 = talloc_strndup(tmp_ctx, s, (int)(p-s));
+               mprAddArray(&ret, count++, mprString(s2));
+               talloc_free(s2);
+               s = p + strlen(separator);
+       }
+       if (*s) {
+               mprAddArray(&ret, count++, mprString(s));
+       }
+       talloc_free(tmp_ctx);
+       mpr_Return(eid, ret);
+       return 0;
+}
+
+
+/*
+  usage:
+     str = join("DC=", list);
+*/
+static int ejs_join(MprVarHandle eid, int argc, struct MprVar **argv)
+{
+       int i;
+       const char *separator;
+       char *ret = NULL;
+       const char **list;
+       TALLOC_CTX *tmp_ctx = talloc_new(mprMemCtx());
+       if (argc != 2 ||
+           argv[0]->type != MPR_TYPE_STRING ||
+           argv[1]->type != MPR_TYPE_OBJECT) {
+               ejsSetErrorMsg(eid, "join invalid arguments");
+               return -1;
+       }
+
+       separator = mprToString(argv[0]);
+       list      = mprToList(tmp_ctx, argv[1]);
+
+       if (list == NULL || list[0] == NULL) {
+               talloc_free(tmp_ctx);
+               mpr_Return(eid, mprString(NULL));
+               return 0;
+       }
+       
+       ret = talloc_strdup(tmp_ctx, list[0]);
+       if (ret == NULL) {
+               goto failed;
+       }
+       for (i=1;list[i];i++) {
+               ret = talloc_asprintf_append(ret, "%s%s", separator, list[i]);
+               if (ret == NULL) {
+                       goto failed;
+               }
+       }
+       mpr_Return(eid, mprString(ret));
+       talloc_free(tmp_ctx);
+       return 0;
+failed:
+       ejsSetErrorMsg(eid, "out of memory");
+       return -1;
+}
+
+/*
+  load a file as a string
+  usage:
+     string = FileLoad(filename);
+*/
+static int ejs_FileLoad(MprVarHandle eid, int argc, char **argv)
+{
+       char *s;
+       if (argc != 1) {
+               ejsSetErrorMsg(eid, "FileLoad invalid arguments");
+               return -1;
+       }
+
+       s = file_load(argv[0], NULL, mprMemCtx());
+       mpr_Return(eid, mprString(s));
+       talloc_free(s);
+       return 0;
+}
 
 /*
   setup C functions that be called from ejs
@@ -67,4 +167,7 @@ void smb_setup_ejs_string(void)
 {
        ejsDefineStringCFunction(-1, "strlower", ejs_strlower, NULL, MPR_VAR_SCRIPT_HANDLE);
        ejsDefineStringCFunction(-1, "strupper", ejs_strupper, NULL, MPR_VAR_SCRIPT_HANDLE);
+       ejsDefineStringCFunction(-1, "split", ejs_split, NULL, MPR_VAR_SCRIPT_HANDLE);
+       ejsDefineStringCFunction(-1, "FileLoad", ejs_FileLoad, NULL, MPR_VAR_SCRIPT_HANDLE);
+       ejsDefineCFunction(-1, "join", ejs_join, NULL, MPR_VAR_SCRIPT_HANDLE);
 }