workbench/tools/IconInject/main.c

   1
   2 #define DEBUG 0
   3 #include <aros/debug.h>
   4
   5 #include <proto/exec.h>
   6 #include <proto/dos.h>
   7 #include <proto/icon.h>
   8 #include <workbench/icon.h>
   9 #include <stdio.h>
  10 #include <string.h>
  11
  12 static const char version[] __attribute__((used)) = "\0$VER: IconInject 0.1 (06.03.09) © The AROS Dev Team" ;
  13
  14 static
  15 const UBYTE iitemplate[] =
  16 "SOURCE,"
  17 "TARGET,"
  18 "IMAGE/S,"
  19 "VERBOSE/S";
  20
  21 struct ArgList
  22 {
  23     STRPTR      sourceIcon;
  24     STRPTR      targetIcon;
  25     IPTR        replaceimages;
  26     IPTR        verbose;
  27 } arglist;
  28
  29 int main(int argc, char **argv)
  30 {
  31     struct RDArgs *iireadargs;
  32     struct ArgList *IIArgList = &arglist;
  33
  34     char *IISourceIcon = NULL;
  35     char *IITargetIcon = NULL;
  36
  37     iireadargs = ReadArgs(iitemplate, (IPTR *)IIArgList, NULL);
  38     if (iireadargs)
  39     {
  40         if ((IIArgList->sourceIcon) && (IIArgList->targetIcon))
  41         {
  42             D(bug("[IconInject] Source @ %p, Dest @ %p\n", IIArgList->sourceIcon, IIArgList->targetIcon));
  43             D(bug("[IconInject] '%s' -> '%s'\n", IIArgList->sourceIcon, IIArgList->targetIcon));
  44             if ((strlen(IIArgList->sourceIcon) > 5) && (strncmp(IIArgList->sourceIcon + strlen(IIArgList->sourceIcon) - 5, ".info",5)))
  45             {
  46                 if ((IISourceIcon = AllocVec(strlen(IIArgList->sourceIcon) - 4, MEMF_CLEAR)) != NULL)
  47                     CopyMem(IIArgList->sourceIcon, IISourceIcon, strlen(IIArgList->sourceIcon) - 5);
  48             }
  49             else
  50             {
  51 /* "TODO: Check if the name is a device and append disk.." */
  52                 if ((IISourceIcon = AllocVec(strlen(IIArgList->sourceIcon) + 1, MEMF_CLEAR)) != NULL)
  53                     CopyMem(IIArgList->sourceIcon, IISourceIcon, strlen(IIArgList->sourceIcon) - 5);
  54             }
  55
  56             if ((strlen(IIArgList->targetIcon) > 5) && (strncmp(IIArgList->targetIcon + strlen(IIArgList->targetIcon) - 5, ".info",5)))
  57             {
  58                 if ((IITargetIcon = AllocVec(strlen(IIArgList->targetIcon) - 4, MEMF_CLEAR)) != NULL)
  59                     CopyMem(IIArgList->targetIcon, IITargetIcon, strlen(IIArgList->targetIcon) - 5);
  60             }
  61             else
  62             {
  63 /* "TODO: Check if the name is a device and append disk.." */
  64                 if ((IITargetIcon = AllocVec(strlen(IIArgList->targetIcon) + 1, MEMF_CLEAR)) != NULL)
  65                     CopyMem(IIArgList->targetIcon, IITargetIcon, strlen(IIArgList->targetIcon) - 5);
  66             }
  67
  68             if (IISourceIcon && IITargetIcon)
  69             {
  70                 struct DiskObject       *sourceIconDOB = NULL;
  71                 struct DiskObject       *targetIconDOB = NULL;
  72                 IPTR                    tmpGadRender = 0;
  73                 IPTR                    tmpSelRender = 0;
  74
  75                 D(bug("[IconInject] '%s' -> '%s'\n", IISourceIcon, IITargetIcon));
  76
  77                 sourceIconDOB = GetIconTags
  78                         (
  79                             IISourceIcon,
  80                             ICONGETA_FailIfUnavailable,        FALSE,
  81                             TAG_DONE
  82                         );
  83
  84                 targetIconDOB = GetIconTags
  85                         (
  86                             IITargetIcon,
  87                             ICONGETA_FailIfUnavailable,        FALSE,
  88                             TAG_DONE
  89                         );
  90
  91                 if (sourceIconDOB && targetIconDOB)
  92                 {
  93                     if (IIArgList->replaceimages)
  94                     {
  95                         if (IIArgList->verbose)
  96                             printf("Replacing ICON IMAGERY in %s.info from %s.info\n", IITargetIcon, IISourceIcon);
  97
  98                         tmpGadRender = (IPTR)targetIconDOB->do_Gadget.GadgetRender;
  99                         tmpSelRender = (IPTR)targetIconDOB->do_Gadget.SelectRender;
 100
 101                         targetIconDOB->do_Gadget.GadgetRender = sourceIconDOB->do_Gadget.GadgetRender;
 102                         targetIconDOB->do_Gadget.SelectRender = sourceIconDOB->do_Gadget.SelectRender;
 103                     }
 104
 105                     // Update the icon ..
 106                     PutIconTagList(IITargetIcon, targetIconDOB, NULL);
 107
 108                     if (IIArgList->replaceimages)
 109                     {
 110                         targetIconDOB->do_Gadget.GadgetRender = (APTR)tmpGadRender;
 111                         targetIconDOB->do_Gadget.SelectRender = (APTR)tmpSelRender;
 112                     }
 113                 }
 114                 {
 115                     // Couldnt obtain one of the Icons ..
 116                 }
 117             }
 118             else
 119             {
 120                 //Failed to allocate Icon name buffer(s) ..
 121             }
 122         }
 123         else
 124         {
 125             //Missing arguments.. display a gui?
 126             printf("no args ..\n");
 127         }
 128     }
 129     if (IISourceIcon) FreeVec(IISourceIcon);
 130     if (IITargetIcon) FreeVec(IITargetIcon);
 131     return RETURN_OK;
 132 }