package model

import (
	"context"
	"errors"
	"fmt"

	"github.com/derailed/k9s/internal"
	"github.com/derailed/k9s/internal/render"
	"k8s.io/apimachinery/pkg/runtime"
)

// Subject represents a subject model.
type Subject struct {
	Resource
}

// List returns a collection of subjects.
func (s *Subject) List(ctx context.Context) ([]runtime.Object, error) {
	kind, ok := ctx.Value(internal.KeySubjectKind).(string)
	if !ok {
		return nil, errors.New("expecting a SubjectKind")
	}

	crbs, err := fetchClusterRoleBindings(s.factory)
	if err != nil {
		return nil, err
	}
	oo := make([]runtime.Object, 0, len(crbs))
	for _, crb := range crbs {
		for _, su := range crb.Subjects {
			if su.Kind != kind || inSubjectRes(oo, su.Name) {
				continue
			}
			oo = append(oo, render.SubjectRef{
				Name:          su.Name,
				Kind:          "ClusterRoleBinding",
				FirstLocation: crb.Name,
			})
		}
	}

	rbs, err := fetchRoleBindings(s.factory)
	if err != nil {
		return nil, err
	}
	for _, rb := range rbs {
		for _, su := range rb.Subjects {
			if su.Kind != kind || inSubjectRes(oo, su.Name) {
				continue
			}
			oo = append(oo, render.SubjectRef{
				Name:          su.Name,
				Kind:          "RoleBinding",
				FirstLocation: rb.Name,
			})
		}
	}

	return oo, nil
}

// Hydrate returns a pod as container rows.
func (s *Subject) Hydrate(oo []runtime.Object, rr render.Rows, re Renderer) error {
	for i, o := range oo {
		res, ok := o.(render.SubjectRef)
		if !ok {
			return fmt.Errorf("expecting unstructured but got %T", o)
		}

		if err := re.Render(res, render.AllNamespaces, &rr[i]); err != nil {
			return err
		}
	}

	return nil
}

func inSubjectRes(oo []runtime.Object, match string) bool {
	for _, o := range oo {
		res, ok := o.(render.SubjectRef)
		if !ok {
			continue
		}
		if res.Name == match {
			return true
		}
	}
	return false
}