package rsacomb

import tech.oxfordsemantic.jrdfox.logic.datalog.{TupleTableAtom, TupleTableName}
import tech.oxfordsemantic.jrdfox.logic.expression.{IRI}

trait RDFTriple {

  implicit class RDFTriple(atom: TupleTableAtom) {

    /* Is this the best way to determine if an atom is an RDF triple?
     * Note that we can't use `getNumberOfArguments()` because is not
     * "consistent":
     * - for an atom created with `rdf(<term1>, <term2>, <term3>)`,
     * `getNumberOfArguments` returns 3
     * - for an atom created with `Atom.create(<tupletablename>, <term1>,
     * <term2>, <term3>)`, `getNumberOfArguments()` returns 3
     *
     * This is probably because `Atom.rdf(...) is implemented as:
     * ```scala
     *  def rdf(term1: Term, term2: Term, term3: Term): Atom =
     *    Atom.create(TupleTableName.create("internal:triple"), term1, term2, term3)
     * ```
     */
    def isRdfTriple: Boolean =
      atom.getTupleTableName.getName.equals("internal:triple")

    def isClassAssertion: Boolean =
      atom.isRdfTriple && atom.getArguments.get(1).equals(IRI.RDF_TYPE)

    def isRoleAssertion: Boolean =
      atom.isRdfTriple && !atom.getArguments.get(1).equals(IRI.RDF_TYPE)

    def suffix(sx: String): TupleTableAtom =
      if (this.isClassAssertion) {
        val newclass = atom.getArguments.get(2) match {
          case iri: IRI => IRI.create(s"${iri.getIRI}_$sx")
          case other    => other
        }
        TupleTableAtom.rdf(
          atom.getArguments.get(0),
          atom.getArguments.get(1),
          newclass
        )
      } else if (this.isRoleAssertion) {
        val newrole = atom.getArguments.get(1) match {
          case iri: IRI => IRI.create(s"${iri.getIRI}_$sx")
          case other    => other
        }
        TupleTableAtom.rdf(
          atom.getArguments.get(0),
          newrole,
          atom.getArguments.get(2)
        )
      } else {
        val newname =
          TupleTableName.create(s"${atom.getTupleTableName.getName}_$sx")
        TupleTableAtom.create(newname, atom.getArguments())
      }
  }

}