const {
  MAX_DB_RETRY_ATTEMPTS,
  SLEEP_BETWEEN_BATCHES,
  SLEEP_TIME_AFTER_FAILED_DB_OPERATION,
  LANDING_ZONE_GRAPH,
  BATCH_SIZE,
} = require('./config');

const prefixes = `
PREFIX besluit: <http://data.vlaanderen.be/ns/besluit#>
PREFIX skos: <http://www.w3.org/2004/02/skos/core#>
PREFIX lblodgeneriek: <https://data.lblod.info/vocabularies/generiek/>
PREFIX org: <http://www.w3.org/ns/org#>
PREFIX code: <http://lblod.data.gift/vocabularies/organisatie/>
PREFIX adms: <http://www.w3.org/ns/adms#>
PREFIX generiek: <https://data.vlaanderen.be/ns/generiek#>
PREFIX ere: <http://data.lblod.info/vocabularies/erediensten/>
PREFIX organisatie: <https://data.vlaanderen.be/ns/organisatie#>
PREFIX mu: <http://mu.semte.ch/vocabularies/core/>
PREFIX euvoc: <http://publications.europa.eu/ontology/euvoc#>
PREFIX prov: <http://www.w3.org/ns/prov#>
PREFIX schema: <http://schema.org/>
PREFIX locn: <http://www.w3.org/ns/locn#>
PREFIX foaf: <http://xmlns.com/foaf/0.1/>
PREFIX ext:<http://mu.semte.ch/vocabularies/ext/>
`

async function batchedDbUpdate(
  muUpdate,
  graph,
  triples,
  extraHeaders,
  endpoint,
  batchSize,
  maxAttempts,
  sleepBetweenBatches = 1000,
  sleepTimeOnFail = 1000,
  operation = 'INSERT'
) {
  for (let i = 0; i < triples.length; i += batchSize) {
    console.log(`Inserting triples in batch: ${i}-${i + batchSize}`);

    const batch = triples.slice(i, i + batchSize).join('\n');

    const insertCall = async () => {
      await muUpdate(`
${operation} DATA {
GRAPH <${graph}> {
${batch}
}
}
`, extraHeaders, endpoint);
    };

    await operationWithRetry(insertCall, 0, maxAttempts, sleepTimeOnFail);

    console.log(`Sleeping before next query execution: ${sleepBetweenBatches}`);
    await new Promise(r => setTimeout(r, sleepBetweenBatches));
  }
}

async function operationWithRetry(callback,
  attempt,
  maxAttempts,
  sleepTimeOnFail) {
  try {
    if (typeof callback === "function")
      return await callback();
    else // Catch error from promise - not how I would do it normally, but allows re use of existing code.
      return await callback;
  }
  catch (e) {
    console.log(`Operation failed for ${callback.toString()}, attempt: ${attempt} of ${maxAttempts}`);
    console.log(`Error: ${e}`);
    console.log(`Sleeping ${sleepTimeOnFail} ms`);

    if (attempt >= maxAttempts) {
      console.log(`Max attempts reached for ${callback.toString()}, giving up`);
      throw e;
    }

    await new Promise(r => setTimeout(r, sleepTimeOnFail));
    return operationWithRetry(callback, ++attempt, maxAttempts, sleepTimeOnFail);
  }
}


async function insertIntoPublicGraph(lib, statements) {
  console.log(`Inserting ${statements.length} statements into public graph`);

  await batchedDbUpdate(
    lib.muAuthSudo.updateSudo,
    'http://mu.semte.ch/graphs/public',
    Array.from(statements),
    {},
    process.env.MU_SPARQL_ENDPOINT,
    BATCH_SIZE,
    MAX_DB_RETRY_ATTEMPTS,
    SLEEP_BETWEEN_BATCHES,
    SLEEP_TIME_AFTER_FAILED_DB_OPERATION,
    'INSERT');
}

async function insertIntoSpecificGraphs(lib, statementsWithGraphs) {

  for( let graph in statementsWithGraphs) {
    console.log(`Inserting ${statementsWithGraphs[graph].length} statements into ${graph} graph`);
    await batchedDbUpdate(
      lib.muAuthSudo.updateSudo,
      graph,
      Array.from(statementsWithGraphs[graph]),
      {},
      process.env.MU_SPARQL_ENDPOINT,
      BATCH_SIZE,
      MAX_DB_RETRY_ATTEMPTS,
      SLEEP_BETWEEN_BATCHES,
      SLEEP_TIME_AFTER_FAILED_DB_OPERATION,
      'INSERT');
  }
  
}

async function deleteFromPublicGraph(lib, statements) {
  console.log(`Deleting ${statements.length} statements from public graph`);

  await batchedDbUpdate(
    lib.muAuthSudo.updateSudo,
    'http://mu.semte.ch/graphs/public',
    Array.from(statements),
    {},
    process.env.MU_SPARQL_ENDPOINT,
    BATCH_SIZE,
    MAX_DB_RETRY_ATTEMPTS,
    SLEEP_BETWEEN_BATCHES,
    SLEEP_TIME_AFTER_FAILED_DB_OPERATION,
    'DELETE');
}

async function deleteFromSpecificGraphs(lib, statementsWithGraphs) {
  

  for( let graph in statementsWithGraphs) {
    console.log(`Deleting ${statementsWithGraphs[graph].length} statements from ${graph} graph`);
    await batchedDbUpdate(
      lib.muAuthSudo.updateSudo,
      graph,
      Array.from(statementsWithGraphs[graph]),
      {},
      process.env.MU_SPARQL_ENDPOINT,
      BATCH_SIZE,
      MAX_DB_RETRY_ATTEMPTS,
      SLEEP_BETWEEN_BATCHES,
      SLEEP_TIME_AFTER_FAILED_DB_OPERATION,
      'DELETE');
  }
  
}


