package backup

import (
	"code.laria.me/petrific/acl"
	"code.laria.me/petrific/fs"
	"code.laria.me/petrific/objects"
	"code.laria.me/petrific/storage"
	"fmt"
	"io"
	"math/rand"
	"os"
)

func RestoreFile(s storage.Storage, id objects.ObjectId, w io.Writer) error {
	file, err := storage.GetObjectOfType(s, id, objects.OTFile)
	if err != nil {
		return err
	}

	for i, fragment := range *file.(*objects.File) {
		blob_obj, err := storage.GetObjectOfType(s, fragment.Blob, objects.OTBlob)
		if err != nil {
			return err
		}
		blob := *blob_obj.(*objects.Blob)

		if uint64(len(blob)) != fragment.Size {
			return fmt.Errorf("RestoreFile: blob size of %s doesn't match size in fragment %d of file %s", fragment.Blob, i, id)
		}

		if _, err := w.Write(blob); err != nil {
			return err
		}
	}

	return nil
}

func execBitFromACL(a acl.ACL) bool {
	return a.ToUnixPerms()&0100 != 0
}

func RestoreDir(s storage.Storage, id objects.ObjectId, root fs.Dir) error {
	tree_obj, err := storage.GetObjectOfType(s, id, objects.OTTree)
	tree := tree_obj.(objects.Tree)

	seen := make(map[string]struct{})

	for name, file_info := range tree {
		switch file_info.Type() {
		case objects.TETFile:
			tmpname := fmt.Sprintf(".petrific-%d-%08x%08x%08x%08x", os.Getpid(), rand.Uint32(), rand.Uint32(), rand.Uint32(), rand.Uint32())
			new_file, err := root.CreateChildFile(tmpname, execBitFromACL(file_info.ACL()))
			if err != nil {
				return err
			}

			wc, err := new_file.OpenWritable()
			if err != nil {
				return err
			}

			if err := RestoreFile(s, file_info.(objects.TreeEntryFile).Ref, wc); err != nil {
				wc.Close()
				return err
			}
			wc.Close()

			if err := root.RenameChild(tmpname, name); err != nil {
				return err
			}
		case objects.TETDir:
			var subdir fs.Dir

			// Try to use existing directory
			child, err := root.GetChild(name)
			if err == nil {
				if child.Type() == fs.FDir {
					subdir = child.(fs.Dir)
				} else {
					if err := child.Delete(); err != nil {
						return err
					}
				}
			} else if err != os.ErrNotExist {
				return err
			}

			// Create directory, if it doesn't exist
			if subdir == nil {
				subdir, err = root.CreateChildDir(name)
				if err != nil {
					return err
				}
			}

			if err := RestoreDir(s, file_info.(objects.TreeEntryDir).Ref, subdir); err != nil {
				return err
			}
		case objects.TETSymlink:
			// Is there already a child of that name? If yes, delete it
			child, err := root.GetChild(name)
			if err == nil {
				if err := child.Delete(); err != nil {
					return err
				}
			} else if err != os.ErrNotExist {
				return err
			}

			if _, err := root.CreateChildSymlink(name, file_info.(objects.TreeEntrySymlink).Target); err != nil {
				return err
			}
		default:
			return fmt.Errorf("child '%s' of %s has unknown tree entry type %s", name, id, file_info.Type())
		}

		seen[name] = struct{}{}
	}

	// We now restored all children, we now need to remove the children of root, that shouldn't be there accoring to the backup
	children, err := root.Readdir()
	if err != nil {
		return err
	}
	for _, c := range children {
		_, ok := seen[c.Name()]
		if !ok {
			if err := c.Delete(); err != nil {
				return err
			}
		}
	}

	return nil
}