1 /*
2  * Copyright (c) 2021 ScalaQuest Team.
3  *
4  * This file is part of ScalaQuest, and is distributed under the terms of the MIT License as
5  * described in the file LICENSE in the ScalaQuest distribution's top directory.
6  */
7 
8 package io.github.scalaquest.core.pipeline.resolver
9 
10 import io.github.scalaquest.core.dictionary.verbs.VerbPrep
11 import io.github.scalaquest.core.model.{Action, ItemDescription, ItemRef}
12 import io.github.scalaquest.core.pipeline.parser.{AbstractSyntaxTree, ParserResult}
13 
14 abstract class AbstractSyntaxTreeResolver extends Resolver {
15 
16   def actions(v: VerbPrep): Either[String, Action]
17 
18   def items(d: ItemDescription): Either[String, ItemRef]
19 
20   override def resolve(parserResult: ParserResult): Either[String, ResolverResult] = {
21     for {
22       statement <- parserResult.tree match {
23         case AbstractSyntaxTree.Intransitive(verb, prep, _) =>
24           for {
25             action <- actions((verb, prep))
26           } yield Statement.Intransitive(action)
27 
28         case AbstractSyntaxTree.Transitive(verb, prep, _, obj) =>
29           for {
30             action  <- actions((verb, prep))
31             itemRef <- items(obj)
32           } yield Statement.Transitive(action, itemRef)
33 
34         case AbstractSyntaxTree.Ditransitive(verb, prep, _, directObj, indirectObj) =>
35           for {
36             action          <- actions((verb, prep))
37             directItemRef   <- items(directObj)
38             indirectItemRef <- items(indirectObj)
39           } yield Statement.Ditransitive(action, directItemRef, indirectItemRef)
40 
41         case _ => Left("The statement is wrong.")
42       }
43     } yield ResolverResult(statement)
44   }
45 
46 }