import { Algebra } from 'sparqlalgebrajs';
import { BaseQuad } from 'rdf-js';
import { defaultGraph } from '@rdfjs/data-model';
import { PatchHandler } from './PatchHandler';
import { Readable } from 'stream';
import { Representation } from '../../ldp/representation/Representation';
import { ResourceIdentifier } from '../../ldp/representation/ResourceIdentifier';
import { ResourceLocker } from '../ResourceLocker';
import { ResourceStore } from '../ResourceStore';
import { someTerms } from 'rdf-terms';
import { SparqlUpdatePatch } from '../../ldp/http/SparqlUpdatePatch';
import { Store } from 'n3';
import { UnsupportedHttpError } from '../../util/errors/UnsupportedHttpError';

/**
 * PatchHandler that supports specific types of SPARQL updates.
 * Currently all DELETE/INSERT types are supported that have empty where bodies and no variables.
 */
export class SimpleSparqlUpdatePatchHandler extends PatchHandler {
  private readonly source: ResourceStore;
  private readonly locker: ResourceLocker;

  public constructor(source: ResourceStore, locker: ResourceLocker) {
    super();
    this.source = source;
    this.locker = locker;
  }

  public async canHandle(input: {identifier: ResourceIdentifier; patch: SparqlUpdatePatch}): Promise<void> {
    if (input.patch.dataType !== 'algebra' || !input.patch.algebra) {
      throw new UnsupportedHttpError('Only SPARQL update patch requests are supported.');
    }
  }

  public async handle(input: {identifier: ResourceIdentifier; patch: SparqlUpdatePatch}): Promise<void> {
    const op = input.patch.algebra;
    if (!this.isDeleteInsert(op)) {
      throw new UnsupportedHttpError('Only DELETE/INSERT SPARQL update operations are supported.');
    }

    const def = defaultGraph();
    const deletes = op.delete || [];
    const inserts = op.insert || [];

    if (!deletes.every((pattern): boolean => pattern.graph.equals(def))) {
      throw new UnsupportedHttpError('GRAPH statements are not supported.');
    }
    if (!inserts.every((pattern): boolean => pattern.graph.equals(def))) {
      throw new UnsupportedHttpError('GRAPH statements are not supported.');
    }
    if (op.where || deletes.some((pattern): boolean => someTerms(pattern, (term): boolean => term.termType === 'Variable'))) {
      throw new UnsupportedHttpError('WHERE statements are not supported.');
    }

    const lock = await this.locker.acquire(input.identifier);
    const quads = await this.source.getRepresentation(input.identifier, { type: [{ value: 'internal/quads', weight: 1 }]});
    const store = new Store<BaseQuad>();
    const importEmitter = store.import(quads.data);
    await new Promise((resolve, reject): void => {
      importEmitter.on('end', resolve);
      importEmitter.on('error', reject);
    });
    store.removeQuads(deletes);
    store.addQuads(inserts);
    const representation: Representation = {
      data: store.match() as Readable,
      dataType: 'quad',
      metadata: {
        raw: [],
        profiles: [],
        contentType: 'internal/quads',
      },
    };
    await this.source.setRepresentation(input.identifier, representation);

    await lock.release();
  }

  private isDeleteInsert(op: Algebra.Operation): op is Algebra.DeleteInsert {
    return op.type === Algebra.types.DELETE_INSERT;
  }
}