// Copyright (c) Microsoft Corporation
// All rights reserved. 
//
// Licensed under the Apache License, Version 2.0 (the ""License""); you may not use this file except in compliance with the License. You may obtain a copy of the License at http://www.apache.org/licenses/LICENSE-2.0 
//
// THIS CODE IS PROVIDED ON AN  *AS IS* BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED, INCLUDING WITHOUT LIMITATION ANY IMPLIED WARRANTIES OR CONDITIONS OF TITLE, FITNESS FOR A PARTICULAR PURPOSE, MERCHANTABLITY OR NON-INFRINGEMENT. 
//
// See the Apache Version 2.0 License for specific language governing permissions and limitations under the License.

#define NTDDI_VERSION 0x06010000

#include <v8.h>
#include <string>
#include <node_object_wrap.h>
#include <ppltasks.h>
#include "CollectionsConverter.h"
#include "CollectionsWrap.h"
#include "node-async.h"
#include "NodeRtUtils.h"
#include "OpaqueWrapper.h"
#include "WrapperBase.h"

#using <Windows.WinMD>

// this undefs fixes the issues of compiling Windows.Data.Json, Windows.Storag.FileProperties, and Windows.Stroage.Search
// Some of the node header files brings windows definitions with the same names as some of the WinRT methods
#undef DocumentProperties
#undef GetObject
#undef CreateEvent
#undef FindText
#undef SendMessage

const char* REGISTRATION_TOKEN_MAP_PROPERTY_NAME = "__registrationTokenMap__";

using namespace v8;
using namespace node;
using namespace concurrency;

namespace NodeRT { namespace Windows { namespace Web { 
  v8::Handle<v8::Value> WrapIUriToStreamResolver(::Windows::Web::IUriToStreamResolver^ wintRtInstance);
  ::Windows::Web::IUriToStreamResolver^ UnwrapIUriToStreamResolver(Handle<Value> value);
  
  v8::Handle<v8::Value> WrapWebError(::Windows::Web::WebError^ wintRtInstance);
  ::Windows::Web::WebError^ UnwrapWebError(Handle<Value> value);
  


