/////////////////////////////////////////////////////////////////////////
// ClientFileQuery.cpp - Tests FileQuery ATL Component                 //
//                                                                     //
// Jim Fawcett, CSE775 - Distributed Objects, MT1-Q4, Spring 2009      //
/////////////////////////////////////////////////////////////////////////

#include "stdafx.h"
#include "..\FileQuery\FileQuery_i.h"
#include <iostream>
#include <iomanip>

int _tmain(int argc, _TCHAR* argv[])
{
  std::cout << "\n  Solution for CSE775 - MT2, Q4 - File Queries";
  std::cout << "\n ==============================================\n";

  ::CoInitialize(NULL);
  try
  {
    CComQIPtr<IFiles> pFiles;
    pFiles.CoCreateInstance(CLSID_Files);
    HRESULT hr;
    if(pFiles)
    {
      std::cout << "\n  create succeeded\n";
      CComBSTR path(L"C:\\temp");
      int FileCount;
      BSTR* FileNames;
      int* FileSizes;
      hr = pFiles->getFiles(path, &FileCount, &FileNames, &FileSizes);
      if(!SUCCEEDED(hr))
      {
        std::cout << "\n query failed\n\n";
        return 1;
      }
      for(int i=0; i<FileCount; ++i)
      {
        std::wcout.setf(std::ios::adjustfield, std::ios::right);
        std::wcout << L"\n  " << std::setw(20) << FileNames[i] << L"  ";
        std::wcout << FileSizes[i];
        ::SysFreeString(FileNames[i]);
      }
      CoTaskMemFree(FileNames);
      CoTaskMemFree(FileSizes);
      std::cout << "\n\n";
    }
  }
  catch(std::exception& ex)
  {
    std::cout << "\n\n  " << ex.what() << "\n\n";
    return 1;
  }
	return 0;
}