SessionController.cs

// Copyright (c) https://github.com/licanhua/YWinAppDriver. All rights reserved.
// Licensed under the MIT License.

using System;
using System.Collections.Generic;
using System.Linq;
using System.Threading.Tasks;
using Microsoft.AspNetCore.Http;
using Microsoft.AspNetCore.Mvc;
using Microsoft.AspNetCore.Routing;
using WinAppDriver.Infra;
using WinAppDriver.Infra.CommandHandler;
using WinAppDriver.Infra.Communication;
using WinAppDriver.Infra.Result;

namespace WinAppDriver.Controllers
{
  [Route("[controller]")]
  [ApiController]
  public class SessionController : WinAppDriverControllerBase
  {
    private ISessionManager _sessionManager;
    private ICommandHandlers _handlers;
    private IElementCommandHandler _elementHandler;

    private IActionResult ExecuteCommand(string sessionId, Func<object> func, Func<string> getSessionIdFunc = null)
    {
      try
      {
        var result = func();
        if (getSessionIdFunc != null)
        {
          sessionId = getSessionIdFunc();
        }
        return ReplyOk(sessionId, result);
      }
      catch (SessionException ex)
      {
        return ReplyFail(ex.HttpCode, ex.Status, ex.ErrorMessage);
      }
      catch (Exception ex)
      {
        return ReplyFail(500, ResponseStatusCode.UnknownError, ex.Message);
      }
    }

    private IActionResult ExecuteCommand(Command command, string sessionId, object body, string elementId)
    {
      return ExecuteCommand(sessionId, () => { return _handlers.ExecuteCommand(command, _sessionManager, sessionId, body, elementId); });
    }

    private IActionResult ExecuteCommand(string sessionId, string elementId, Func<IElement, object> func)
    {
      return ExecuteCommand(sessionId, () => { return _elementHandler.ExecuteCommand(_sessionManager, sessionId, elementId, func); });
    }

    private IActionResult ExecuteCommand(string sessionId, string elementId, Action<IElement> func)
    {
      return ExecuteCommand(sessionId, elementId, (element) => { func(element); return null; });
    }

    public SessionController(ISessionManager sessionManager, ICommandHandlers handlers, IElementCommandHandler elementHandler) {
      _sessionManager = sessionManager;
      _handlers = handlers;
      _elementHandler = elementHandler;
    }

    [HttpPost]
    [Route("")]
    public IActionResult CreateSession([FromBody] object content) {

      string sessionId = null;
      return ExecuteCommand(null, () => {
        var result = (NewSessionIntermediateResult)_handlers.ExecuteCommand(Command.NewSession, _sessionManager, null, content, null);
        sessionId = result.sessionId;
        return result.capabilities;
      }, () => { return sessionId; });

    }

    [HttpGet]
    [Route("{sessionId}")]
    public IActionResult GetCapabilites(string sessionId, [FromBody] object content)
    {
      return ExecuteCommand(Command.GetCapabilities, sessionId, null, null);
    }

    [HttpPost]
    [Route("{sessionId}/element")]
    public IActionResult FindElement(string sessionId, [FromBody] object content)
    {
      return ExecuteCommand(Command.FindElement, sessionId, content, null);
    }

    [HttpPost]
    [Route("{sessionId}/timeouts")]
    [Route("{sessionId}/timeouts/implicit_wait")]
    public IActionResult SetTimeouts(string sessionId, [FromBody] object content)
    {
      return ExecuteCommand(Command.SetImplicitTimeout, sessionId, content, null);
    }

    [HttpPost]
    [Route("{sessionId}/elements")]
    public IActionResult FindElements(string sessionId, [FromBody] object content)
    {
      return ExecuteCommand(Command.FindElements, sessionId, content, null);
    }

    [HttpGet]
    [Route("{sessionId}/window_handle")]
    public IActionResult GetWindowHandle(string sessionId)
    {
      return ExecuteCommand(Command.GetWindowHandle, sessionId, null, null);
    }

    [HttpGet]
    [Route("{sessionId}/window_handles")]
    public IActionResult GetWindowHandles(string sessionId)
    {
      return ExecuteCommand(Command.GetWindowHandles, sessionId, null, null);
    }

    [HttpPost]
    [Route("{sessionId}/element/{elementId}/element")]
    public IActionResult FindElementFromElement(string sessionId, string elementId, [FromBody] object content)
    {
      return ExecuteCommand(Command.FindElement, sessionId, content, elementId);
    }

