Map Geo Code Maintenance.fh_lua

--[[
@Title: Map - Maintain Place Geo-Code Overrides
@Author: Jane Taubman
@LastUpdated: July 2011
@Version: 1.1
@Description: Maintain Place overrides for GeoCoding for Map Plugins.
Works with the Map Live Events plugin allowing the entry of longitude and latitudes for place names, or simply modern names for places so they can be recognised by Google maps.
]]
  require('luacom')
------------------------------------------------------ Start Table Load Save
-- require '_tableloadsave'
--[[
   Save Table to File/Stringtable
   Load Table from File/Stringtable
   v 0.94
   
   Lua 5.1 compatible
   
   Userdata and indices of these are not saved
   Functions are saved via string.dump, so make sure it has no upvalues
   References are saved
   ----------------------------------------------------
   table.save( table [, filename] )
   
   Saves a table so it can be called via the table.load function again
   table must a object of type 'table'
   filename is optional, and may be a string representing a filename or true/1
   
   table.save( table )
      on success: returns a string representing the table (stringtable)
      (uses a string as buffer, ideal for smaller tables)
   table.save( table, true or 1 )
      on success: returns a string representing the table (stringtable)
      (uses io.tmpfile() as buffer, ideal for bigger tables)
   table.save( table, "filename" )
      on success: returns 1
      (saves the table to file "filename")
   on failure: returns as second argument an error msg
   ----------------------------------------------------
   table.load( filename or stringtable )
   
   Loads a table that has been saved via the table.save function
   
   on success: returns a previously saved table
   on failure: returns as second argument an error msg
   ----------------------------------------------------
   
   chillcode, http://lua-users.org/wiki/SaveTableToFile
   Licensed under the same terms as Lua itself.
]]--
do
   -- declare local variables
   --// exportstring( string )
   --// returns a "Lua" portable version of the string
   local function exportstring( s )
      s = string.format( "%q",s )
      -- to replace
      s = string.gsub( s,"\\\n","\\n" )
      s = string.gsub( s,"\r","\\r" )
      s = string.gsub( s,string.char(26),"\"..string.char(26)..\"" )
      return s
   end
--// The Save Function
function table.save(  tbl,filename )
   local charS,charE = "   ","\n"
   local file,err
   -- create a pseudo file that writes to a string and return the string
   if not filename then
      file =  { write = function( self,newstr ) self.str = self.str..newstr end, str = "" }
      charS,charE = "",""
   -- write table to tmpfile
   elseif filename == true or filename == 1 then
      charS,charE,file = "","",io.tmpfile()
   -- write table to file
   -- use io.open here rather than io.output, since in windows when clicking on a file opened with io.output will create an error
   else
      file,err = io.open( filename, "w" )
      if err then return _,err end
   end
   -- initiate variables for save procedure
   local tables,lookup = { tbl },{ [tbl] = 1 }
   file:write( "return {"..charE )
   for idx,t in ipairs( tables ) do
      if filename and filename ~= true and filename ~= 1 then
         file:write( "-- Table: {"..idx.."}"..charE )
      end
      file:write( "{"..charE )
      local thandled = {}
      for i,v in ipairs( t ) do
         thandled[i] = true
         -- escape functions and userdata
         if type( v ) ~= "userdata" then
            -- only handle value
            if type( v ) == "table" then
               if not lookup[v] then
                  table.insert( tables, v )
                  lookup[v] = #tables
               end
               file:write( charS.."{"..lookup[v].."},"..charE )
            elseif type( v ) == "function" then
               file:write( charS.."loadstring("..exportstring(string.dump( v )).."),"..charE )
            else
               local value =  ( type( v ) == "string" and exportstring( v ) ) or tostring( v )
               file:write(  charS..value..","..charE )
            end
         end
      end
      for i,v in pairs( t ) do
         -- escape functions and userdata
         if (not thandled[i]) and type( v ) ~= "userdata" then
            -- handle index
            if type( i ) == "table" then
               if not lookup[i] then
                  table.insert( tables,i )
                  lookup[i] = #tables
               end
               file:write( charS.."[{"..lookup[i].."}]=" )
            else
               local index = ( type( i ) == "string" and "["..exportstring( i ).."]" ) or string.format( "[%d]",i )
               file:write( charS..index.."=" )
            end
            -- handle value
            if type( v ) == "table" then
               if not lookup[v] then
                  table.insert( tables,v )
                  lookup[v] = #tables
               end
               file:write( "{"..lookup[v].."},"..charE )
            elseif type( v ) == "function" then
               file:write( "loadstring("..exportstring(string.dump( v )).."),"..charE )
            else
               local value =  ( type( v ) == "string" and exportstring( v ) ) or tostring( v )
               file:write( value..","..charE )
            end
         end
      end
      file:write( "},"..charE )
   end
   file:write( "}" )
   -- Return Values
   -- return stringtable from string
   if not filename then
      -- set marker for stringtable
      return file.str.."--|"
   -- return stringttable from file
   elseif filename == true or filename == 1 then
      file:seek ( "set" )
      -- no need to close file, it gets closed and removed automatically
      -- set marker for stringtable
      return file:read( "*a" ).."--|"
   -- close file and return 1
   else
      file:close()
      return 1
   end
end

--// The Load Function
function table.load( sfile )
   -- catch marker for stringtable
   if string.sub( sfile,-3,-1 ) == "--|" then
      tables,err = loadstring( sfile )
   else
      tables,err = loadfile( sfile )
   end
   if err then return _,err
   end
   tables = tables()
   for idx = 1,#tables do
      local tolinkv,tolinki = {},{}
      for i,v in pairs( tables[idx] ) do
         if type( v ) == "table" and tables[v[1]] then
            table.insert( tolinkv,{ i,tables[v[1]] } )
         end
         if type( i ) == "table" and tables[i[1]] then
            table.insert( tolinki,{ i,tables[i[1]] } )
         end
      end
      -- link values, first due to possible changes of indices
      for _,v in ipairs( tolinkv ) do
         tables[idx][v[1]] = v[2]
      end
      -- link indices
      for _,v in ipairs( tolinki ) do
         tables[idx][v[2]],tables[idx][v[1]] =  tables[idx][v[1]],nil
      end
   end
   return tables[1]