async function moveDataAfterInitialSync(muUpdate, endpoint) {
  // Move primary sites admin units
  await muUpdate(`
    ${prefixes}
    INSERT {
      GRAPH <http://mu.semte.ch/graphs/administrative-unit> {
        ?adminUnit org:hasPrimarySite ?site.
        ?site ?pSite ?oSite.
        ?contact ?pContact ?oContact.
        ?address ?pAddress ?oAddress.
        ?addressContact ?pAddressContact ?oAddressContact.
      }
    } WHERE {
      GRAPH <http://mu.semte.ch/graphs/administrative-unit> {
        ?adminUnit a ?type.
      }
      GRAPH <${LANDING_ZONE_GRAPH}> {
        ?adminUnit org:hasPrimarySite ?site.
        ?site ?pSite ?oSite.
        {
          ?site org:siteAddress ?contact.
          ?contact ?pContact ?oContact.
          ?contact locn:address ?addressContact.
          ?addressContact ?pAddressContact ?oAddressContact.
        }
        UNION
        {
          ?site org:siteAddress ?contact.
          ?contact ?pContact ?oContact.
        }
        UNION
        {
          ?site organisatie:bestaatUit ?address.
          ?address ?pAddress ?oAddress.
        }
      }        
    }
  `, undefined, endpoint)

  // Move sites administrative units
  await muUpdate(`
    ${prefixes}
    INSERT {
      GRAPH <http://mu.semte.ch/graphs/administrative-unit> {
        ?adminUnit org:hasSite ?site.
        ?site ?pSite ?oSite.
        ?contact ?pContact ?oContact.
        ?address ?pAddress ?oAddress.
        ?addressContact ?pAddressContact ?oAddressContact.
      }
    } WHERE {
      GRAPH <http://mu.semte.ch/graphs/administrative-unit> {
        ?adminUnit a ?type.
      }
      GRAPH <${LANDING_ZONE_GRAPH}> {
        ?adminUnit org:hasSite ?site.
        ?site ?pSite ?oSite.
        {
          ?site org:siteAddress ?contact.
          ?contact ?pContact ?oContact.
          ?contact locn:address ?addressContact.
          ?addressContact ?pAddressContact ?oAddressContact.
        }
        UNION
        {
          ?site org:siteAddress ?contact.
          ?contact ?pContact ?oContact.
        }
        UNION
        {
          ?site organisatie:bestaatUit ?address.
          ?address ?pAddress ?oAddress.
        }     
      }   
    }
  `, undefined, endpoint)

  // Move primary sites worships
  await muUpdate(`
    ${prefixes}
    INSERT {
      GRAPH <http://mu.semte.ch/graphs/worship-service> {
        ?adminUnit org:hasPrimarySite ?site.
        ?site ?pSite ?oSite.
        ?contact ?pContact ?oContact.
        ?address ?pAddress ?oAddress.
        ?addressContact ?pAddressContact ?oAddressContact.
      }
    } WHERE {
      GRAPH <http://mu.semte.ch/graphs/worship-service> {
        ?adminUnit a ?type.
      }
      GRAPH <${LANDING_ZONE_GRAPH}> {
        ?adminUnit org:hasPrimarySite ?site.
        ?site ?pSite ?oSite.
        {
          ?site org:siteAddress ?contact.
          ?contact ?pContact ?oContact.
          ?contact locn:address ?addressContact.
          ?addressContact ?pAddressContact ?oAddressContact.
        }
        UNION
        {
          ?site org:siteAddress ?contact.
          ?contact ?pContact ?oContact.
        }
        UNION
        {
          ?site organisatie:bestaatUit ?address.
          ?address ?pAddress ?oAddress.
        }        
      }
    }
  `, undefined, endpoint)

  // Move sites worships
  await muUpdate(`
    ${prefixes}
    INSERT {
      GRAPH <http://mu.semte.ch/graphs/worship-service> {
        ?adminUnit org:hasSite ?site.
        ?site ?pSite ?oSite.
        ?contact ?pContact ?oContact.
        ?address ?pAddress ?oAddress.
        ?addressContact ?pAddressContact ?oAddressContact.
      }
    } WHERE {
      GRAPH <http://mu.semte.ch/graphs/worship-service> {
        ?adminUnit a ?type.
      }
      GRAPH <${LANDING_ZONE_GRAPH}> {
        ?adminUnit org:hasSite ?site.
        ?site ?pSite ?oSite.
        {
          ?site org:siteAddress ?contact.
          ?contact ?pContact ?oContact.
          ?contact locn:address ?addressContact.
          ?addressContact ?pAddressContact ?oAddressContact.
        }
        UNION
        {
          ?site org:siteAddress ?contact.
          ?contact ?pContact ?oContact.
        }
        UNION
        {
          ?site organisatie:bestaatUit ?address.
          ?address ?pAddress ?oAddress.
        }        
      }
    }
  `, undefined, endpoint)

  // Move sites worships
  await muUpdate(`
    ${prefixes}
    INSERT {
      GRAPH <http://mu.semte.ch/graphs/public> {
        ?type a <http://lblod.data.gift/vocabularies/organisatie/TypeVestiging>.
        ?type ?a ?b.
      }
    } WHERE {
      GRAPH <${LANDING_ZONE_GRAPH}> {
        ?type a <http://lblod.data.gift/vocabularies/organisatie/TypeVestiging>.
        ?type ?a ?b.
      }
    }
  `, undefined, endpoint)
}

module.exports = {
  batchedDbUpdate,
  moveDataAfterInitialSync,
  insertIntoPublicGraph,
  deleteFromPublicGraph,
  insertIntoSpecificGraphs,
  deleteFromSpecificGraphs,
};