export definitions (Driver and Session) and fix subscribe

Author: pocesar

lib/v1/driver.d.ts

@@ -42,7 +42,7 @@ declare class Driver {
   close(): void;
 }
 
-declare class _ConnectionStreamObserver extends StreamObserver {
+declare class _ConnectionStreamObserver<T> extends StreamObserver<T> {
   constructor(driver: Driver, conn: Connection);
 
   onError(error: Error): void;

lib/v1/error.d.ts

@@ -7,7 +7,7 @@ declare class Neo4jError extends Error {
   code: string;
   message: string;
 
-  constructor(message: any, code: string)
+  constructor(message: any, code: string);
 }
 
 export {

lib/v1/index.d.ts

@@ -4,6 +4,7 @@ import { Neo4jError, SERVICE_UNAVAILABLE, SESSION_EXPIRED } from "./error";
 import Result from "./result";
 import ResultSummary from "./result-summary";
 import Record from "./record";
+import Session from './session';
 import { Driver, READ, WRITE, AuthCredentials, ConfigurationOptions } from "./driver";
 import RoutingDriver from "./routing-driver";
 import VERSION from "../version";
@@ -63,8 +64,10 @@ declare const forExport: {
   types: typeof types;
   session: typeof session;
   error: typeof error;
+  Driver: Driver;
   AuthCredentials: AuthCredentials;
   ConfigurationOptions: ConfigurationOptions;
+  Session: Session;
 };
 
 export {
@@ -78,6 +81,7 @@ export {
   session,
   error,
   AuthCredentials,
+  Session,
   ConfigurationOptions,
 }
 

lib/v1/internal/stream-observer.d.ts

@@ -1,6 +1,14 @@
+import ResultSummary from '../result-summary'
+
 declare type ErrorTransformer = <T extends Error>(err: Error) => T;
 
-declare class StreamObserver {
+export declare interface StreamObserverInterface<T> {
+  onNext?(record: T): void;
+  onCompleted?(summary?: ResultSummary): void;
+  onError?(err: Error): void;
+}
+
+declare class StreamObserver<T> implements StreamObserverInterface<T> {
   constructor(errorTransformer: ErrorTransformer);
 }
 

lib/v1/record.d.ts

@@ -10,8 +10,9 @@ declare class Record<T> {
   constructor(keys: string[], fields: string[], fieldLookup: { [index: string]: string });
 
   forEach(visitor: Visitor<T>): void;
-  toObject(): Partial<{ [ret: string]: T>;
-  get<K extends T>(key: string | number): K;
+  toObject(): Partial<{ [ret: string]: T }>;
+  get<K extends keyof T>(key: K): T[K];
+  get<K extends T>(key: number): K;
   has(key: string | number): boolean;
 }
 

lib/v1/result-summary.d.ts

@@ -1,7 +1,7 @@
 import Integer, { int, isInt } from "./integer";
 
 declare class ResultSummary {
-  statement: {text: string, parameters: { [index: string]: any }};
+  statement: { text: string, parameters: { [index: string]: any } };
   statementType: string;
   counters: StatementStatistic;
   //for backwards compatibility, remove in future version

lib/v1/result.d.ts

@@ -1,6 +1,6 @@
 import ResultSummary from "./result-summary";
 import Record from "./record";
-import StreamObserver from "./internal/stream-observer";
+import StreamObserver, { StreamObserverInterface } from "./internal/stream-observer";
 
 declare type ResultType<T> = {
   records: Array<Record<T>>;
@@ -10,21 +10,21 @@ declare type ResultType<T> = {
 declare type PromiseResult<T> = PromiseLike<ResultType<T>>
 
 declare class Result<T, Params extends { [index: string]: any }> extends Promise<ResultType<T>> {
-  _streanObserver: StreamObserver;
+  _streamObserver: StreamObserver<T>;
   _p: PromiseResult<T> | null;
   _statement: string;
   _parameters: Params;
   _metaSupplier: () => any;
 
   constructor(
-    streamObserver: StreamObserver,
+    streamObserver: StreamObserver<T>,
     statement: string,
     parameters?: Params,
     metaSupplier?: () => any
   );
 
   protected _createPromise(): PromiseResult<T>;
-  subscribe(observer: StreamObserver): void;
+  subscribe(observer: StreamObserverInterface<Record<T>>): void;
 }
 
 export default Result;

lib/v1/session.d.ts

@@ -23,7 +23,7 @@ declare class Session {
   protected _onRunFailure(): Function;
 }
 
-declare class _RunObserver extends StreamObserver {
+declare class _RunObserver<T> extends StreamObserver<T> {
   constructor(
     onError: Function
   );

lib/v1/transaction.d.ts

@@ -21,7 +21,7 @@ declare class Transaction {
 declare function _runDiscardAll<T>(
   msg: any,
   connectionPromise: PromiseLike<Connection>,
-  observer: StreamObserver
+  observer: StreamObserver<T>
 ): Result<T, {}>;
 
 export default Transaction;