######################################################################
# README
#
# lookup.py - by Adam Doppelt - http://www.gurge.com/amd/
#
# Based on code and documentation by Scott Leighton, available at:
#
#   http://www.geocities.com/Heartland/Acres/3216/palmrecs.htm
#
# Since this script is based on his code, permission is granted to
# freely distribute it provided that no money changes hands, otherwise
# contact Scott at: helphand@pacbell.net
#
# Note: This code has only been tested with Palm 3.0 for Windows. Use
# at your own risk.
#
# INSTALLATION
#
# To install, modify the DIR variable to point at your pilot address
# directory.
#
# You will also need python, naturally, and maybe a batch file to
# launch lookup.py if you aren't using a decent shell.
######################################################################

import string
import struct
import sys

# This MUST BE CHANGED in order for the script to work.
DIR = "d:/Program Files/Palm/AdamP/address/"

#######################################################
# utilities
#######################################################

def chop(s):
    return s[0:len(s)-1]

#######################################################
# dat file
#######################################################

class DATFile:
    def __init__(self, path):
        self.file = open(path, "rb")
        
    def close():
        close(self.file)

    # read a long
    def long(self):
        return struct.unpack('<l', self.file.read(4))[0]

    # read a short
    def short(self):
        return struct.unpack('<h', self.file.read(2))[0]

    # read a cstring
    def cstring(self):
        len = ord(self.file.read(1))
        if (len == 0xFF):
            len = self.short()
        if (len == 0):
            return ""
        return self.file.read(len)

    # read a field
    def field(self):
        type = self.long();
        
        # long or boolean
        if type == 1 or type == 6:
            return self.long()
        
        # padded cstring
        if type == 5:
            # skip padding
            self.long()
            return self.cstring()
        
        raise Exception, "field type " + str(type) +  " not suppored."
    

#######################################################
# category
#######################################################

class Category:
    def __init__(self, file):
        self.index = file.long();
        self.id = file.long();
        self.dirty = file.long();
        self.name = file.cstring();
        self.short = file.cstring();


#######################################################
# file header
#######################################################

class Header:
    def __init__(self,file):
        self.version = file.long();
        self.filename = file.cstring();
        self.custom = file.cstring();
        self.nextID = file.long();

        self.categories = file.long();
        for i in range(self.categories):
            category = Category(file)
        
        self.id = file.long();
        self.perRow = file.long();
        self.idPos = file.long();
        self.statusPos = file.long();
        self.placementPos = file.long();
        
        self.fields = file.short();
        self.entries = []
        for i in range(self.fields):
            self.entries.append(file.short())
        
        self.count = file.long();

#######################################################
# address
#######################################################

class Address:
    PHONES = ("Work", "Home", "Fax", "Other", "Email", "Main",
              "Pager", "Mobile")
    
    pending = ""
    buf = ""
    
    def __init__(self,file):
        self.data = {}
        for key in ["id", "status", "pos", "last",
                    "first", "title", "company"]:
            self.data[key] = file.field()
            
        # the phone numbers
        for key in self.PHONES:
            self.data[key] = ""
            
        for i in range(5):
            key = file.field()
            self.data[self.PHONES[key]] = file.field()

        for key in ["address", "city", "state", "zip", "country", "note",
                    "private", "category", "custom1", "custom2", "custom3",
                    "custom4", "display"]:
            self.data[key] = file.field()

    def pretty(self):
        buf = ""

        # first/last/company
        if self.data["first"] != "" or self.data["last"] != "":
            if self.data["first"] == "":
                self.addValue(self.data["last"], 1)
            elif self.data["last"] == "":
                self.addValue(self.data["first"], 1)
            else:
                self.addValue(self.data["first"], 0)
                self.buf = self.buf + " "
                self.addValue(self.data["last"], 1)
                
                self.addValue(self.data["title"], 1)
                self.addValue(self.data["company"], 1)
        elif self.data["company"] != "":
            self.addValue(self.data["company"], 1)
            self.addValue(self.data["title"], 1)

        self.pending = "\n"

        # phone numbers
        for key in self.PHONES:
            self.namedValue(key, self.data[key], 0, 1)

        self.pending = "\n"

        # address
        self.addValue(self.data["address"], 1)
        if self.data["city"] != "":
            self.addValue(self.data["city"], 0)
            self.pending = ", "

        self.addValue(self.data["state"], 0)
        self.pending = " "
    
        self.addValue(self.data["zip"], 1)
        self.addValue(self.data["country"], 1)
    
        self.pending = "\n"

        # customs
        for key in ("custom1", "custom2", "custom3", "custom4"):
            self.namedValue(key, self.data[key], 0, 1)        
    
        self.pending = "\n"

        # note
        self.namedValue("Note", self.data["note"], 1, 0)

        # strip off the trailing newline
        if self.buf[len(self.buf) - 1] == '\n':
            self.buf = chop(self.buf)
            
        return self.buf

    def addPending(self):
        self.buf = self.buf + self.pending
        self.pending = ""

    def namedValue(self, key, value, keynewline, valuenewline):
        if value != "":
            self.addPending()
            self.buf = self.buf + key + ": "
            if (keynewline):
                self.addnewline()
            self.addValue(value, valuenewline)
                
    def addValue(self, value, newline):
        if value != "":
            self.addPending()
            self.buf = self.buf + value
            if (newline):
                self.addnewline()

    def addnewline(self):
        self.buf = self.buf + "\n"


#######################################################
# main
#######################################################

try:
    file = DATFile(DIR + "address.dat")
except (IOError), e:
    print "Could not read " + DIR + "address.dat :\n" + e[1] + "."
    sys.exit(1)

# determine the search string

search = ""
for i in range(1, len(sys.argv)):
    search = search + sys.argv[i] + " "
search = string.lower(chop(search))

# parse the file

header = Header(file)
for i in range(header.count / header.fields):
    a = Address(file)
    pretty = a.pretty()
    if string.find(string.lower(pretty), search) != -1:
        print "------------------------"
        print pretty