  static v8::Handle<v8::Value> InitWebErrorStatusEnum(const Handle<Object> exports)
  {
    HandleScope scope;
    
    Handle<Object> enumObject = Object::New();
    exports->Set(String::NewSymbol("WebErrorStatus"), enumObject);

    enumObject->Set(String::NewSymbol("unknown"), Integer::New(0));
    enumObject->Set(String::NewSymbol("certificateCommonNameIsIncorrect"), Integer::New(1));
    enumObject->Set(String::NewSymbol("certificateExpired"), Integer::New(2));
    enumObject->Set(String::NewSymbol("certificateContainsErrors"), Integer::New(3));
    enumObject->Set(String::NewSymbol("certificateRevoked"), Integer::New(4));
    enumObject->Set(String::NewSymbol("certificateIsInvalid"), Integer::New(5));
    enumObject->Set(String::NewSymbol("serverUnreachable"), Integer::New(6));
    enumObject->Set(String::NewSymbol("timeout"), Integer::New(7));
    enumObject->Set(String::NewSymbol("errorHttpInvalidServerResponse"), Integer::New(8));
    enumObject->Set(String::NewSymbol("connectionAborted"), Integer::New(9));
    enumObject->Set(String::NewSymbol("connectionReset"), Integer::New(10));
    enumObject->Set(String::NewSymbol("disconnected"), Integer::New(11));
    enumObject->Set(String::NewSymbol("httpToHttpsOnRedirection"), Integer::New(12));
    enumObject->Set(String::NewSymbol("httpsToHttpOnRedirection"), Integer::New(13));
    enumObject->Set(String::NewSymbol("cannotConnect"), Integer::New(14));
    enumObject->Set(String::NewSymbol("hostNameNotResolved"), Integer::New(15));
    enumObject->Set(String::NewSymbol("operationCanceled"), Integer::New(16));
    enumObject->Set(String::NewSymbol("redirectFailed"), Integer::New(17));
    enumObject->Set(String::NewSymbol("unexpectedStatusCode"), Integer::New(18));
    enumObject->Set(String::NewSymbol("unexpectedRedirection"), Integer::New(19));
    enumObject->Set(String::NewSymbol("unexpectedClientError"), Integer::New(20));
    enumObject->Set(String::NewSymbol("unexpectedServerError"), Integer::New(21));
    enumObject->Set(String::NewSymbol("multipleChoices"), Integer::New(22));
    enumObject->Set(String::NewSymbol("movedPermanently"), Integer::New(23));
    enumObject->Set(String::NewSymbol("found"), Integer::New(24));
    enumObject->Set(String::NewSymbol("seeOther"), Integer::New(25));
    enumObject->Set(String::NewSymbol("notModified"), Integer::New(26));
    enumObject->Set(String::NewSymbol("useProxy"), Integer::New(27));
    enumObject->Set(String::NewSymbol("temporaryRedirect"), Integer::New(28));
    enumObject->Set(String::NewSymbol("badRequest"), Integer::New(29));
    enumObject->Set(String::NewSymbol("unauthorized"), Integer::New(30));
    enumObject->Set(String::NewSymbol("paymentRequired"), Integer::New(31));
    enumObject->Set(String::NewSymbol("forbidden"), Integer::New(32));
    enumObject->Set(String::NewSymbol("notFound"), Integer::New(33));
    enumObject->Set(String::NewSymbol("methodNotAllowed"), Integer::New(34));
    enumObject->Set(String::NewSymbol("notAcceptable"), Integer::New(35));
    enumObject->Set(String::NewSymbol("proxyAuthenticationRequired"), Integer::New(36));
    enumObject->Set(String::NewSymbol("requestTimeout"), Integer::New(37));
    enumObject->Set(String::NewSymbol("conflict"), Integer::New(38));
    enumObject->Set(String::NewSymbol("gone"), Integer::New(39));
    enumObject->Set(String::NewSymbol("lengthRequired"), Integer::New(40));
    enumObject->Set(String::NewSymbol("preconditionFailed"), Integer::New(41));
    enumObject->Set(String::NewSymbol("requestEntityTooLarge"), Integer::New(42));
    enumObject->Set(String::NewSymbol("requestUriTooLong"), Integer::New(43));
    enumObject->Set(String::NewSymbol("unsupportedMediaType"), Integer::New(44));
    enumObject->Set(String::NewSymbol("requestedRangeNotSatisfiable"), Integer::New(45));
    enumObject->Set(String::NewSymbol("expectationFailed"), Integer::New(46));
    enumObject->Set(String::NewSymbol("internalServerError"), Integer::New(47));
    enumObject->Set(String::NewSymbol("notImplemented"), Integer::New(48));
    enumObject->Set(String::NewSymbol("badGateway"), Integer::New(49));
    enumObject->Set(String::NewSymbol("serviceUnavailable"), Integer::New(50));
    enumObject->Set(String::NewSymbol("gatewayTimeout"), Integer::New(51));
    enumObject->Set(String::NewSymbol("httpVersionNotSupported"), Integer::New(52));

    return scope.Close(Undefined());
  }



  
  class IUriToStreamResolver : public WrapperBase
  {
  public:    
    static v8::Handle<v8::Value> Init(const Handle<Object> exports)
    {
      HandleScope scope;
      
      s_constructorTemplate = Persistent<FunctionTemplate>::New(FunctionTemplate::New(New));
      s_constructorTemplate->SetClassName(String::NewSymbol("IUriToStreamResolver"));
      s_constructorTemplate->InstanceTemplate()->SetInternalFieldCount(1);
      
      Handle<Value> asyncSymbol = String::NewSymbol("__winRtAsync__");
      Handle<Function> func;
                  
      func = FunctionTemplate::New(UriToStreamAsync)->GetFunction();
      func->Set(asyncSymbol, True(), PropertyAttribute::DontEnum);
      s_constructorTemplate->PrototypeTemplate()->Set(String::NewSymbol("uriToStreamAsync"), func);
      
                  
      Local<Function> constructor = s_constructorTemplate->GetFunction();


      exports->Set(String::NewSymbol("IUriToStreamResolver"), constructor);
      return scope.Close(Undefined());
    }


