package main

import (
	"encoding/json"
	"errors"
	"fmt"
	"os"

	"github.com/restic/restic"
	"github.com/restic/restic/backend"
)

func init() {
	commands["cat"] = commandCat
}

func commandCat(be backend.Server, key *restic.Key, args []string) error {
	if len(args) != 2 {
		return errors.New("usage: cat [blob|tree|snapshot|key|lock] ID")
	}

	tpe := args[0]

	id, err := backend.ParseID(args[1])
	if err != nil {
		id = nil

		if tpe != "snapshot" {
			return err
		}

		// find snapshot id with prefix
		id, err = backend.Find(be, backend.Snapshot, args[1])
		if err != nil {
			return err
		}
	}

	ch, err := restic.NewContentHandler(be, key)
	if err != nil {
		return err
	}

	err = ch.LoadAllMaps()
	if err != nil {
		return err
	}

	switch tpe {
	case "blob":
		// try id
		data, err := ch.Load(backend.Data, id)
		if err == nil {
			_, err = os.Stdout.Write(data)
			return err
		}

		// try storage id
		buf, err := be.Get(backend.Data, id)
		if err != nil {
			return err
		}

		// decrypt
		buf, err = key.Decrypt(buf)
		if err != nil {
			return err
		}

		_, err = os.Stdout.Write(buf)
		return err

	case "tree":
		var tree restic.Tree
		// try id
		err := ch.LoadJSON(backend.Tree, id, &tree)
		if err != nil {
			// try storage id
			buf, err := be.Get(backend.Tree, id)
			if err != nil {
				return err
			}

			// decrypt
			buf, err = key.Decrypt(buf)
			if err != nil {
				return err
			}

			// unmarshal
			err = json.Unmarshal(backend.Uncompress(buf), &tree)
			if err != nil {
				return err
			}
		}

		buf, err := json.MarshalIndent(&tree, "", "  ")
		if err != nil {
			return err
		}

		fmt.Println(string(buf))

		return nil
	case "map":
		var bl restic.BlobList
		err := ch.LoadJSONRaw(backend.Map, id, &bl)
		if err != nil {
			return err
		}

		buf, err := json.MarshalIndent(&bl, "", "  ")
		if err != nil {
			return err
		}

		fmt.Println(string(buf))

		return nil
	case "snapshot":
		var sn restic.Snapshot

		err = ch.LoadJSONRaw(backend.Snapshot, id, &sn)
		if err != nil {
			return err
		}

		buf, err := json.MarshalIndent(&sn, "", "  ")
		if err != nil {
			return err
		}

		fmt.Println(string(buf))

		return nil
	case "key":
		data, err := be.Get(backend.Key, id)
		if err != nil {
			return err
		}

		var key restic.Key
		err = json.Unmarshal(data, &key)
		if err != nil {
			return err
		}

		buf, err := json.MarshalIndent(&key, "", "  ")
		if err != nil {
			return err
		}

		fmt.Println(string(buf))

		return nil
	case "lock":
		return errors.New("not yet implemented")
	default:
		return errors.New("invalid type")
	}
}