require_relative '../test-util'

describe 'Pione::Transformer::RuleDefinitionTransformer' do
  transformer_spec("input_line", :input_line) do
    tc("input '*.txt'") do
      Naming.InputLine(DataExpr.new("*.txt").to_seq)
    end
  end

  transformer_spec("output_line", :output_line) do
    tc("output '*.txt'") do
      Naming.OutputLine(DataExpr.new("*.txt").to_seq)
    end
  end

  transformer_spec("param_line", :param_line) do
    tc("param {var: 1}") do
      Naming.ParamLine(
        Parameters.new(
          {Variable.new("var") => IntegerSequence.new([PioneInteger.new(1)])}
        )
      )
    end
    tc("param {var1: 1, var2: 2}") do
      Naming.ParamLine(
        Parameters.new(
          { Variable.new("var1") => IntegerSequence.new([PioneInteger.new(1)]),
            Variable.new("var2") => IntegerSequence.new([PioneInteger.new(2)]) }
        )
      )
    end
  end

  transformer_spec("feature_line", :feature_line) do
    tc("feature +A") do
      Naming.FeatureLine(Feature::RequisiteExpr.new("A"))
    end
  end

  transformer_spec("rule_definition", :rule_definition) do
    tc(<<-STRING) do
      Rule Test
        input  '*.a'
        output '{$INPUT[1].MATCH[1]}.b'
      Action
        echo "test" > {$OUTPUT[1].NAME}
      End
    STRING
      ActionRule.new(
        RuleExpr.new(Package.new("main"), "Test"),
        RuleCondition.new(
          [ DataExpr.new('*.a').to_seq ],
          [ DataExpr.new('{$INPUT[1].MATCH[1]}.b').to_seq ]
        ),
        ActionBlock.new("        echo \"test\" > {$OUTPUT[1].NAME}\n")
      )
    end

    tc(<<-STRING) do
      Rule Test
        input '*.a'
        output '{$INPUT[1].MATCH[1]}.b'
      Flow
        rule TestA
        rule TestB
      End
    STRING
      FlowRule.new(
        RuleExpr.new(Package.new("main"), "Test"),
        RuleCondition.new(
          [ DataExpr.new('*.a') ],
          [ DataExpr.new('{$INPUT[1].MATCH[1]}.b') ]
        ),
        FlowBlock.new(
          CallRule.new(RuleExpr.new(Package.new("main"), "TestA")),
          CallRule.new(RuleExpr.new(Package.new("main"), "TestB"))
        )
      )
    end

    tc(<<-STRING) do
      Rule Main
        input '*.txt'.except('summary.txt')
        output 'summary.txt'
        param {ConvertCharSet: true}
      Flow
      if $ConvertCharset
        rule NKF.params("-w")
      end
      rule CountChar
      rule Summarize
      End
    STRING
      FlowRule.new(
        RuleExpr.new(Package.new("main"), "Main"),
        RuleCondition.new(
          [Message.new(
              "except",
              DataExpr.new("*.txt"),
              DataExpr.new("summary.txt"))],
          [DataExpr.new("summary.txt")],
          params: Parameters.new(
            Variable.new("ConvertCharSet") =>
            BooleanSequence.new([PioneBoolean.true])
          )
        ),
        FlowBlock.new(
          ConditionalBlock.new(
            Variable.new("ConvertCharset"),
            { BooleanSequence.new([PioneBoolean.true]) =>
              FlowBlock.new(
                CallRule.new(Message.new(
                    "params",
                    RuleExpr.new(Package.new("main"), "NKF"),
                    StringSequence.new([PioneString.new("-w")])
                ))
            )}
          ),
          CallRule.new(
            RuleExpr.new(Package.new("main"), "CountChar")
          ),
          CallRule.new(
            RuleExpr.new(Package.new("main"), "Summarize")
          )
        )
      )
    end

    tc(<<-STRING) do
      Rule EmptyRule
        input '*.a'
        output '*.a'.touch
      End
    STRING
      rule_expr = RuleExpr.new(Package.new("main"), "EmptyRule")
      condition = RuleCondition.new(
        [DataExpr.new("*.a")], [Message.new("touch", DataExpr.new("*.a"))]
      )
      EmptyRule.new(rule_expr, condition, EmptyBlock.instance)
    end

  end

  transformer_spec("rule_definitions", :toplevel_elements) do
    tc(<<-STRING) do
      Rule TestA
        input  '*.a'
        output '{$INPUT[1].MATCH[1]}.b'
      Action
      cat {$INPUT[1].NAME} > {$OUTPUT[1].NAME}
      End

      Rule TestB
        input  '*.b'
        output '{$INPUT[1].MATCH[1]}.c'
      Action
      cat {$INPUT[1].NAME} > {$OUTPUT[1].NAME}
      End
    STRING
      [
        ActionRule.new(
          RuleExpr.new(Package.new("main"), "TestA"),
          RuleCondition.new(
            [ DataExpr.new("*.a") ],
            [ DataExpr.new('{$INPUT[1].MATCH[1]}.b') ]
          ),
          ActionBlock.new("      cat {$INPUT[1].NAME} > {$OUTPUT[1].NAME}\n")
        ),
        ActionRule.new(
          RuleExpr.new(Package.new("main"), "TestB"),
          RuleCondition.new(
            [ DataExpr.new("*.b") ],
            [ DataExpr.new('{$INPUT[1].MATCH[1]}.c') ]
          ),
          ActionBlock.new("      cat {$INPUT[1].NAME} > {$OUTPUT[1].NAME}\n")
        )
      ]
    end
  end
end