end
-- close do
end
------------------------------------------------------ End Table Load Save
---------------------------------------- Extract XML Data from request
function getXMLvalue(source,key)
local startkey = '<'..key..'>'
local endkey = '</'..key..'>'
local pattern = startkey..'.-'..endkey
local result = source:match(pattern)
result = result:gsub(startkey,'')
result = result:gsub(endkey,'')
return result
end

function create_dialog_PlaceMaint()
  local containers = {}
-- Input Fields
txtModern =     iup.text{minsize = "400x25"}
txtPlace  =     iup.text{minsize = "400x25",readonly='YES',bgcolor='DLGBGCOLOR'}
txtLong   =     iup.text{minsize = "400x25"}
txtLat    =     iup.text{minsize = "400x25"} 
lstPlaces =     iup.list{
      dropdown   = "YES",
		dropexpand = "YES",
		VISIBLE_ITEMS = '20',
	   sort       = "YES"
    }
--- Define Buttons
btnSave =  iup.button{
      title = "Save and Close",
      minsize = "150x25",
    }
btnCancel = iup.button{
      title = "Cancel",
      minsize = "75x25",
    }
btnLookup = iup.button{
      title = "Lookup",
      minsize = "75x25",
    }

  containers[4] = iup.hbox{
    iup.label{title = "Placename",minsize = "100x25"},
    txtPlace,
  }

  containers[5] = iup.hbox{
    iup.label{title = "Modern Name",minsize = "100x25"},
	txtModern,
  }

  containers[6] = iup.hbox{
    iup.label{title = "Long",minsize = "100x25"},
    txtLong,}

  containers[7] = iup.hbox{
    iup.label{title = "Lat",minsize = "100x25"},
	 txtLat,}



containers[8] = iup.hbox{
btnLookup,
		btnUpdate,
      btnAdd,
		btnSave,
		btnCancel
  }
  containers[3] = iup.vbox{
	lstPlaces,
    containers[4],
    containers[5],
    containers[7],
    containers[6],
    containers[8],
    expandchildren = "YES",
    gap = "5",
  }

  containers[2] = iup.frame{
    containers[3],
  }

  containers[1] = iup.dialog{
	 title = 'Maintain Geo-Location Overrides',
    containers[2],
    
  }

  return containers[1]
end


dlg = create_dialog_PlaceMaint()
function btnCancel:action()
  return iup.Close() 
end

function url_encode(str)
  if (str) then
    str = string.gsub (str, "\n", "\r\n")
    str = string.gsub (str, "([^%w ])",
        function (c) return string.format ("%%%02X", string.byte(c)) end)
    str = string.gsub (str, " ", "+")
  end
  return str	
end

function btnLookup:action()
local http = luacom.CreateObject("winhttp.winhttprequest.5.1")
local place = url_encode(txtModern.value) 

http:Open("GET","http://where.yahooapis.com/geocode?q="..place.."&appid=NVkuOj5a",false)
http:Send()
local XML = http.Responsetext 
print(XML)
print(getXMLvalue(XML,'latitude'))
print(getXMLvalue(XML,'longitude'))
		txtLong.value = getXMLvalue(XML,'longitude')
	   	txtLat.value = getXMLvalue(XML,'latitude')
end
function lstPlaces:action(t, i, v)
if txtPlace ~= '' then
-- Update current record before moving
  tblPlaceOverride[txtPlace.value] = {place = txtPlace.value, modern=txtModern.value, long = txtLong.value,lat=txtLat.value}
end
  txtPlace.value = t
	if tblPlaceOverride[t] == nil then
		txtModern.value = t
		txtLong.value = ''
	   	txtLat.value = ''
	else
		txtModern.value = tblPlaceOverride[t]['modern']
		txtLong.value = tblPlaceOverride[t]['long']
	   	txtLat.value = tblPlaceOverride[t]['lat']
	end

  return iup.DEFAULT
end


function btnSave:action()
	if txtPlace.value ~= '' then
	  tblPlaceOverride[txtPlace.value] = {place = txtPlace.value, modern=txtModern.value, long = txtLong.value,lat=txtLat.value}
   end
	table.save(tblPlaceOverride,settingsFile)  
   return iup.Close() 
end

--  Build Places from Gedcom File
function buildplacelist()
local ptr = fhNewItemPtr()
local tblPlaces = {}
for i,rcdtype in pairs({'INDI','FAM'}) do
	ptr:MoveToFirstRecord(rcdtype)
	while ptr:IsNotNull() do
	if fhGetTag(ptr) == 'PLAC' then
		tblPlaces[fhGetValueAsText(ptr)] = (tblPlaces[fhGetValueAsText(ptr)] or 0) +1
	end
	ptr:MoveNextSpecial()
	end
end
return tblPlaces
end
--------------------------------------------------------------- Main Code
settingsFile = fhGetPluginDataFileName()
tblPlaceOverride =table.load(settingsFile)
if tblPlaceOverride == nil then
	tblPlaceOverride = {}
	table.save(tblPlaceOverride,settingsFile)
end
tblplaces = buildplacelist()

dlg:show()
for strplace,count in pairs(tblplaces) do
	iup.SetAttribute(lstPlaces, "APPENDITEM", strplace) 
end

iup.MainLoop()