package steam

import (
	"archive/tar"
	"io"
	"os"
	"path/filepath"
	"strings"
)

// Package writes the package, returning bytes written and an error if any
func (g *Game) Package(wr io.Writer) error {
	if err := os.Chdir(g.LibraryPath); err != nil {
		return err
	}
	acf, err := FindACF(g.LibraryPath, g.Name)
	if err != nil {
		return err
	}

	twriter := tar.NewWriter(wr)

	for _, pth := range []string{"common/" + g.Name, acf} {
		if err := filepath.Walk(pth, TarWalkfn(twriter)); err != nil {
			return err
		}
	}

	return twriter.Flush()
}

func (l *Library) Extract(r io.Reader) error {
	if err := os.Chdir(l.Folder); err != nil {
		return err
	}

	treader := tar.NewReader(r)

	for {
		hdr, err := treader.Next()
		if err == io.EOF {
			// We've reached the end! Whoee
			break
		}
		if err != nil {
			return err
		}

		// Fix windows slashes...
		fileName := strings.Replace(hdr.Name, "\\", "/", -1)

		info := hdr.FileInfo()
		if info.IsDir() {
			// I don't like hard-coded permissions but it
			// it helps with overall platform compatibility
			if err = os.MkdirAll(fileName, 0775); err != nil {
				return err
			}
			continue
		}

		if err = os.MkdirAll(filepath.Dir(fileName), 0775); err != nil {
			return err
		}

		// Create a file handle to work with
		f, err := os.OpenFile(fileName, os.O_CREATE|os.O_TRUNC|os.O_WRONLY, 0664)
		if err != nil {
			return err
		}
		defer f.Close()
		if _, err := io.Copy(f, treader); err != nil {
			return err
		}

	}

	return nil
}

func (g *Game) Delete() error {
	if err := os.Chdir(g.LibraryPath); err != nil {
		return err
	}

	acf, err := FindACF(g.LibraryPath, g.Name)
	if err != nil {
		return err
	}
	if err := os.Remove(acf); err != nil {
		return err
	}
	if err := os.RemoveAll("common/" + g.Name); err != nil {
		return err
	}

	return nil
}

// Get the Size of a game in a pretty format
func (g Game) GetSize() string {
	return formatBytes(g.Size)
}

func (g *Game) setSizeInfo() error {
	pth := filepath.Join(g.LibraryPath, "common", g.Name)
	return filepath.Walk(pth, func(path string, info os.FileInfo, err error) error {
		if err != nil {
			return err
		}

		if info.Mode().IsRegular() {
			g.Size += info.Size()
		}

		return nil
	})
}