    [HttpPost]
    [Route("{sessionId}/element/{elementId}/elements")]
    public IActionResult FindElementsFromElement(string sessionId, string elementId, [FromBody] object content)
    {
      return ExecuteCommand(Command.FindElements, sessionId, content, elementId);
    }
    [HttpPost]
    [Route("{sessionId}/element/{elementId}/click")]
    public IActionResult ElementClick(string sessionId, string elementId)
    {
      return ExecuteCommand(Command.ClickOnElement, sessionId, null, elementId);
    }

    [HttpGet]
    [Route("{sessionId}/element/{elementId}/equals/{otherId}")]
    public IActionResult ElementEquals(string sessionId, string elementId, string otherId)
    {
      return ExecuteCommand(Command.ElementEquals, sessionId, otherId, otherId);
    }

    [HttpGet]
    [Route("{sessionId}/element/{elementId}/attribute/{propertyName}")]
    public IActionResult ElementGetName(string sessionId, string elementId, string propertyName)
    {
      return ExecuteCommand(sessionId, elementId, (element) => { return element.GetAttribute(propertyName); });
    }

    [HttpPost]
    [Route("{sessionId}/element/{elementId}/doubleclick")]
    public IActionResult ElementDoubleClick(string sessionId, string elementId)
    {
      return ExecuteCommand(sessionId, elementId, (element) => { element.DoubleClick(); });
    }

    [HttpGet]
    [Route("{sessionId}/element/{elementId}/name")]
    public IActionResult ElementGetTagName(string sessionId, string elementId)
    {
      return ExecuteCommand(sessionId, elementId, (element) => { return element.GetTagName(); });
    }

    [HttpGet]
    [Route("{sessionId}/element/{elementId}/text")]
    public IActionResult ElementText(string sessionId, string elementId)
    {
      return ExecuteCommand(sessionId, elementId, (element) => { return element.GetText(); });
    }

    [HttpGet]
    [Route("{sessionId}/element/{elementId}/selected")]
    public IActionResult ElementSelected(string sessionId, string elementId)
    {
      return ExecuteCommand(sessionId, elementId, (element) => { return element.IsSelected(); });
    }

    [HttpGet]
    [Route("{sessionId}/element/{elementId}/enabled")]
    public IActionResult ElementEnabled(string sessionId, string elementId)
    {
      return ExecuteCommand(sessionId, elementId, (element) => { return element.IsEnabled(); });
    }

    [HttpGet]
    [Route("{sessionId}/element/{elementId}/displayed")]
    public IActionResult ElementDisplayed(string sessionId, string elementId)
    {
      return ExecuteCommand(sessionId, elementId, (element) => { return element.IsDisplayed(); });
    }

    [HttpGet]
    [Route("{sessionId}/element/{elementId}/location")]
    public IActionResult ElementLocation(string sessionId, string elementId)
    {
      return ExecuteCommand(sessionId, elementId, (element) => { return element.GetLocation(); });
    }

    [HttpGet]
    [Route("{sessionId}/element/{elementId}/size")]
    public IActionResult ElementSize(string sessionId, string elementId)
    {
      return ExecuteCommand(sessionId, elementId, (element) => { return element.GetSize(); });
    }

    [HttpGet]
    [Route("{sessionId}/title")]
    public IActionResult GetTitle(string sessionId)
    {
      return ExecuteCommand(Command.GetTitle, sessionId, null, null);
    }

    [HttpGet]
    [Route("{sessionId}/element/active")]
    public IActionResult GetFocusedElement(string sessionId)
    {
      return ExecuteCommand(Command.GetFocusedElement, sessionId, null, null);
    }

    [HttpPost]
    [Route("{sessionId}/element/{elementId}/value")]
    public IActionResult SetValue(string sessionId, string elementId, [FromBody] object body)
    {
      return ExecuteCommand(Command.ElementSendKeys, sessionId, body, elementId);
    }

    [HttpPost]
    [Route("{sessionId}/keys")]
    public IActionResult SendKeys(string sessionId, [FromBody] object body)
    {
      return ExecuteCommand(Command.SessionSendKeys, sessionId, body, null);
    }

    [HttpPost]
    [Route("{sessionId}/element/{elementId}/clear")]
    public IActionResult ElementClear(string sessionId, string elementId)
    {
      return ExecuteCommand(sessionId, elementId, (element) => { element.Clear(); });
    }

    [HttpGet]
    [Route("{sessionId}/source")]
    public IActionResult GetSource(string sessionId)
    {
      return ExecuteCommand(Command.GetSource, sessionId, null, null);
    }

    [HttpDelete]
    [Route("{sessionId}")]
    public IActionResult DeleteSession(string sessionId)
    {
      return ExecuteCommand(Command.DeleteSession, sessionId, null, null);
    }