    virtual ::Platform::Object^ GetObjectInstance() const override
    {
      return _instance;
    }

  private:
    
    IUriToStreamResolver(::Windows::Web::IUriToStreamResolver^ instance)
    {
      _instance = instance;
    }
    
    
    static v8::Handle<v8::Value> New(const v8::Arguments& args)
    {
      HandleScope scope;

      // in case the constructor was called without the new operator
      if (!s_constructorTemplate->HasInstance(args.This()))
      {
        if (args.Length() > 0)
        {
          std::unique_ptr<Handle<Value> []> constructorArgs(new Handle<Value>[args.Length()]);

          Handle<Value> *argsPtr = constructorArgs.get();
          for (int i = 0; i < args.Length(); i++)
          {
            argsPtr[i] = args[i];
          }

          return s_constructorTemplate->GetFunction()->CallAsConstructor(args.Length(), constructorArgs.get());
        }
        else
        {
          return s_constructorTemplate->GetFunction()->CallAsConstructor(args.Length(), nullptr);
        }
      }
      
      ::Windows::Web::IUriToStreamResolver^ winRtInstance;


      if (args.Length() == 1 && OpaqueWrapper::IsOpaqueWrapper(args[0]) &&
        NodeRT::Utils::IsWinRtWrapperOf<::Windows::Web::IUriToStreamResolver^>(args[0]))
      {
        try 
        {
          winRtInstance = (::Windows::Web::IUriToStreamResolver^) NodeRT::Utils::GetObjectInstance(args[0]);
        }
        catch (Platform::Exception ^exception)
        {
          NodeRT::Utils::ThrowWinRtExceptionInJs(exception);
          return scope.Close(Undefined());
        }
      }
      else
      {
        ThrowException(Exception::Error(NodeRT::Utils::NewString(L"Invalid arguments, no suitable constructor found")));
        return scope.Close(Undefined());
      }

      args.This()->SetHiddenValue(String::NewSymbol("__winRtInstance__"), True());

      IUriToStreamResolver *wrapperInstance = new IUriToStreamResolver(winRtInstance);
      wrapperInstance->Wrap(args.This());

      return args.This();
    }


    static Handle<Value> UriToStreamAsync(const v8::Arguments& args)
    {
      HandleScope scope;

      if (!NodeRT::Utils::IsWinRtWrapperOf<::Windows::Web::IUriToStreamResolver^>(args.This()))
      {
        return scope.Close(Undefined());
      }

      if (args.Length() == 0 || !args[args.Length() -1]->IsFunction())
      {
          ThrowException(Exception::Error(NodeRT::Utils::NewString(L"Bad arguments: No callback was given")));
          return scope.Close(Undefined());
      }

      IUriToStreamResolver *wrapper = IUriToStreamResolver::Unwrap<IUriToStreamResolver>(args.This());

      ::Windows::Foundation::IAsyncOperation<::Windows::Storage::Streams::IInputStream^>^ op;
    

      if (args.Length() == 2
        && NodeRT::Utils::IsWinRtWrapperOf<::Windows::Foundation::Uri^>(args[0]))
      {
        try
        {
          ::Windows::Foundation::Uri^ arg0 = dynamic_cast<::Windows::Foundation::Uri^>(NodeRT::Utils::GetObjectInstance(args[0]));
          
          op = wrapper->_instance->UriToStreamAsync(arg0);
        }
        catch (Platform::Exception ^exception)
        {
          NodeRT::Utils::ThrowWinRtExceptionInJs(exception);
          return scope.Close(Undefined());
        }
      }
      else 
      {
        ThrowException(Exception::Error(NodeRT::Utils::NewString(L"Bad arguments: no suitable overload found")));
        return scope.Close(Undefined());
      }
    
      auto opTask = create_task(op);
      uv_async_t* asyncToken = NodeUtils::Async::GetAsyncToken(args[args.Length() -1].As<Function>());

      opTask.then( [asyncToken] (task<::Windows::Storage::Streams::IInputStream^> t) 
      {	
        try
        {
          auto result = t.get();
          NodeUtils::Async::RunCallbackOnMain(asyncToken, [result](NodeUtils::InvokeCallbackDelegate invokeCallback) {


            TryCatch tryCatch;
            Handle<Value> error; 
            Handle<Value> arg1 = NodeRT::Utils::CreateExternalWinRTObject("Windows.Storage.Streams", "IInputStream", result);
            if (tryCatch.HasCaught())
            {
              error = tryCatch.Exception()->ToObject();
            }
            else 
            {
              error = Undefined();
            }
            if (arg1.IsEmpty()) arg1 = Undefined();
            Handle<Value> args[] = {error, arg1};

            invokeCallback(_countof(args), args);
          });
        }
        catch (Platform::Exception^ exception)
        {
          NodeUtils::Async::RunCallbackOnMain(asyncToken, [exception](NodeUtils::InvokeCallbackDelegate invokeCallback) {
             
            Handle<Value> error = NodeRT::Utils::WinRtExceptionToJsError(exception);
        
            Handle<Value> args[] = {error};
            invokeCallback(_countof(args), args);
          });
        }  		
      });

      return scope.Close(Undefined());
    }
  





