#include <util/wintypes.h>
#include <stdio.h>
#include <string.h>
#include <util/rapitypes.h>

#include "debug.h"
#include "args.h"

#include "vectorutils.h"
#include "stringutils.h"
#include "csidlpaths.h"


// todo: add wildcard support
// todo: option to recurse directories
// todo: add wince specific attrs

/*
ATTRIB [+R | -R] [+A | -A ] [+S | -S] [+H | -H] [drive:][path][filename]
       [/S [/D]]

  +   Sets an attribute.
  -   Clears an attribute.
  R   Read-only file attribute.
  A   Archive file attribute.
  S   System file attribute.
  H   Hidden file attribute.
  [drive:][path][filename]
      Specifies a file or files for attrib to process.
  /S  Processes matching files in the current folder
      and all subfolders.
  /D  Processes folders as well.

*/

void usage()
{
    printf("(C) 2003-2008 Willem jan Hengeveld  itsme@xs4all.nl\n");
    printf("Usage: pchmod [+|-][RASH] path ...\n");
}
int main( int argc, char *argv[])
{
    DebugStdOut();

    StringList namelist;

    DWORD removeflags= 0;
    DWORD addflags= 0;

    for (int i=1 ; i<argc ; i++)
    {
        if (argv[i][0]=='-' || argv[i][0]=='+') {
            DWORD &flags= argv[i][0]=='-' ? removeflags : addflags;
            for (int j=1 ; j<strlen(argv[i]) ; j++) {
                switch(tolower(argv[i][j]))
                {
                case 'a': flags |= FILE_ATTRIBUTE_ARCHIVE; break;
                case 'r': flags |= FILE_ATTRIBUTE_READONLY; break;
                case 'h': flags |= FILE_ATTRIBUTE_HIDDEN; break;
                case 's': flags |= FILE_ATTRIBUTE_SYSTEM; break;
                default:
                    usage();
                    return 1;
                }
            }
        }
        else
        {
            namelist.push_back(argv[i]);
        }
    }
    if (FAILED(CeRapiInit()))
    {
        printf( "Failed\n");
        return 1;
    }

    DWORD res= 0;
    for (StringList::iterator i= namelist.begin() ; i!=namelist.end(); ++i) {
        DWORD attr= CeGetFileAttributes( expand_csidl(ToWString(*i)).c_str() );
        if (attr==0xFFFFFFFF) {
            ceerror("CeGetFileAttributes(%s)", (*i).c_str());
            res= 1;
            continue;
        }
        attr |= addflags;
        attr &= ~removeflags;
        if (!CeSetFileAttributes( expand_csidl(ToWString(*i)).c_str(), attr )) {
            ceerror("CeSetFileAttributes(%s)", (*i).c_str());
            res= 1;
        }
    }

    CeRapiUninit();

    return res;
}