package gui

import (
	"fmt"

	"github.com/isacikgoz/gitbatch/core/command"
	"github.com/isacikgoz/gitbatch/core/git"
	"github.com/jroimartin/gocui"
)

var (
	statusHeaderViewFeature = viewFeature{Name: "status-header", Title: " Status Header "}
	stageViewFeature        = viewFeature{Name: "staged", Title: " Staged "}
	unstageViewFeature      = viewFeature{Name: "unstaged", Title: " Not Staged "}
	stashViewFeature        = viewFeature{Name: "stash", Title: " Stash "}

	statusViews = []viewFeature{stageViewFeature, unstageViewFeature, stashViewFeature}

	commitMesageReturnView string
	stagedFiles            []*command.File
	unstagedFiles          []*command.File
)

// open the status layout
func (gui *Gui) openStatusView(g *gocui.Gui, v *gocui.View) error {
	if err := populateFileLists(gui.getSelectedRepository()); err != nil {
		return err
	}
	gui.openStatusHeaderView(g)
	gui.openStageView(g)
	gui.openUnStagedView(g)
	gui.openStashView(g)
	return nil
}

// focus to next view
func (gui *Gui) nextStatusView(g *gocui.Gui, v *gocui.View) error {
	return gui.nextViewOfGroup(g, v, statusViews)
}

// focus to previous view
func (gui *Gui) previousStatusView(g *gocui.Gui, v *gocui.View) error {
	return gui.previousViewOfGroup(g, v, statusViews)
}

// moves the cursor downwards for the main view and if it goes to bottom it
// prevents from going further
func (gui *Gui) statusCursorDown(g *gocui.Gui, v *gocui.View) error {
	if v == nil {
		return nil
	}

	cx, cy := v.Cursor()
	ox, oy := v.Origin()
	ly := len(v.BufferLines()) - 2 // why magic number? have no idea

	// if we are at the end we just return
	if cy+oy == ly {
		return nil
	}
	if err := v.SetCursor(cx, cy+1); err != nil {

		if err := v.SetOrigin(ox, oy+1); err != nil {
			return err
		}
	}
	r := gui.getSelectedRepository()
	return refreshStatusView(v.Name(), g, r, false)
}

// moves the cursor upwards for the main view
func (gui *Gui) statusCursorUp(g *gocui.Gui, v *gocui.View) error {
	if v == nil {
		return nil
	}

	ox, oy := v.Origin()
	cx, cy := v.Cursor()
	if err := v.SetCursor(cx, cy-1); err != nil && oy > 0 {
		if err := v.SetOrigin(ox, oy-1); err != nil {
			return err
		}
	}
	r := gui.getSelectedRepository()
	return refreshStatusView(v.Name(), g, r, false)
}

// header og the status layout
func (gui *Gui) openStatusHeaderView(g *gocui.Gui) error {
	maxX, _ := g.Size()
	r := gui.getSelectedRepository()
	v, err := g.SetView(statusHeaderViewFeature.Name, 6, 2, maxX-6, 4)
	if err != nil {
		if err != gocui.ErrUnknownView {
			return err
		}
		fmt.Fprintln(v, r.AbsPath)
		// v.Frame = false
		v.Wrap = true
	}
	return nil
}

// close the opened stat views
func (gui *Gui) closeStatusView(g *gocui.Gui, v *gocui.View) error {
	for _, view := range statusViews {
		if err := g.DeleteView(view.Name); err != nil {
			return err
		}
	}
	if err := g.DeleteView(statusHeaderViewFeature.Name); err != nil {
		return err
	}
	stagedFiles = make([]*command.File, 0)
	unstagedFiles = make([]*command.File, 0)

	return gui.closeViewCleanup(mainViewFeature.Name)
}

// generate file lists by git status command
func populateFileLists(r *git.Repository) error {
	files, err := command.Status(r)
	if err != nil {
		return err
	}
	stagedFiles = make([]*command.File, 0)
	unstagedFiles = make([]*command.File, 0)
	for _, file := range files {
		if file.X != command.StatusNotupdated && file.X != command.StatusUntracked && file.X != command.StatusIgnored && file.X != command.StatusUpdated {
			stagedFiles = append(stagedFiles, file)
		}
		if file.Y != command.StatusNotupdated {
			unstagedFiles = append(unstagedFiles, file)
		}
	}
	return err
}

func refreshStatusView(viewName string, g *gocui.Gui, r *git.Repository, reload bool) error {
	if reload {
		populateFileLists(r)
	}
	var err error
	switch viewName {
	case stageViewFeature.Name:
		err = refreshStagedView(g)
	case unstageViewFeature.Name:
		err = refreshUnstagedView(g)
	case stashViewFeature.Name:
		err = refreshStashView(g, r)
	}
	return err
}

func refreshAllStatusView(g *gocui.Gui, r *git.Repository, reload bool) error {
	for _, v := range statusViews {
		if err := refreshStatusView(v.Name, g, r, reload); err != nil {
			return err
		}
	}
	return nil
}