using System;
using System.Collections.Generic;
using System.Configuration;
using System.Threading.Tasks;
using InfluxDB.Client.Api.Domain;
using InfluxDB.Client.Writes;
using NUnit.Framework;

namespace InfluxDB.Client.Test
{
    [TestFixture]
    public class ItDeleteApiTest : AbstractItClientTest
    {
        [SetUp]
        public new async Task SetUp()
        {
            _organization = await FindMyOrg();

            var retention = new BucketRetentionRules(BucketRetentionRules.TypeEnum.Expire, 3600);

            _bucket = await Client.GetBucketsApi()
                .CreateBucketAsync(GenerateName("h2o"), retention, _organization);

            //
            // Add Permissions to read and write to the Bucket
            //
            var resource =
                new PermissionResource(PermissionResource.TypeBuckets, _bucket.Id, null,
                    _organization.Id);

            var readBucket = new Permission(Permission.ActionEnum.Read, resource);
            var writeBucket = new Permission(Permission.ActionEnum.Write, resource);

            var loggedUser = await Client.GetUsersApi().MeAsync();
            Assert.IsNotNull(loggedUser);

            var authorization = await Client.GetAuthorizationsApi()
                .CreateAuthorizationAsync(await FindMyOrg(),
                    new List<Permission> { readBucket, writeBucket });

            _token = authorization.Token;

            Client.Dispose();
            var options = new InfluxDBClientOptions(InfluxDbUrl)
            {
                Token = _token,
                Org = _organization.Id,
                Bucket = _bucket.Id
            };
            Client = new InfluxDBClient(options);
            _queryApi = Client.GetQueryApi();
        }

        [TearDown]
        protected new void After()
        {
            _writeApi.Dispose();
        }

        private Bucket _bucket;
        private QueryApi _queryApi;
        private WriteApi _writeApi;
        private DeleteApi _deleteApi;
        private Organization _organization;
        private string _token;

        [Test]
        public async Task Delete()
        {
            Client.Dispose();

            WriteData();

            var query = "from(bucket:\"" + _bucket.Name +
                        "\") |> range(start: 1970-01-01T00:00:00.000000001Z) |> pivot(rowKey:[\"_time\"], columnKey: [\"_field\"], valueColumn: \"_value\")";

            _queryApi = Client.GetQueryApi();
            var tables = await _queryApi.QueryAsync(query, _organization.Id);

            Assert.AreEqual(2, tables.Count);
            Assert.AreEqual(1, tables[0].Records.Count);
            Assert.AreEqual(5, tables[1].Records.Count);

            _deleteApi = Client.GetDeleteApi();

            await _deleteApi.Delete(DateTime.Now.AddHours(-1), DateTime.Now, "", _bucket, _organization);

            var tablesAfterDelete1 = await _queryApi.QueryAsync(query, _organization.Id);

            Assert.AreEqual(0, tablesAfterDelete1.Count);
        }

        [Test]
        public async Task DeleteWithPredicate()
        {
            Client.Dispose();

            WriteData();

            var query = "from(bucket:\"" + _bucket.Name +
                        "\") |> range(start: 1970-01-01T00:00:00.000000001Z) |> pivot(rowKey:[\"_time\"], columnKey: [\"_field\"], valueColumn: \"_value\")";

            _queryApi = Client.GetQueryApi();
            var tables = await _queryApi.QueryAsync(query, _organization.Id);

            Assert.AreEqual(2, tables.Count);
            Assert.AreEqual(1, tables[0].Records.Count);
            Assert.AreEqual(5, tables[1].Records.Count);

            var location = "east";
            var predicate = $"location=\"{location}\"";

            _deleteApi = Client.GetDeleteApi();
            await _deleteApi.Delete(DateTime.Now.AddHours(-1), DateTime.Now, predicate, _bucket, _organization);

            var tablesAfterDelete = await _queryApi.QueryAsync(query, _organization.Id);

            Assert.AreEqual(1, tablesAfterDelete.Count);
            Assert.AreEqual(5, tablesAfterDelete[0].Records.Count);

            await _deleteApi.Delete(DateTime.Now.AddHours(-1), DateTime.Now, "location = \"west\"", _bucket,
                _organization);

            var tablesAfterDelete2 = await _queryApi.QueryAsync(query, _organization.Id);

            Assert.AreEqual(0, tablesAfterDelete2.Count);
        }

        private void WriteData()
        {
            Environment.SetEnvironmentVariable("point-datacenter", "LA");
            ConfigurationManager.AppSettings["point-sensor.version"] = "1.23a";

            var options = new InfluxDBClientOptions(InfluxDbUrl)
            {
                Token = _token,
                DefaultTags = new Dictionary<string, string>
                {
                    { "id", "132-987-655" },
                    { "customer", "California Miner" },
                    { "env-var", "${env.point-datacenter}" },
                    { "sensor-version", "${point-sensor.version}" }
                }
            };

            Client = new InfluxDBClient(options);

            var point = PointData.Measurement("h2o_feet").Tag("location", "east").Field("water_level", 1);
            var point2 = PointData.Measurement("h2o_feet").Tag("location", "west").Field("water_level", 2);
            var point3 = PointData.Measurement("h2o_feet").Tag("location", "west").Field("water_level", 3);
            var point4 = PointData.Measurement("h2o_feet").Tag("location", "west").Field("water_level", 4);
            var point5 = PointData.Measurement("h2o_feet").Tag("location", "west").Field("water_level", 5);
            var point6 = PointData.Measurement("h2o_feet").Tag("location", "west").Field("water_level", 6);

            _writeApi = Client.GetWriteApi();
            var listener = new WriteApiTest.EventListener(_writeApi);
            _writeApi.WritePoint(point, _bucket.Name, _organization.Id);
            _writeApi.Flush();

            listener.WaitToSuccess();

            _writeApi.WritePoint(point2, _bucket.Name, _organization.Id);
            _writeApi.Flush();

            listener.WaitToSuccess();

            _writeApi.WritePoint(point3, _bucket.Name, _organization.Id);
            _writeApi.Flush();

            listener.WaitToSuccess();

            _writeApi.WritePoint(point4, _bucket.Name, _organization.Id);
            _writeApi.Flush();

            listener.WaitToSuccess();

            _writeApi.WritePoint(point5, _bucket.Name, _organization.Id);
            _writeApi.Flush();

            listener.WaitToSuccess();

            _writeApi.WritePoint(point6, _bucket.Name, _organization.Id);
            _writeApi.Flush();

            listener.WaitToSuccess();
        }
    }
}