  private:
    ::Windows::Web::IUriToStreamResolver^ _instance;
    static Persistent<FunctionTemplate> s_constructorTemplate;

    friend v8::Handle<v8::Value> WrapIUriToStreamResolver(::Windows::Web::IUriToStreamResolver^ wintRtInstance);
    friend ::Windows::Web::IUriToStreamResolver^ UnwrapIUriToStreamResolver(Handle<Value> value);
    friend bool IsIUriToStreamResolverWrapper(Handle<Value> value);
  };
  Persistent<FunctionTemplate> IUriToStreamResolver::s_constructorTemplate;

  v8::Handle<v8::Value> WrapIUriToStreamResolver(::Windows::Web::IUriToStreamResolver^ winRtInstance)
  {
    HandleScope scope;

    if (winRtInstance == nullptr)
    {
      return scope.Close(Undefined());
    }

    Handle<Object> opaqueWrapper = CreateOpaqueWrapper(winRtInstance);
    Handle<Value> args[] = {opaqueWrapper};
    return scope.Close(IUriToStreamResolver::s_constructorTemplate->GetFunction()->NewInstance(_countof(args), args));
  }

  ::Windows::Web::IUriToStreamResolver^ UnwrapIUriToStreamResolver(Handle<Value> value)
  {
     return IUriToStreamResolver::Unwrap<IUriToStreamResolver>(value.As<Object>())->_instance;
  }

  void InitIUriToStreamResolver(Handle<Object> exports)
  {
    IUriToStreamResolver::Init(exports);
  }

  class WebError : public WrapperBase
  {
  public:    
    static v8::Handle<v8::Value> Init(const Handle<Object> exports)
    {
      HandleScope scope;
      
      s_constructorTemplate = Persistent<FunctionTemplate>::New(FunctionTemplate::New(New));
      s_constructorTemplate->SetClassName(String::NewSymbol("WebError"));
      s_constructorTemplate->InstanceTemplate()->SetInternalFieldCount(1);
      
                              
      Local<Function> constructor = s_constructorTemplate->GetFunction();

      constructor->Set(String::NewSymbol("getStatus"), FunctionTemplate::New(GetStatus)->GetFunction());

      exports->Set(String::NewSymbol("WebError"), constructor);
      return scope.Close(Undefined());
    }


    virtual ::Platform::Object^ GetObjectInstance() const override
    {
      return _instance;
    }

  private:
    
