Reorganisation

server/extension/Makefile

@@ -0,0 +1,26 @@
+# Makefile to generate the extension out of separate sql source files.
+# Once a version is released, it is not meant to be changed. E.g: once version 0.0.1 is out, it SHALL NOT be changed.
+EXTENSION = cdb_geocoder_server
+EXTVERSION = $(shell grep default_version $(EXTENSION).control | sed -e "s/default_version[[:space:]]*=[[:space:]]*'\([^']*\)'/\1/")
+
+DATA = $(EXTENSION)--$(EXTVERSION).sql
+
+REGRESS = $(notdir $(basename $(wildcard sql/*test.sql)))
+
+# postgres build stuff
+PG_CONFIG = pg_config
+PGXS := $(shell $(PG_CONFIG) --pgxs)
+include $(PGXS)
+
+
+SOURCES_DATA = $(wildcard sql/$(EXTVERSION)/*.sql)
+
+$(DATA): $(SOURCES_DATA)
+	rm -f $@
+	cat $(SOURCES_DATA) >> $@
+
+all: $(DATA)
+
+# Only meant for development time, do not use once a version is released
+devclean:
+	rm -f $(DATA)

server/extension/cdb_geocoder_server--0.0.1.sql

@@ -0,0 +1,37 @@
+-- Complain if script is sourced in psql, rather than via CREATE EXTENSION
+\echo Use "CREATE EXTENSION cdb_geocoder_server" to load this file. \quit
+-- Check if a given host is up by performing a ping -c 1 call.
+CREATE OR REPLACE FUNCTION cdb_geocoder_server._check_host(hostname TEXT)
+  RETURNS BOOLEAN
+AS $$
+  import os
+
+  response = os.system("ping -c 1 " + hostname)
+
+  return False if response else True
+$$ LANGUAGE plpythonu VOLATILE;
+
+-- Returns current pwd
+CREATE OR REPLACE FUNCTION cdb_geocoder_server._pwd()
+  RETURNS TEXT
+AS $$
+  import os
+
+  return os.getcwd()
+$$ LANGUAGE plpythonu VOLATILE;
+-- Check if a given host is up by performing a ping -c 1 call.
+CREATE OR REPLACE FUNCTION cdb_geocoder_server.geocode_street(searchtext TEXT, state TEXT DEFAULT NULL, country TEXT DEFAULT NULL)
+  RETURNS Geometry
+AS $$
+  from heremaps import heremapsgeocoder
+
+  geocoder = heremapsgeocoder.Geocoder(app_id, app_code)
+
+  results = geocoder.geocode_address(searchtext=searchtext, state=state, country=country)
+  coordinates = geocoder.extract_lng_lat_from_result(results[0])
+
+  plan = plpy.prepare("SELECT ST_SetSRID(ST_MakePoint($1, $2), 4326); ", ["double precision", "double precision"])
+  point = plpy.execute(plan, [coordinates[0], coordinates[1]], 1)[0]
+
+  return point['st_setsrid']
+$$ LANGUAGE plpythonu VOLATILE;

server/extension/cdb_geocoder_server.control

@@ -0,0 +1,6 @@
+# cdb geocoder server extension
+comment = 'CartoDB server geocoder extension'
+default_version = '0.0.1'
+requires = 'plpythonu, schema_triggers, postgis'
+superuser = true
+schema = cdb_geocoder_server

server/extension/expected/00_install_test.out

@@ -0,0 +1,6 @@
+-- Install dependencies
+CREATE EXTENSION postgis;
+CREATE EXTENSION schema_triggers;
+CREATE EXTENSION plpythonu;
+-- Install the extension
+CREATE EXTENSION cdb_geocoder_server;

server/extension/expected/10_helper_test.out

@@ -0,0 +1,14 @@
+-- Check that check_host is working
+select cdb_geocoder_server._check_host('cartodb.com');
+ _check_host 
+-------------
+ t
+(1 row)
+
+-- Check that check_pwd is working
+select cdb_geocoder_server._pwd();
+             _pwd             
+------------------------------
+ /var/lib/postgresql/9.3/main
+(1 row)
+

server/extension/sql/0.0.1/00_header.sql

@@ -0,0 +1,2 @@
+-- Complain if script is sourced in psql, rather than via CREATE EXTENSION
+\echo Use "CREATE EXTENSION cdb_geocoder_server" to load this file. \quit

server/extension/sql/0.0.1/10_helper.sql

@@ -0,0 +1,19 @@
+-- Check if a given host is up by performing a ping -c 1 call.
+CREATE OR REPLACE FUNCTION cdb_geocoder_server._check_host(hostname TEXT)
+  RETURNS BOOLEAN
+AS $$
+  import os
+
+  response = os.system("ping -c 1 " + hostname)
+
+  return False if response else True
+$$ LANGUAGE plpythonu VOLATILE;
+
+-- Returns current pwd
+CREATE OR REPLACE FUNCTION cdb_geocoder_server._pwd()
+  RETURNS TEXT
+AS $$
+  import os
+
+  return os.getcwd()
+$$ LANGUAGE plpythonu VOLATILE;

server/extension/sql/0.0.1/20_geocoding.sql

@@ -0,0 +1,16 @@
+-- Check if a given host is up by performing a ping -c 1 call.
+CREATE OR REPLACE FUNCTION cdb_geocoder_server.geocode_street(searchtext TEXT, state TEXT DEFAULT NULL, country TEXT DEFAULT NULL)
+  RETURNS Geometry
+AS $$
+  from heremaps import heremapsgeocoder
+
+  geocoder = heremapsgeocoder.Geocoder(app_id, app_code)
+
+  results = geocoder.geocode_address(searchtext=searchtext, state=state, country=country)
+  coordinates = geocoder.extract_lng_lat_from_result(results[0])
+
+  plan = plpy.prepare("SELECT ST_SetSRID(ST_MakePoint($1, $2), 4326); ", ["double precision", "double precision"])
+  point = plpy.execute(plan, [coordinates[0], coordinates[1]], 1)[0]
+
+  return point['st_setsrid']
+$$ LANGUAGE plpythonu VOLATILE;

server/extension/sql/00_install_test.sql

@@ -0,0 +1,8 @@
+-- Install dependencies
+CREATE EXTENSION postgis;
+CREATE EXTENSION schema_triggers;
+CREATE EXTENSION plpythonu;
+
+-- Install the extension
+CREATE EXTENSION cdb_geocoder_server;
+

server/extension/sql/10_helper_test.sql

@@ -0,0 +1,5 @@
+-- Check that check_host is working
+select cdb_geocoder_server._check_host('cartodb.com');
+
+-- Check that check_pwd is working
+select cdb_geocoder_server._pwd();