    [HttpPost]
    [Route("{sessionId}/appium/app/close")]
    public IActionResult AppiumAppClose(string sessionId)
    {
      return ExecuteCommand(Command.AppiumCloseApp, sessionId, null, null);
    }

    [HttpPost]
    [Route("{sessionId}/window")]
    public IActionResult ActivateWindow(string sessionId, [FromBody] object content)
    {
      return ExecuteCommand(Command.ActivateWindow, sessionId, content, null);
    }

    [HttpPost]
    [Route("{sessionId}/window/{windowHandle}/size")]
    public IActionResult SetWindowSize(string sessionId, string windowHandle, [FromBody] object content)
    {
      return ExecuteCommand(Command.SetWindowSize, sessionId, content, windowHandle);
    }

    [HttpPost]
    [Route("{sessionId}/window/{windowHandle}/position")]
    public IActionResult SetWindowPosition(string sessionId, string windowHandle, [FromBody] object content)
    {
      return ExecuteCommand(Command.SetWindowPosition, sessionId, content, windowHandle);
    }


    [HttpGet]
    [Route("{sessionId}/window/{windowHandle}/size")]
    public IActionResult GetWindowSize(string sessionId, string windowHandle)
    {
      return ExecuteCommand(Command.GetWindowSize, sessionId, null, windowHandle);
    }

    [HttpGet]
    [Route("{sessionId}/window/{windowHandle}/position")]
    public IActionResult GetWindowPosition(string sessionId, string windowHandle)
    {
      return ExecuteCommand(Command.GetWindowPosition, sessionId, null, windowHandle);
    }

    [HttpPost]
    [Route("{sessionId}/window/{windowHandle}/maximize")]
    public IActionResult MaximizeWindow(string sessionId, string windowHandle, [FromBody] object content)
    {
      return ExecuteCommand(Command.MaximizeWindow, sessionId, content, windowHandle);
    }
    [HttpDelete]
    [Route("{sessionId}/window")]
    public IActionResult DeleteWindow(string sessionId)
    {
      return ExecuteCommand(Command.AppiumCloseApp, sessionId, null, null);
    }

    [HttpPost]
    [Route("{sessionId}/appium/app/launch")]
    public IActionResult AppLaunchApp(string sessionId)
    {
      return ExecuteCommand(Command.AppiumLaunchApp, sessionId, null, null);
    }

    [HttpGet]
    [Route("{sessionId}/screenshot")]
    public IActionResult TakeScreenshot(string sessionId)
    {
      return ExecuteCommand(Command.TakeScreenshot, sessionId, null, null);
    }

    [HttpGet]
    [Route("{sessionId}/element/{elementId}/screenshot")]
    public IActionResult TakeScreenshot(string sessionId, string elementId)
    {
      return ExecuteCommand(Command.TakeScreenshot, sessionId, null, elementId);
    }

    [HttpPost]
    [Route("{sessionId}/timeouts/async_script")]
    [Route("{sessionId}/url")]
    [Route("{sessionId}/forward")]
    [Route("{sessionId}/back")]
    [Route("{sessionId}/refresh")]
    [Route("{sessionId}/execute")]
    [Route("{sessionId}/execute_async")]
    [Route("{sessionId}/ime/{any}")]
    [Route("{sessionId}/frame")]
    [Route("{sessionId}/frame/parent")]
    [Route("{sessionId}/cookie")]
    [Route("{sessionId}/element/{elementId}/submit")]
    [Route("{sessionId}/orientation")]
    [Route("{sessionId}/accept_alert")]
    [Route("{sessionId}/dismiss_alert")]
    [Route("{sessionId}/alert_text")]
    [Route("{sessionId}/location")]
    [Route("{sessionId}/session_storage")]
    [Route("{sessionId}/log")]
    [Route("{sessionId}/local_storage")]
    public IActionResult UnknownPost(string sessionId)
    {
      return ExecuteCommand(Command.UnknownCommand, sessionId, null, null);
    }

    [HttpGet]
    [Route("{sessionId}/url")]
    [Route("{sessionId}/cookie")]
    [Route("{sessionId}/element/{id}/location_in_view")]
    [Route("{sessionId}/orientation")]
    [Route("{sessionId}/alert_text")]
    [Route("{sessionId}/location")]
    [Route("{sessionId}/local_storage")]
    [Route("{sessionId}/local_storage/key/{key}")]
    [Route("{sessionId}/local_storage/size")]
    [Route("{sessionId}/session_storage")]
    [Route("{sessionId}/session_storage/key/{key}")]
    [Route("{sessionId}/session_storage/size")]
    [Route("{sessionId}/log/types")]
    [Route("{sessionId}/application_cache/status")]
    [Route("{sessionId}/element/{id}/css/{propertyName}")]
    public IActionResult UnknownGet(string sessionId)
    {
      return ExecuteCommand(Command.UnknownCommand, sessionId, null, null);
    }

