sharee.bike-App/TINKLib/Model/Connector/Query/QueryLoggedIn.cs

using System;
using System.Linq;
using System.Threading.Tasks;
using TINK.Model.Bikes;
using TINK.Model.Connector.Updater;
using TINK.Model.Services.CopriApi;
using TINK.Repository;

namespace TINK.Model.Connector
{
    /// <summary> Provides query functionality for a logged in user. </summary>
    public class QueryLoggedIn : BaseLoggedIn, IQuery
    {
        /// <summary> Copri server. </summary>
        private readonly ICopriServer server;

        /// <summary>Constructs a copri query object.</summary>
        /// <param name="copriServer">Server which implements communication.</param>
        public QueryLoggedIn(ICopriServerBase copriServer,
            string sessionCookie,
            string mail,
            Func<DateTime> dateTimeProvider) : base(copriServer, sessionCookie, mail, dateTimeProvider)
        {
            server = copriServer as ICopriServer;
            if (server == null)
            {
                throw new ArgumentException($"Copri server is not of expected typ. Type detected is {copriServer.GetType()}.");
            }

            server = copriServer as ICopriServer;
        }

        /// <summary> Gets all stations including postions.</summary>
        public async Task<Result<StationsAndBikesContainer>> GetBikesAndStationsAsync()
        {
            var stationResponse = await server.GetStationsAsync();
            var bikesAvailableResponse = await server.GetBikesAvailableAsync();
            var bikesOccupiedResponse = await server.GetBikesOccupiedAsync();

            return new Result<StationsAndBikesContainer>(
                typeof(CopriCallsMonkeyStore),
                new StationsAndBikesContainer(
                    stationResponse.GetStationsAllMutable(),
                    UpdaterJSON.GetBikesAll(
                        bikesAvailableResponse?.bikes?.Values,
                        bikesOccupiedResponse?.bikes_occupied?.Values,
                        Mail,
                        DateTimeProvider)),
                stationResponse.GetGeneralData());
        }

        /// <summary> Gets bikes occupied and bikes for which feedback is required. </summary>
        /// <returns>Collection of bikes.</returns>
        public async Task<Result<BikeCollection>> GetBikesOccupiedAsync()
        {
            var bikesFeedbackRequired = await server.GetBikesAvailableAsync();
            var bikesOccupiedResponse = await server.GetBikesOccupiedAsync();

            return new Result<BikeCollection>(
                typeof(CopriCallsMonkeyStore),
                UpdaterJSON.GetBikesAll(
                    bikesFeedbackRequired.bikes?.Values?.Select(bike => bike)?.Where(bike => bike.GetState() == State.InUseStateEnum.FeedbackPending),
                    bikesOccupiedResponse?.bikes_occupied?.Values,
                    Mail,
                    DateTimeProvider),
                bikesOccupiedResponse.GetGeneralData());
        }

        /// <summary> Gets bikes available and bikes occupied. </summary>
        /// <returns>Collection of bikes.</returns>
        public async Task<Result<BikeCollection>> GetBikesAsync()
        {
            var bikesAvailableResponse = await server.GetBikesAvailableAsync();
            var bikesOccupiedResponse = await server.GetBikesOccupiedAsync();

            return new Result<BikeCollection>(
                typeof(CopriCallsMonkeyStore),
                UpdaterJSON.GetBikesAll(
                    bikesAvailableResponse?.bikes?.Values,
                    bikesOccupiedResponse?.bikes_occupied?.Values,
                    Mail,
                    DateTimeProvider),
                bikesAvailableResponse.GetGeneralData());
        }
    }
}