Merge branch 'master' of git.icynet.eu:evert/geobase

src/modules/countries/countries.controller.ts

@@ -9,8 +9,8 @@ export class CountriesController {
   constructor(private readonly service: CountriesService) {}
 
   @Get()
-  async getAllCountries(@Query() { q, fields }: CountriesQueryDto) {
-    return this.service.search(q, fields);
+  async getAllCountries(@Query() query: CountriesQueryDto) {
+    return this.service.search(query);
   }
 
   @Get(':iso')

src/modules/countries/countries.interfaces.ts

@@ -1,4 +1,6 @@
 export interface CountriesQueryDto {
   fields?: string[];
   q?: string;
+  limit?: string;
+  offset?: string;
 }

src/modules/countries/countries.service.ts

@@ -3,6 +3,7 @@ import { InjectRepository } from '@nestjs/typeorm';
 import { ILike, Repository } from 'typeorm';
 import { Country } from './countries.entity';
 import { intOrNull } from 'src/utils/int-or-null';
+import { CountriesQueryDto } from './countries.interfaces';
 
 const COUNTRIES_URL =
   'https://download.geonames.org/export/dump/countryInfo.txt';
@@ -83,9 +84,17 @@ export class CountriesService {
     });
   }
 
-  async search(query?: string, fields = ACCEPT_FIELDS) {
+  async search({
+    q: query,
+    fields = ACCEPT_FIELDS,
+    limit,
+    offset,
+  }: CountriesQueryDto) {
     const select = this.mapAllowedQuery(fields);
     const filter = {};
+    const take = Math.max(Math.min(intOrNull(limit as string) || 50, 1000), 1);
+
+    const skip = intOrNull(offset as string) || 0;
 
     if (query) {
       filter['country'] = ILike(`%${query}%`);
@@ -94,6 +103,8 @@ export class CountriesService {
     return this.countryRepository.find({
       where: filter,
       select,
+      skip,
+      take,
     });
   }
 

src/modules/geonames/geonames.controller.ts

@@ -1,5 +1,5 @@
 import { Controller, Get, Query } from '@nestjs/common';
-import { GeonameQuery } from './geonames.interfaces';
+import { FeatureQuery, GeonameQuery } from './geonames.interfaces';
 import { GeonamesService } from './geonames.service';
 
 @Controller({
@@ -12,4 +12,9 @@ export class GeonamesController {
   async search(@Query() query: GeonameQuery) {
     return this.service.search(query);
   }
+
+  @Get('features')
+  async features(@Query() query: FeatureQuery) {
+    return this.service.getFeatureList(query);
+  }
 }

src/modules/geonames/geonames.interfaces.ts

@@ -11,3 +11,8 @@ export type GeonameQuery = Record<keyof Geoname, unknown> & {
   offset?: string;
   q?: string;
 };
+
+export interface FeatureQuery {
+  q?: string;
+  class?: string;
+}

src/modules/geonames/geonames.service.ts

@@ -3,12 +3,17 @@ import { InjectDataSource, InjectRepository } from '@nestjs/typeorm';
 import { DataSource, ILike, In, Repository } from 'typeorm';
 import { pipeline } from 'stream/promises';
 import { join } from 'path';
-import { GeonameQuery, WorkDirectory } from './geonames.interfaces';
+import {
+  FeatureQuery,
+  GeonameQuery,
+  WorkDirectory,
+} from './geonames.interfaces';
 import { createInterface } from 'readline';
 import { Geoname } from './geonames.entity';
 import { intOrNull } from 'src/utils/int-or-null';
 import * as fs from 'fs';
 import * as unzipper from 'unzipper';
+import { GeoFeature } from './feature.dictionary';
 
 const GEONAMES_DUMP =
   'https://download.geonames.org/export/dump/allCountries.zip';
@@ -216,4 +221,33 @@ export class GeonamesService {
 
     Logger.log('Done!', context);
   }
+
+  getFeatureList({ q, class: className }: FeatureQuery) {
+    return Object.keys(GeoFeature)
+      .filter((code) => {
+        if (!className) return true;
+        return GeoFeature[code][0] === className.toUpperCase();
+      })
+      .filter((code) => {
+        if (!q) return true;
+        const lower = q.toLowerCase();
+        return (
+          code.includes(lower) ||
+          GeoFeature[code][1]?.includes(lower) ||
+          GeoFeature[code][2]?.includes(lower)
+        );
+      })
+      .reduce(
+        (prev, curr) => [
+          ...prev,
+          {
+            code: curr,
+            class: GeoFeature[curr][0],
+            name: GeoFeature[curr][1],
+            description: GeoFeature[curr][2],
+          },
+        ],
+        [],
+      );
+  }
 }

src/modules/timezones/timezones.module.ts

@@ -0,0 +1,8 @@
+import { Module } from '@nestjs/common';
+import { TimezonesService } from './timezones.service';
+
+@Module({
+  providers: [TimezonesService],
+  exports: [TimezonesService],
+})
+export class TimezonesModule {}

src/modules/timezones/timezones.service.ts

@@ -0,0 +1,4 @@
+import { Injectable } from '@nestjs/common';
+
+@Injectable()
+export class TimezonesService {}