    [HttpDelete]
    [Route("{sessionId}/cookie/{name}")]
    [Route("{sessionId}/cookie")]
    [Route("{sessionId}/local_storage")]
    [Route("{sessionId}/local_storage/key/{key}")]
    [Route("{sessionId}/session_storage")]
    public IActionResult UnknownDelete(string sessionId)
    {
      return ExecuteCommand(Command.UnknownCommand, sessionId, null, null);
    }

    [HttpPost]
    [Route("{sessionId}/appium/device/push_file")]
    public IActionResult SessionDevicePushFile(string sessionId, [FromBody] object content)
    {
      return ExecuteCommand(Command.DevicePushFile, sessionId, content, null);
    }

    [HttpPost]
    [Route("{sessionId}/appium/device/pull_file")]
    public IActionResult SessionDevicePullFile(string sessionId, [FromBody] object content)
    {
      return ExecuteCommand(Command.DevicePullFile, sessionId, content, null);
    }

    [HttpPost]
    [Route("{sessionId}/moveto")]
    public IActionResult SessionMoveTo(string sessionId, [FromBody] object content)
    {
      return ExecuteCommand(Command.SessionMoveTo, sessionId, content, null);
    }

    [HttpPost]
    [Route("{sessionId}/click")]
    public IActionResult SessionClick(string sessionId, [FromBody] object content)
    {
      return ExecuteCommand(Command.SessionMouseAction, sessionId, content, MouseAction.Click.ToString());
    }

    [HttpPost]
    [Route("{sessionId}/buttondown")]
    public IActionResult SessionButtonDown(string sessionId, [FromBody] object content)
    {
      return ExecuteCommand(Command.SessionMouseAction, sessionId, content, MouseAction.Down.ToString());
    }

    [HttpPost]
    [Route("{sessionId}/buttonup")]
    public IActionResult SessionButtonUp(string sessionId, [FromBody] object content)
    {
      return ExecuteCommand(Command.SessionMouseAction, sessionId, content, MouseAction.Up.ToString());
    }

    [HttpPost]
    [Route("{sessionId}/doubleclick")]
    public IActionResult SessionDoubleClick(string sessionId, [FromBody] object content)
    {
      return ExecuteCommand(Command.SessionMouseAction, sessionId, content, MouseAction.DoubleClick.ToString());
    }

    [HttpPost]
    [Route("{sessionId}/touch/click")]
    public IActionResult SessionTouchClick(string sessionId, [FromBody] object content)
    {
      return ExecuteCommand(Command.SessionTouchActionOnElement, sessionId, content, TouchAction.Click.ToString());
    }

    [HttpPost]
    [Route("{sessionId}/touch/doubleclick")]
    public IActionResult SessionTouchDoubleClick(string sessionId, [FromBody] object content)
    {
      return ExecuteCommand(Command.SessionTouchActionOnElement, sessionId, content, TouchAction.DoubleClick.ToString());
    }

    [HttpPost]
    [Route("{sessionId}/touch/longclick")]
    public IActionResult SessionTouchLongClick(string sessionId, [FromBody] object content)
    {
      return ExecuteCommand(Command.SessionTouchActionOnElement, sessionId, content, TouchAction.PressAndHold.ToString());
    }

    [HttpPost]
    [Route("{sessionId}/touch/up")]
    public IActionResult SessionTouchUp(string sessionId, [FromBody] object content)
    {
      return ExecuteCommand(Command.SessionTouchUpDownMove, sessionId, content, TouchAction.Up.ToString());
    }

    [HttpPost]
    [Route("{sessionId}/touch/down")]
    public IActionResult SessionTouchDown(string sessionId, [FromBody] object content)
    {
      return ExecuteCommand(Command.SessionTouchUpDownMove, sessionId, content, TouchAction.Down.ToString());
    }

    [HttpPost]
    [Route("{sessionId}/touch/move")]
    public IActionResult SessionTouchMove(string sessionId, [FromBody] object content)
    {
      return ExecuteCommand(Command.SessionTouchUpDownMove, sessionId, content, TouchAction.Move.ToString());
    }
    [HttpPost]
    [Route("{sessionId}/touch/{action}", Order = 10)]
    public IActionResult TBD(string sessionId)
    {
      return ReplyFail(500, ResponseStatusCode.UnknownError, "Not implement yet");
    }

  }
}