text: /*
   * Copyright 2010 Jacek Caban for CodeWeavers
   *
   * This library is free software; you can redistribute it and/or
   * modify it under the terms of the GNU Lesser General Public
   * License as published by the Free Software Foundation; either
   * version 2.1 of the License, or (at your option) any later version.
   *
   * This library is distributed in the hope that it will be useful,
   * but WITHOUT ANY WARRANTY; without even the implied warranty of
   * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
   * Lesser General Public License for more details.
   *
   * You should have received a copy of the GNU Lesser General Public
   * License along with this library; if not, write to the Free Software
   * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
   */
  
  #include <stdarg.h>
  
  #define COBJMACROS
  
  #include <windef.h>
  #include <winbase.h>
  #include <winreg.h>
  #include <ole2.h>
  #include <shellapi.h>
  #include <activscp.h>
  #include <initguid.h>
  
  #include "wscript.h"
  
  #include <wine/debug.h>
  #include <wine/unicode.h>
  
  #ifdef _WIN64
  
  #define IActiveScriptParse_Release IActiveScriptParse64_Release
  #define IActiveScriptParse_InitNew IActiveScriptParse64_InitNew
  #define IActiveScriptParse_ParseScriptText IActiveScriptParse64_ParseScriptText
  
  #else
  
  #define IActiveScriptParse_Release IActiveScriptParse32_Release
  #define IActiveScriptParse_InitNew IActiveScriptParse32_InitNew
  #define IActiveScriptParse_ParseScriptText IActiveScriptParse32_ParseScriptText
  
  #endif
  
  WINE_DEFAULT_DEBUG_CHANNEL(wscript);
  
  static const WCHAR wscriptW[] = {'W','S','c','r','i','p','t',0};
  static const WCHAR wshW[] = {'W','S','H',0};
  WCHAR scriptFullName[MAX_PATH];
  
  ITypeInfo *host_ti;
  ITypeInfo *arguments_ti;
  
  static HRESULT query_interface(REFIID,void**);
  
  static HRESULT WINAPI ActiveScriptSite_QueryInterface(IActiveScriptSite *iface,
                                                        REFIID riid, void **ppv)
  {
      return query_interface(riid, ppv);
  }
  
  static ULONG WINAPI ActiveScriptSite_AddRef(IActiveScriptSite *iface)
  {
      return 2;
  }
  
  static ULONG WINAPI ActiveScriptSite_Release(IActiveScriptSite *iface)
  {
      return 1;
  }
  
  static HRESULT WINAPI ActiveScriptSite_GetLCID(IActiveScriptSite *iface, LCID *plcid)
  {
      WINE_TRACE("()\n");
  
      *plcid = GetUserDefaultLCID();
      return S_OK;
  }
  
  static HRESULT WINAPI ActiveScriptSite_GetItemInfo(IActiveScriptSite *iface,
          LPCOLESTR pstrName, DWORD dwReturnMask, IUnknown **ppunkItem, ITypeInfo **ppti)
  {
      WINE_TRACE("(%s %x %p %p)\n", wine_dbgstr_w(pstrName), dwReturnMask, ppunkItem, ppti);
  
      if(strcmpW(pstrName, wshW) && strcmpW(pstrName, wscriptW))
          return E_FAIL;
  
      if(dwReturnMask & SCRIPTINFO_ITYPEINFO) {
          ITypeInfo_AddRef(host_ti);
          *ppti = host_ti;
      }
  
      if(dwReturnMask & SCRIPTINFO_IUNKNOWN) {
          IHost_AddRef(&host_obj);
          *ppunkItem = (IUnknown*)&host_obj;
      }
  
      return S_OK;
  }
  
  static HRESULT WINAPI ActiveScriptSite_GetDocVersionString(IActiveScriptSite *iface,
          BSTR *pbstrVersion)
  {
      WINE_FIXME("()\n");
      return E_NOTIMPL;
  }
  
  static HRESULT WINAPI ActiveScriptSite_OnScriptTerminate(IActiveScriptSite *iface,
          const VARIANT *pvarResult, const EXCEPINFO *pexcepinfo)
  {
      WINE_FIXME("()\n");
      return E_NOTIMPL;
  }
  
  static HRESULT WINAPI ActiveScriptSite_OnStateChange(IActiveScriptSite *iface,
          SCRIPTSTATE ssScriptState)
  {
      WINE_TRACE("(%x)\n", ssScriptState);
      return S_OK;
  }
  
  static HRESULT WINAPI ActiveScriptSite_OnScriptError(IActiveScriptSite *iface,
          IActiveScriptError *pscripterror)
  {
      WINE_FIXME("()\n");
      return E_NOTIMPL;
  }
  
  static HRESULT WINAPI ActiveScriptSite_OnEnterScript(IActiveScriptSite *iface)
  {
      WINE_TRACE("()\n");
      return S_OK;
  }
  
  static HRESULT WINAPI ActiveScriptSite_OnLeaveScript(IActiveScriptSite *iface)
  {
      WINE_TRACE("()\n");
      return S_OK;
  }
  
  static IActiveScriptSiteVtbl ActiveScriptSiteVtbl = {
      ActiveScriptSite_QueryInterface,
      ActiveScriptSite_AddRef,
      ActiveScriptSite_Release,
      ActiveScriptSite_GetLCID,
      ActiveScriptSite_GetItemInfo,
      ActiveScriptSite_GetDocVersionString,
      ActiveScriptSite_OnScriptTerminate,
      ActiveScriptSite_OnStateChange,
      ActiveScriptSite_OnScriptError,
      ActiveScriptSite_OnEnterScript,
      ActiveScriptSite_OnLeaveScript
  };
  
  static IActiveScriptSite script_site = { &ActiveScriptSiteVtbl };
  
  static HRESULT WINAPI ActiveScriptSiteWindow_QueryInterface(IActiveScriptSiteWindow *iface, REFIID riid, void **ppv)
  {
      return query_interface(riid, ppv);
  }
  
  static ULONG WINAPI ActiveScriptSiteWindow_AddRef(IActiveScriptSiteWindow *iface)
  {
      return 2;
  }
  
  static ULONG WINAPI ActiveScriptSiteWindow_Release(IActiveScriptSiteWindow *iface)
  {
      return 1;
  }
  
  static HRESULT WINAPI ActiveScriptSiteWindow_GetWindow(IActiveScriptSiteWindow *iface, HWND *phwnd)
  {
      TRACE("(%p)\n", phwnd);
  
      *phwnd = NULL;
      return S_OK;
  }
  
  static HRESULT WINAPI ActiveScriptSiteWindow_EnableModeless(IActiveScriptSiteWindow *iface, BOOL fEnable)
  {
      TRACE("(%x)\n", fEnable);
      return S_OK;
  }
  
  static const IActiveScriptSiteWindowVtbl ActiveScriptSiteWindowVtbl = {
      ActiveScriptSiteWindow_QueryInterface,
      ActiveScriptSiteWindow_AddRef,
      ActiveScriptSiteWindow_Release,
      ActiveScriptSiteWindow_GetWindow,
      ActiveScriptSiteWindow_EnableModeless
  };
  
  static IActiveScriptSiteWindow script_site_window = { &ActiveScriptSiteWindowVtbl };
  
  static HRESULT query_interface(REFIID riid, void **ppv)
  {
      if(IsEqualGUID(riid, &IID_IUnknown)) {
          TRACE("(IID_IUnknown %p)\n", ppv);
          *ppv = &script_site;
      }else if(IsEqualGUID(riid, &IID_IActiveScriptSite)) {
          TRACE("(IID_IActiveScriptSite %p)\n", ppv);
          *ppv = &script_site;
      }else if(IsEqualGUID(riid, &IID_IActiveScriptSiteWindow)) {
          TRACE("(IID_IActiveScriptSiteWindow %p)\n", ppv);
          *ppv = &script_site_window;
      }else {
          *ppv = NULL;
          TRACE("(%s %p)\n", wine_dbgstr_guid(riid), ppv);
          return E_NOINTERFACE;
      }
  
      IUnknown_AddRef((IUnknown*)*ppv);
      return S_OK;
  }
  
  static BOOL load_typelib(void)
  {
      ITypeLib *typelib;
      HRESULT hres;
  
      static const WCHAR wscript_exeW[] = {'w','s','c','r','i','p','t','.','e','x','e',0};
  
      hres = LoadTypeLib(wscript_exeW, &typelib);
      if(FAILED(hres))
          return FALSE;
  
      hres = ITypeLib_GetTypeInfoOfGuid(typelib, &IID_IHost, &host_ti);
      if(SUCCEEDED(hres))
          hres = ITypeLib_GetTypeInfoOfGuid(typelib, &IID_IArguments2, &arguments_ti);
  
      ITypeLib_Release(typelib);
      return SUCCEEDED(hres);
  }
  
  static BOOL get_engine_clsid(const WCHAR *ext, CLSID *clsid)
  {
      WCHAR fileid[64], progid[64];
      DWORD res;
      LONG size;
      HKEY hkey;
      HRESULT hres;
  
      static const WCHAR script_engineW[] =
          {'\\','S','c','r','i','p','t','E','n','g','i','n','e',0};
  
      res = RegOpenKeyW(HKEY_CLASSES_ROOT, ext, &hkey);
      if(res != ERROR_SUCCESS)
          return FALSE;
  
      size = sizeof(fileid)/sizeof(WCHAR);
      res = RegQueryValueW(hkey, NULL, fileid, &size);
      RegCloseKey(hkey);
      if(res != ERROR_SUCCESS)
          return FALSE;
  
      WINE_TRACE("fileid is %s\n", wine_dbgstr_w(fileid));
  
      strcatW(fileid, script_engineW);
      res = RegOpenKeyW(HKEY_CLASSES_ROOT, fileid, &hkey);
      if(res != ERROR_SUCCESS)
          return FALSE;
  
      size = sizeof(progid)/sizeof(WCHAR);
      res = RegQueryValueW(hkey, NULL, progid, &size);
      RegCloseKey(hkey);
      if(res != ERROR_SUCCESS)
          return FALSE;
  
      WINE_TRACE("ProgID is %s\n", wine_dbgstr_w(progid));
  
      hres = CLSIDFromProgID(progid, clsid);
      return SUCCEEDED(hres);
  }
  
  static BOOL create_engine(CLSID *clsid, IActiveScript **script_ret,
          IActiveScriptParse **parser)
  {
      IActiveScript *script;
      IUnknown *unk;
      HRESULT hres;
  
      hres = CoCreateInstance(clsid, NULL, CLSCTX_INPROC_SERVER|CLSCTX_INPROC_HANDLER,
              &IID_IUnknown, (void**)&unk);
      if(FAILED(hres))
          return FALSE;
  
      hres = IUnknown_QueryInterface(unk, &IID_IActiveScript, (void**)&script);
      IUnknown_Release(unk);
      if(FAILED(hres))
          return FALSE;
  
      hres = IActiveScript_QueryInterface(script, &IID_IActiveScriptParse, (void**)parser);
      if(FAILED(hres)) {
          IActiveScript_Release(script);
          return FALSE;
      }
  
      *script_ret = script;
      return TRUE;
  }
  
  static BOOL init_engine(IActiveScript *script, IActiveScriptParse *parser)
  {
      HRESULT hres;
  
      if(!load_typelib())
          return FALSE;
  
      hres = IActiveScript_SetScriptSite(script, &script_site);
      if(FAILED(hres))
          return FALSE;
  
      hres = IActiveScriptParse_InitNew(parser);
      if(FAILED(hres))
          return FALSE;
  
      hres = IActiveScript_AddNamedItem(script, wscriptW, SCRIPTITEM_ISVISIBLE);
      if(FAILED(hres))
          return FALSE;
  
      hres = IActiveScript_AddNamedItem(script, wshW, SCRIPTITEM_ISVISIBLE);
      if(FAILED(hres))
          return FALSE;
  
      hres = IActiveScript_SetScriptState(script, SCRIPTSTATE_INITIALIZED);
      return SUCCEEDED(hres);
  }
  
  static BSTR get_script_str(const WCHAR *filename)
  {
      const char *file_map;
      HANDLE file, map;
      DWORD size, len;
      BSTR ret;
  
      file = CreateFileW(filename, GENERIC_READ, 0, NULL, OPEN_EXISTING, FILE_ATTRIBUTE_READONLY, NULL);
      if(file == INVALID_HANDLE_VALUE)
          return NULL;
  
      size = GetFileSize(file, NULL);
      map = CreateFileMappingW(file, NULL, PAGE_READONLY, 0, 0, NULL);
      CloseHandle(file);
      if(map == INVALID_HANDLE_VALUE)
          return NULL;
  
      file_map = MapViewOfFile(map, FILE_MAP_READ, 0, 0, 0);
      CloseHandle(map);
      if(!file_map)
          return NULL;
  
      len = MultiByteToWideChar(CP_ACP, 0, file_map, size, NULL, 0);
      ret = SysAllocStringLen(NULL, len);
      MultiByteToWideChar(CP_ACP, 0, file_map, size, ret, len);
  
      UnmapViewOfFile(file_map);
      return ret;
  }
  
  static void run_script(const WCHAR *filename, IActiveScript *script, IActiveScriptParse *parser)
  {
      BSTR text;
      HRESULT hres;
  
      text = get_script_str(filename);
      if(!text) {
          WINE_FIXME("Could not get script text\n");
          return;
      }
  
      hres = IActiveScriptParse_ParseScriptText(parser, text, NULL, NULL, NULL, 1, 1,
              SCRIPTTEXT_HOSTMANAGESSOURCE|SCRIPTITEM_ISVISIBLE, NULL, NULL);
      SysFreeString(text);
      if(FAILED(hres)) {
          WINE_FIXME("ParseScriptText failed: %08x\n", hres);
          return;
      }
  
      hres = IActiveScript_SetScriptState(script, SCRIPTSTATE_STARTED);
      if(FAILED(hres))
          WINE_FIXME("SetScriptState failed: %08x\n", hres);
  }
  
  static BOOL set_host_properties(const WCHAR *prop)
  {
      static const WCHAR nologoW[] = {'n','o','l','o','g','o',0};
      static const WCHAR iactive[] = {'i',0};
      static const WCHAR batch[] = {'b',0};
  
      if(*prop == '/') {
          ++prop;
          if(*prop == '/')
              ++prop;
      }
      else
          ++prop;
  
      if(strcmpiW(prop, iactive) == 0)
          wshInteractive = VARIANT_TRUE;
      else if(strcmpiW(prop, batch) == 0)
          wshInteractive = VARIANT_FALSE;
      else if(strcmpiW(prop, nologoW) == 0)
          WINE_FIXME("ignored %s switch\n", debugstr_w(nologoW));
      else
      {
          WINE_FIXME("unsupported switch %s\n", debugstr_w(prop));
          return FALSE;
      }
      return TRUE;
  }
  
  int WINAPI wWinMain(HINSTANCE hInst, HINSTANCE hPrevInst, LPWSTR cmdline, int cmdshow)
  {
      WCHAR *ext, *filepart, *filename = NULL;
      IActiveScriptParse *parser;
      IActiveScript *script;
      WCHAR **argv;
      CLSID clsid;
      int argc, i;
      DWORD res;
  
      WINE_TRACE("(%p %p %s %x)\n", hInst, hPrevInst, wine_dbgstr_w(cmdline), cmdshow);
  
      argv = CommandLineToArgvW(GetCommandLineW(), &argc);
      if(!argv)
          return 1;
  
      for(i=1; i<argc; i++) {
          if(*argv[i] == '/' || *argv[i] == '-') {
              if(!set_host_properties(argv[i]))
                  return 1;
          }else {
              filename = argv[i];
              argums = argv+i+1;
              numOfArgs = argc-i-1;
              break;
          }
      }
  
      if(!filename) {
          WINE_FIXME("No file name specified\n");
          return 1;
      }
      res = GetFullPathNameW(filename, sizeof(scriptFullName)/sizeof(WCHAR), scriptFullName, &filepart);
      if(!res || res > sizeof(scriptFullName)/sizeof(WCHAR))
          return 1;
  
      ext = strrchrW(filepart, '.');
      if(!ext || !get_engine_clsid(ext, &clsid)) {
          WINE_FIXME("Could not find engine for %s\n", wine_dbgstr_w(ext));
          return 1;
      }
  
      CoInitialize(NULL);
  
      if(!create_engine(&clsid, &script, &parser)) {
          WINE_FIXME("Could not create script engine\n");
          CoUninitialize();
          return 1;
      }
  
      if(init_engine(script, parser)) {
          run_script(filename, script, parser);
          IActiveScript_Close(script);
          ITypeInfo_Release(host_ti);
      }else {
          WINE_FIXME("Script initialization failed\n");
      }
  
      IActiveScript_Release(script);
      IActiveScriptParse_Release(parser);
  
      CoUninitialize();
  
      return 0;
  }
  