package fstore

import (
	"path/filepath"
	"strings"
)

type StoreState struct {
	Path     string
	IsDir    bool
	Dirs     map[string]StoreState
	Files    map[string]StoreState
	FileData string
}

func NewStoreState(in map[string]string) StoreState {
	root := StoreState{
		Path:  "/",
		IsDir: true,
		Dirs:  map[string]StoreState{},
		Files: map[string]StoreState{},
	}

	for path, fileData := range in {
		slugs := strings.Split(path[1:], "/") // Remove leading slash.

		parent := root

		// Add directories.
		for _, part := range slugs[:len(slugs)-1] {
			if _, ok := parent.Dirs[part]; !ok {
				parent.Dirs[part] = StoreState{
					Path:  filepath.Join(parent.Path, part),
					IsDir: true,
					Dirs:  map[string]StoreState{},
					Files: map[string]StoreState{},
				}
			}
			parent = parent.Dirs[part]
		}

		parent.Files[slugs[len(slugs)-1]] = StoreState{
			Path:     path,
			IsDir:    false,
			FileData: fileData,
		}
	}

	return root
}