addons/User/source/IoUser.c

   1 /*#io
   2 User ioDoc(
   3                  docCopyright("Steve Dekorte", 2004)
   4                  docLicense("BSD revised")
   5                  docDescription("This object provides access to the local operating system's information about the current user.")
   6                  //docCredits("Windows code by Mike Austin")
   7                  docCategory("Server")
   8 */
   9
  10 #include "IoUser.h"
  11 #include "IoState.h"
  12 #include "IoNumber.h"
  13 #include "IoList.h"
  14 #include "IoDirectory.h"
  15 #include <stdlib.h>
  16
  17 IoTag *IoUser_newTag(void *state)
  18 {
  19         IoTag *tag = IoTag_newWithName_("User");
  20         IoTag_state_(tag, state);
  21         IoTag_cloneFunc_(tag, (IoTagCloneFunc *)IoUser_rawClone);
  22         return tag;
  23 }
  24
  25 IoUser *IoUser_proto(void *state)
  26 {
  27         IoObject *self = IoObject_new(state);
  28         IoObject_tag_(self, IoUser_newTag(state));
  29
  30         IoState_registerProtoWithFunc_(state, self, IoUser_proto);
  31
  32         {
  33                 IoMethodTable methodTable[] = {
  34                 {"name", IoUser_protoName},
  35                 {"homeDirectory", IoUser_homeDirectory},
  36                 {NULL, NULL},
  37                 };
  38                 IoObject_addMethodTable_(self, methodTable);
  39         }
  40         return self;
  41 }
  42
  43 IoUser *IoUser_rawClone(IoUser *proto)
  44 {
  45         IoUser *self = IoObject_rawClonePrimitive(proto);
  46         return self;
  47 }
  48
  49 /* ----------------------------------------------------------- */
  50
  51 IoUser *IoUser_new(void *state)
  52 {
  53         IoObject *proto = IoState_protoWithInitFunction_(state, IoUser_proto);
  54         return IOCLONE(proto);
  55 }
  56
  57 /* ----------------------------------------------------------- */
  58
  59 #ifdef WIN32
  60
  61 #include <windows.h>
  62 #include <shlobj.h>
  63
  64 IoObject *IoUser_protoName(IoUser *self, IoObject *locals, IoMessage *m)
  65 {
  66         TCHAR userName[256]; DWORD userSize = 255;
  67         /* Copies up to userSize, then sets userSize to actual size */
  68         GetUserName(userName, &userSize);
  69         return IOSYMBOL(userName);
  70 }
  71
  72 IoObject *IoUser_homeDirectory(IoUser *self, IoObject *locals, IoMessage *m)
  73 {
  74         TCHAR homePath[MAX_PATH];
  75         SHGetFolderPath( NULL, CSIDL_APPDATA, NULL, 0, homePath );
  76         return IoDirectory_newWithPath_(IOSTATE, IOSYMBOL(homePath));
  77 }
  78
  79 #else /* Unix */
  80
  81 #include <unistd.h>
  82 #include <sys/types.h>
  83 #include <pwd.h>
  84
  85 IoObject *IoUser_protoName(IoUser *self, IoObject *locals, IoMessage *m)
  86 {
  87         /*#io
  88         docSlot("name", "Returns the current user's name.")
  89         */
  90
  91         char *userName = (char *)getlogin();
  92
  93         if (userName == NULL)
  94         {
  95                 userName = getenv("LOGNAME");
  96         }
  97
  98         if (userName == NULL)
  99         {
 100                 return IONIL(self);
 101         }
 102
 103         return IOSYMBOL(userName);
 104 }
 105
 106 #define IODIRECTORY(path) IoDirectory_newWithPath_(IOSTATE, IOSYMBOL(path));
 107
 108 IoObject *IoUser_homeDirectory(IoUser *self, IoObject *locals, IoMessage *m)
 109 {
 110         /*#io
 111         docSlot("homeDirectory", "Returns the current user's home directory as a Directory object.")
 112         */
 113
 114         char *login = (char *)getlogin();
 115         /*
 116          IoSymbol *homePath;
 117          */
 118
 119         if (login)
 120         {
 121                 struct passwd *pw = getpwnam(login);
 122
 123                 if (pw && pw->pw_dir)
 124                 {
 125                         return IODIRECTORY(pw->pw_dir);
 126                 }
 127         }
 128
 129         {
 130                 char *path = getenv("HOME");
 131
 132                 if (path)
 133                 {
 134                         return IODIRECTORY(path);
 135                 }
 136                 else
 137                 {
 138                         return IODIRECTORY("~");
 139                 }
 140         }
 141 }
 142
 143 #endif
 144