From dfc150aa39b80aa2e8024a2681cae40d5b8942fb Mon Sep 17 00:00:00 2001
From: Andrey Smirnov <Smirnov.Andrey@gmail.com>
Date: Mon, 16 Dec 2013 23:04:36 +0400
Subject: [PATCH] Preliminary support for Package, very simple.

---
 debian/package.go      | 79 ++++++++++++++++++++++++++++++++++++++++++
 debian/package_test.go | 48 +++++++++++++++++++++++++
 2 files changed, 127 insertions(+)
 create mode 100644 debian/package.go
 create mode 100644 debian/package_test.go

diff --git a/debian/package.go b/debian/package.go
new file mode 100644
index 00000000..9cad393c
--- /dev/null
+++ b/debian/package.go
@@ -0,0 +1,79 @@
+package debian
+
+import (
+	"bytes"
+	debc "github.com/smira/godebiancontrol"
+	"github.com/ugorji/go/codec"
+	"strings"
+)
+
+// Package is single instance of Debian package
+//
+// TODO: support source & binary
+type Package struct {
+	Name         string
+	Version      string
+	Filename     string
+	Architecture string
+	Depends      []string
+	PreDepends   []string
+	Suggests     []string
+	Recommends   []string
+	Extra        debc.Paragraph
+}
+
+func parseDependencies(input debc.Paragraph, key string) []string {
+	value, ok := input[key]
+	if !ok {
+		return nil
+	}
+
+	delete(input, key)
+
+	return strings.Split(value, ", ")
+}
+
+// NewPackageFromControlFile creates Package from parsed Debian control file
+func NewPackageFromControlFile(input debc.Paragraph) *Package {
+	result := &Package{
+		Name:         input["Package"],
+		Version:      input["Version"],
+		Filename:     input["Filename"],
+		Architecture: input["Architecture"],
+	}
+
+	delete(input, "Package")
+	delete(input, "Version")
+	delete(input, "Filename")
+	delete(input, "Architecture")
+
+	result.Depends = parseDependencies(input, "Depends")
+	result.PreDepends = parseDependencies(input, "Pre-Depends")
+	result.Suggests = parseDependencies(input, "Suggests")
+	result.Recommends = parseDependencies(input, "Recommends")
+
+	result.Extra = input
+
+	return result
+}
+
+// Key returns unique key identifying package
+func (p *Package) Key() []byte {
+	return []byte(p.Name + " " + p.Version)
+}
+
+// Encode does msgpack encoding of Package
+func (p *Package) Encode() []byte {
+	var buf bytes.Buffer
+
+	encoder := codec.NewEncoder(&buf, &codec.MsgpackHandle{})
+	encoder.Encode(p)
+
+	return buf.Bytes()
+}
+
+// Decode decodes msgpack representation into Package
+func (p *Package) Decode(input []byte) error {
+	decoder := codec.NewDecoderBytes(input, &codec.MsgpackHandle{})
+	return decoder.Decode(p)
+}
diff --git a/debian/package_test.go b/debian/package_test.go
new file mode 100644
index 00000000..207151c0
--- /dev/null
+++ b/debian/package_test.go
@@ -0,0 +1,48 @@
+package debian
+
+import (
+	debc "github.com/smira/godebiancontrol"
+	. "launchpad.net/gocheck"
+)
+
+var packagePara = debc.Paragraph{"Source": "alien-arena", "Depends": "libc6 (>= 2.7), alien-arena-data (>= 7.40)", "Filename": "pool/contrib/a/alien-arena/alien-arena-common_7.40-2_i386.deb", "SHA1": "46955e48cad27410a83740a21d766ce362364024", "SHA256": "eb4afb9885cba6dc70cccd05b910b2dbccc02c5900578be5e99f0d3dbf9d76a5", "Priority": "extra", "Maintainer": "Debian Games Team <pkg-games-devel@lists.alioth.debian.org>", "Description": "Common files for Alien Arena client and server ALIEN ARENA is a standalone 3D first person online deathmatch shooter\n crafted from the original source code of Quake II and Quake III, released\n by id Software under the GPL license. With features including 32 bit\n graphics, new particle engine and effects, light blooms, reflective water,\n hi resolution textures and skins, hi poly models, stain maps, ALIEN ARENA\n pushes the envelope of graphical beauty rivaling today's top games.\n .\n This package installs the common files for Alien Arena.\n", "Homepage": "http://red.planetarena.org", "Tag": "role::app-data, role::shared-lib, special::auto-inst-parts", "Installed-Size": "456", "Version": "7.40-2", "Replaces": "alien-arena (<< 7.33-1)", "Size": "187518", "MD5sum": "1e8cba92c41420aa7baa8a5718d67122", "Package": "alien-arena-common", "Section": "contrib/games", "Architecture": "i386"}
+
+type PackageSuite struct {
+	para debc.Paragraph
+}
+
+var _ = Suite(&PackageSuite{})
+
+func (s *PackageSuite) SetUpTest(c *C) {
+	s.para = make(debc.Paragraph)
+	for k, v := range packagePara {
+		s.para[k] = v
+	}
+}
+
+func (s *PackageSuite) TestNewFromPara(c *C) {
+	p := NewPackageFromControlFile(s.para)
+
+	c.Check(p.Name, Equals, "alien-arena-common")
+	c.Check(p.Version, Equals, "7.40-2")
+	c.Check(p.Architecture, Equals, "i386")
+	c.Check(p.Filename, Equals, "pool/contrib/a/alien-arena/alien-arena-common_7.40-2_i386.deb")
+	c.Check(p.Depends, DeepEquals, []string{"libc6 (>= 2.7)", "alien-arena-data (>= 7.40)"})
+	c.Check(p.Suggests, IsNil)
+}
+
+func (s *PackageSuite) TestKey(c *C) {
+	p := NewPackageFromControlFile(s.para)
+
+	c.Check(p.Key(), DeepEquals, []byte("alien-arena-common 7.40-2"))
+}
+
+func (s *PackageSuite) TestEncodeDecode(c *C) {
+	p := NewPackageFromControlFile(s.para)
+	encoded := p.Encode()
+	p2 := &Package{}
+	err := p2.Decode(encoded)
+
+	c.Assert(err, IsNil)
+	c.Assert(p2, DeepEquals, p)
+}