Merge pull request #11 from CartoDB/internal-geocoder-server-functions

Internal geocoder server functions

server/extension/sql/10_admin0.sql

@@ -0,0 +1,41 @@
+-- Interface of the server extension
+
+CREATE OR REPLACE FUNCTION geocode_admin0_polygons(user_id name, rx_id bigint, country_name text)
+RETURNS Geometry AS $$
+    plpy.debug('Entering geocode_admin0_polygons')
+    plpy.debug('user_name = %s' % user_name)
+
+    #-- Access control
+    #-- TODO: this should be part of cdb python library
+    if user_name == 'publicuser':
+        plpy.error('The api_key must be provided')
+
+    #--TODO: rate limiting check
+    #--TODO: quota check
+
+    #-- Copied from the doc, see http://www.postgresql.org/docs/9.4/static/plpython-database.html
+    plan = plpy.prepare("SELECT _geocode_admin0_polygons($1) AS mypolygon", ["text"])
+    rv = plpy.execute(plan, [country_name], 1)
+
+    plpy.debug('Returning from Returning from geocode_admin0_polygons')
+    return rv[0]["mypolygon"]
+$$ LANGUAGE plpythonu;
+
+
+--------------------------------------------------------------------------------
+
+-- Implementation of the server extension
+
+CREATE OR REPLACE FUNCTION _geocode_admin0_polygons(country_name text)
+RETURNS Geometry AS $$
+  DECLARE
+    ret Geometry;
+  BEGIN
+    SELECT n.the_geom as geom INTO ret
+      FROM (SELECT q, lower(regexp_replace(q, '[^a-zA-Z\u00C0-\u00ff]+', '', 'g'))::text x
+        FROM (SELECT country_name q) g) d
+      LEFT OUTER JOIN admin0_synonyms s ON name_ = d.x
+      LEFT OUTER JOIN ne_admin0_v3 n ON s.adm0_a3 = n.adm0_a3 GROUP BY d.q, n.the_geom, s.adm0_a3;
+    RETURN ret;
+  END
+$$ LANGUAGE plpgsql;