Added ability to extract games. Initial code to delete games. Initial basic progress on a command line application.

4 files changed, 191 insertions, 6 deletions

diff --git a/archive/unarchive.go b/archive/unarchive.go
new file mode 100644
index 0000000..5089370
--- /dev/null
+++ b/archive/unarchive.go
@@ -0,0 +1,64 @@
+package archive
+
+import (
+	"archive/tar"
+	fp "path/filepath"
+
+	"io"
+	"os"
+)
+
+type Unarchive struct {
+	Input     string
+	inputFile *os.File
+	tarReader *tar.Reader
+}
+
+// Extracts a tar arcive to the current working directory
+// This will overwrite everything. So be careful
+func (u *Unarchive) UnTar() error {
+	var err error
+	u.inputFile, err = os.Open(u.Input)
+	if err != nil {
+		return err
+	}
+	defer u.inputFile.Close()
+	u.tarReader = tar.NewReader(u.inputFile)
+
+	for {
+		hdr, err := u.tarReader.Next()
+		if err == io.EOF {
+			// We've reached the end! Whoee
+			break
+		}
+		if err != nil {
+			return err
+		}
+
+		info := hdr.FileInfo()
+		if info.IsDir() {
+			if err = os.MkdirAll(hdr.Name, info.Mode()); err != nil {
+				return err
+			}
+			continue
+		}
+
+		if err = os.MkdirAll(fp.Dir(hdr.Name), info.Mode()); err != nil {
+			return err
+		}
+
+		// Create a file handle to work with
+		// f, err := os.Create(hdr.Name)
+		f, err := os.OpenFile(hdr.Name, os.O_CREATE|os.O_TRUNC|os.O_WRONLY, info.Mode())
+		if err != nil {
+			return err
+		}
+		defer f.Close()
+		if _, err := io.Copy(f, u.tarReader); err != nil {
+			return err
+		}
+
+	}
+
+	return nil
+}
diff --git a/steam-export-cli.go b/steam-export-cli.go
new file mode 100644
index 0000000..46f96dd
--- /dev/null
+++ b/steam-export-cli.go
@@ -0,0 +1,70 @@
+package main
+
+import (
+	"errors"
+	"flag"
+	"fmt"
+	"os"
+
+	"git.riedstra.us/mitch/steam-export/steam"
+)
+
+var (
+	errorHandling flag.ErrorHandling = flag.ExitOnError
+
+	defaultLib string         = "C:\\Program Files (x86)\\Steam\\steamapps"
+	steamLib   *steam.Library = &steam.Library{}
+)
+
+func parseArgs(args []string) error {
+	if len(args) < 2 {
+		return errors.New("Not enough arguments")
+	}
+	switch a := args[1]; a {
+	case "list":
+		return listGames(args[2:])
+	default:
+		printHelp()
+	}
+
+	return nil
+}
+
+func printHelp() {
+	fmt.Printf(`Program usage:
+steam-export-cli $subcommand $options
+Subcommands:
+	list -l $steam_library
+	package -l $steam_library -f $output_file [ -G $game_index | -g "game_name" ]
+	extract -l $steam_library -f $input_file
+	delete -l $steam_library [ -G $game_index | -g $game_name ]
+	server -c $config_file -l $steam_library -L $listen_addr`)
+
+}
+
+func listGames(args []string) error {
+	fl := flag.NewFlagSet("list", errorHandling)
+	lib := fl.String("l", defaultLib,
+		"Path to library in question. All the way to the 'steamapps' folder")
+
+	l := defaultLib
+	fl.Parse(args)
+	if fl.Parsed() {
+		l = string(*lib)
+	}
+
+	if err := steamLib.ProcessLibrary(l); err != nil {
+		return err
+	}
+
+	fmt.Println(steamLib)
+
+	return nil
+}
+
+func main() {
+	if err := parseArgs(os.Args); err != nil {
+		fmt.Fprintln(os.Stderr, err)
+		printHelp()
+	}
+}
diff --git a/steam/package.go b/steam/package.go
index 72d29ee..8a4ff8b 100644
--- a/steam/package.go
+++ b/steam/package.go
@@ -1,27 +1,30 @@
 package steam
 
 import (
-	// "fmt"
 	"git.riedstra.us/mitch/steam-export/archive"
 	"os"
-	// "strings"
+	"path/filepath"
 )
 
-func (l *Library) PackageGame(g string) error {
+func (l *Library) PackageGameToFile(index int, file string) error {
+	g := l.Games[index]
+
 	working_dir, err := os.Getwd()
 	if err != nil {
 		return err
 	}
-	output := working_dir + "/" + g + ".tar"
+	// output := working_dir + "/" + g + ".tar"
+	output, err := filepath.Abs(file)
+	if err != nil {
+		return err
+	}
 
 	os.Chdir(l.Folder)
 	acf, err := l.FindACF(g)
 	if err != nil {
 		return err
 	}
-	// acf = strings.Replace(acf, l.Folder, "", -1)
 	input := []string{"common/" + g, acf}
-	// fmt.Fprintf(os.Stderr, "input arguments for archive: %s\n", input)
 	a := archive.Archive{Output: output, Input: input}
 	err = a.Tar()
 	if err != nil {
@@ -32,3 +35,49 @@ func (l *Library) PackageGame(g string) error {
 
 	return nil
 }
+
+func (l *Library) ExtractGameFromFile(f string) error {
+	working_dir, err := os.Getwd()
+	if err != nil {
+		return err
+	}
+
+	f, err = filepath.Abs(f)
+	if err != nil {
+		return err
+	}
+
+	if err = os.Chdir(l.Folder); err != nil {
+		return err
+	}
+	u := &archive.Unarchive{
+		Input: f,
+	}
+	if err := u.UnTar(); err != nil {
+		return err
+	}
+
+	if err = os.Chdir(working_dir); err != nil {
+		return err
+	}
+	return nil
+}
+
+/*
+func (l *Library) DeleteGame(i index) error {
+	g := l.Games[i]
+
+	working_dir, err := os.Getwd()
+	if err != nil {
+		return err
+	}
+
+	// os.RemoveAll()
+
+	if err = os.Chdir(working_dir); err != nil {
+		return err
+	}
+
+	return nil
+}
+*/
diff --git a/steam/steam.go b/steam/steam.go
index 9715f38..ea27e5a 100644
--- a/steam/steam.go
+++ b/steam/steam.go
@@ -46,6 +46,8 @@ func (s *Library) ProcessLibrary(r string) error {
 				s.Games = append(s.Games, f.Name())
 			}
 		}
+	} else {
+		return errors.New(fmt.Sprintf("No common directory in: %s", r))
 	}
 	return nil
 }