    WebError(::Windows::Web::WebError^ instance)
    {
      _instance = instance;
    }
    
    
    static v8::Handle<v8::Value> New(const v8::Arguments& args)
    {
      HandleScope scope;

      // in case the constructor was called without the new operator
      if (!s_constructorTemplate->HasInstance(args.This()))
      {
        if (args.Length() > 0)
        {
          std::unique_ptr<Handle<Value> []> constructorArgs(new Handle<Value>[args.Length()]);

          Handle<Value> *argsPtr = constructorArgs.get();
          for (int i = 0; i < args.Length(); i++)
          {
            argsPtr[i] = args[i];
          }

          return s_constructorTemplate->GetFunction()->CallAsConstructor(args.Length(), constructorArgs.get());
        }
        else
        {
          return s_constructorTemplate->GetFunction()->CallAsConstructor(args.Length(), nullptr);
        }
      }
      
      ::Windows::Web::WebError^ winRtInstance;


      if (args.Length() == 1 && OpaqueWrapper::IsOpaqueWrapper(args[0]) &&
        NodeRT::Utils::IsWinRtWrapperOf<::Windows::Web::WebError^>(args[0]))
      {
        try 
        {
          winRtInstance = (::Windows::Web::WebError^) NodeRT::Utils::GetObjectInstance(args[0]);
        }
        catch (Platform::Exception ^exception)
        {
          NodeRT::Utils::ThrowWinRtExceptionInJs(exception);
          return scope.Close(Undefined());
        }
      }
      else
      {
        ThrowException(Exception::Error(NodeRT::Utils::NewString(L"Invalid arguments, no suitable constructor found")));
        return scope.Close(Undefined());
      }

      args.This()->SetHiddenValue(String::NewSymbol("__winRtInstance__"), True());

      WebError *wrapperInstance = new WebError(winRtInstance);
      wrapperInstance->Wrap(args.This());

      return args.This();
    }


  


    static Handle<Value> GetStatus(const v8::Arguments& args)
    {
      HandleScope scope;

      if (args.Length() == 1
        && args[0]->IsInt32())
      {
        try
        {
          int arg0 = static_cast<int>(args[0]->Int32Value());
          
          ::Windows::Web::WebErrorStatus result;
          result = ::Windows::Web::WebError::GetStatus(arg0);
          return scope.Close(Integer::New(static_cast<int>(result)));
        }
        catch (Platform::Exception ^exception)
        {
          NodeRT::Utils::ThrowWinRtExceptionInJs(exception);
          return scope.Close(Undefined());
        }
      }
      else 
      {
        ThrowException(Exception::Error(NodeRT::Utils::NewString(L"Bad arguments: no suitable overload found")));
        return scope.Close(Undefined());
      }

      return scope.Close(Undefined());
    }



  private:
    ::Windows::Web::WebError^ _instance;
    static Persistent<FunctionTemplate> s_constructorTemplate;

    friend v8::Handle<v8::Value> WrapWebError(::Windows::Web::WebError^ wintRtInstance);
    friend ::Windows::Web::WebError^ UnwrapWebError(Handle<Value> value);
    friend bool IsWebErrorWrapper(Handle<Value> value);
  };
  Persistent<FunctionTemplate> WebError::s_constructorTemplate;

  v8::Handle<v8::Value> WrapWebError(::Windows::Web::WebError^ winRtInstance)
  {
    HandleScope scope;

    if (winRtInstance == nullptr)
    {
      return scope.Close(Undefined());
    }

    Handle<Object> opaqueWrapper = CreateOpaqueWrapper(winRtInstance);
    Handle<Value> args[] = {opaqueWrapper};
    return scope.Close(WebError::s_constructorTemplate->GetFunction()->NewInstance(_countof(args), args));
  }

  ::Windows::Web::WebError^ UnwrapWebError(Handle<Value> value)
  {
     return WebError::Unwrap<WebError>(value.As<Object>())->_instance;
  }

  void InitWebError(Handle<Object> exports)
  {
    WebError::Init(exports);
  }

} } } 

void init(Handle<Object> exports)
{
  if (FAILED(CoInitializeEx(nullptr, COINIT_MULTITHREADED)))
  {
    ThrowException(v8::Exception::Error(NodeRT::Utils::NewString(L"error in CoInitializeEx()")));
    return;
  }
  
  NodeRT::Windows::Web::InitWebErrorStatusEnum(exports);
  NodeRT::Windows::Web::InitIUriToStreamResolver(exports);
  NodeRT::Windows::Web::InitWebError(exports);

  NodeRT::Utils::RegisterNameSpace("Windows.Web", exports);
}


NODE_MODULE(binding, init)