package sub

import (
	"github.com/fatih/color"
	"golang.flu0r1ne.net/planr"
        "fmt"
        "strings"
)

var (
  col_pass = color.New(color.FgGreen)
  col_fail = color.New(color.FgRed)
  col_title = color.New(color.FgHiWhite)
  col_label = color.New(color.FgCyan)
);

func tcTitle(tc planr.TestCase) string {
  title := tc.Cname

  if tc.Config.Title != nil {
    title = *tc.Config.Title
  }

  return title
}

func tcStatus(tc planr.TestCase) string {
    status := "SILENT"

    if tc.Result != nil {
      if tc.Result.Status == planr.PASSING {
        status = "PASS"
      } else {
        status = "FAIL"
      }
    }

    return status
}

func pprintLabeled(label, value string) {
  col_label.Printf("  %s: ", label)
  fmt.Println(value)
}

const (
  FENCE_WIDTH = 78
)

func pprintFenced(title, value string) {
  wingWidth := FENCE_WIDTH - len(title) - 2
  wing := strings.Repeat("-", wingWidth / 2)
  fence := strings.Repeat("-", FENCE_WIDTH) 

  fmt.Printf("  %s %s %s\n", wing, title, wing)
  
  fmt.Print("  " + strings.ReplaceAll(value, "\n", "\n  "))

  fmt.Println(fence)
}

func tcStatusLine(tc planr.TestCase) {
  title  := tcTitle(tc)
  status := tcStatus(tc)

  if status == "PASS" {
    col_pass.Printf("[%s] ", status);
  } else {
    col_fail.Printf("[%s] ", status);
  }

  col_title.Println(title);  
}

func tcPprint(tc planr.TestCase) {
  tcStatusLine(tc)

  if tc.Config.Points != nil {
    points := fmt.Sprintf("%.1f", *tc.Config.Points)
    pprintLabeled("points", points)
  }

  if tc.Config.Description != nil {
    pprintLabeled("description", *tc.Config.Description)
  }

  res := tc.Result

  if res.Status == planr.COMPILATION_FAILURE {
    
    if res.DebugOutput != "" {
      fmt.Println()
      pprintFenced("compilation output", tc.Result.DebugOutput);
    } else {
      fmt.Println("WARN: No debug output provided")
    }

  } else if res.Status == planr.RUNTIME_FAILURE {

    if tc.Result.TestOutput != "" {
      fmt.Println()
      pprintFenced("test output", tc.Result.TestOutput);
    }
    
  }

  fmt.Println()
}

func printResults(passed, tc_total int, earned, points_total float64) {
  col_title.Println("Final Results:")
  
  pprintLabeled("passed", fmt.Sprintf("%d/%d", passed, tc_total));

  percent := earned / points_total * 100

  pprintLabeled("score", fmt.Sprintf(
    "%.2f/%.2f ~= %.1f%%", earned, points_total, percent,
  ));
}