/**
 * @license
 * Copyright Google Inc. All Rights Reserved.
 *
 * Use of this source code is governed by an MIT-style license that can be
 * found in the LICENSE file at https://angular.io/license
 */

import {AotCompiler, AotCompilerHost, AotCompilerOptions, CompileSummaryKind, GeneratedFile, createAotCompiler} from '@angular/compiler';
import {fakeAsync, tick} from '@angular/core/testing';

import {MockDirectory, compile, setup} from './test_util';

describe('aot summaries for jit', () => {
  let angularFiles = setup();

  function compileApp(rootDir: MockDirectory, options: {useSummaries?: boolean} = {}):
      {genFiles: GeneratedFile[], outDir: MockDirectory} {
    let result: {genFiles: GeneratedFile[], outDir: MockDirectory} = null !;
    let error: Error|null = null;
    compile([rootDir, angularFiles], options).then((r) => result = r, (e) => error = e);
    tick();
    if (error) {
      throw error;
    }
    return result;
  }

  it('should create @Injectable summaries', fakeAsync(() => {
       const appDir = {
         'app.module.ts': `
        import { Injectable } from '@angular/core';

        export class Dep {}

        @Injectable()
        export class MyService {
          constructor(d: Dep) {}
        }
      `
       };
       const rootDir = {'app': appDir};

       const genFile =
           compileApp(rootDir).genFiles.find(f => f.genFileUrl === '/app/app.module.ngsummary.ts');

       expect(genFile.source).toContain(`import * as import0 from '/app/app.module'`);
       expect(genFile.source).toContain('export function MyServiceNgSummary()');
       // Note: CompileSummaryKind.Injectable = 3
       expect(genFile.source).toMatch(/summaryKind: 3,\s*type: \{\s*reference: import0.MyService/);
       expect(genFile.source).toContain('token: {identifier: {reference: import0.Dep}}');
     }));

  it('should create @Pipe summaries', fakeAsync(() => {
       const appDir = {
         'app.module.ts': `
        import { Pipe, NgModule } from '@angular/core';

        export class Dep {}

        @Pipe({name: 'myPipe'})
        export class MyPipe {
          constructor(d: Dep) {}
        }

        @NgModule({declarations: [MyPipe]})
        export class MyModule {}
      `
       };
       const rootDir = {'app': appDir};

       const genFile =
           compileApp(rootDir).genFiles.find(f => f.genFileUrl === '/app/app.module.ngsummary.ts');

       expect(genFile.source).toContain(`import * as import0 from '/app/app.module'`);
       expect(genFile.source).toContain('export function MyPipeNgSummary()');
       // Note: CompileSummaryKind.Pipe = 1
       expect(genFile.source).toMatch(/summaryKind: 0,\s*type: \{\s*reference: import0.MyPipe/);
       expect(genFile.source).toContain('token: {identifier: {reference: import0.Dep}}');
     }));

  it('should create @Directive summaries', fakeAsync(() => {
       const appDir = {
         'app.module.ts': `
        import { Directive, NgModule } from '@angular/core';

        export class Dep {}

        @Directive({selector: '[myDir]'})
        export class MyDirective {
          constructor(a: Dep) {}
        }

        @NgModule({declarations: [MyDirective]})
        export class MyModule {}
      `
       };
       const rootDir = {'app': appDir};

       const genFile =
           compileApp(rootDir).genFiles.find(f => f.genFileUrl === '/app/app.module.ngsummary.ts');

       expect(genFile.source).toContain(`import * as import0 from '/app/app.module'`);
       expect(genFile.source).toContain('export function MyDirectiveNgSummary()');
       // Note: CompileSummaryKind.Directive = 1
       expect(genFile.source)
           .toMatch(/summaryKind: 1,\s*type: \{\s*reference: import0.MyDirective/);
       expect(genFile.source).toContain('token: {identifier: {reference: import0.Dep}}');
     }));

  it('should create @NgModule summaries', fakeAsync(() => {
       const appDir = {
         'app.module.ts': `
        import { NgModule } from '@angular/core';

        export class Dep {}

        @NgModule()
        export class MyModule {
          constructor(d: Dep) {}
        }
      `
       };
       const rootDir = {'app': appDir};

       const genFile =
           compileApp(rootDir).genFiles.find(f => f.genFileUrl === '/app/app.module.ngsummary.ts');

       expect(genFile.source).toContain(`import * as import0 from '/app/app.module'`);
       expect(genFile.source).toContain('export function MyModuleNgSummary()');
       // Note: CompileSummaryKind.NgModule = 2
       expect(genFile.source).toMatch(/summaryKind: 2,\s*type: \{\s*reference: import0.MyModule/);
       expect(genFile.source).toContain('token: {identifier: {reference: import0.Dep}}');
     }));

  it('should embed useClass provider summaries in @Directive summaries', fakeAsync(() => {
       const appDir = {
         'app.service.ts': `
        import { Injectable } from '@angular/core';

        export class Dep {}

        @Injectable()
        export class MyService {
          constructor(d: Dep) {}
        }
      `,
         'app.module.ts': `
        import { Directive, NgModule } from '@angular/core';
        import { MyService } from './app.service';

        @Directive({
          selector: '[myDir]',
          providers: [MyService]
        })
        export class MyDirective {}

        @NgModule({declarations: [MyDirective]})
        export class MyModule {}
      `
       };
       const rootDir = {'app': appDir};

       const genFile =
           compileApp(rootDir).genFiles.find(f => f.genFileUrl === '/app/app.module.ngsummary.ts');

       expect(genFile.source).toMatch(/useClass: \{\s*reference: import1.MyService/);
       // Note: CompileSummaryKind.Injectable = 3
       expect(genFile.source).toMatch(/summaryKind: 3,\s*type: \{\s*reference: import1.MyService/);
       expect(genFile.source).toContain('token: {identifier: {reference: import1.Dep}}');
     }));

  it('should embed useClass provider summaries into @NgModule summaries', fakeAsync(() => {
       const appDir = {
         'app.service.ts': `
        import { Injectable } from '@angular/core';

        export class Dep {}

        @Injectable()
        export class MyService {
          constructor(d: Dep) {}
        }
      `,
         'app.module.ts': `
        import { NgModule } from '@angular/core';
        import { MyService } from './app.service';

        @NgModule({
          providers: [MyService]
        })
        export class MyModule {}
      `
       };
       const rootDir = {'app': appDir};

       const genFile =
           compileApp(rootDir).genFiles.find(f => f.genFileUrl === '/app/app.module.ngsummary.ts');

       expect(genFile.source).toMatch(/useClass: \{\s*reference: import1.MyService/);
       // Note: CompileSummaryKind.Injectable = 3
       expect(genFile.source).toMatch(/summaryKind: 3,\s*type: \{\s*reference: import1.MyService/);
       expect(genFile.source).toContain('token: {identifier: {reference: import1.Dep}}');
     }));

  it('should reference declared @Directive and @Pipe summaries in @NgModule summaries',
     fakeAsync(() => {
       const appDir = {
         'app.module.ts': `
        import { Directive, Pipe, NgModule } from '@angular/core';

        @Directive({selector: '[myDir]'})
        export class MyDirective {}

        @Pipe({name: 'myPipe'})
        export class MyPipe {}

        @NgModule({declarations: [MyDirective, MyPipe]})
        export class MyModule {}
      `
       };
       const rootDir = {'app': appDir};

       const genFile =
           compileApp(rootDir).genFiles.find(f => f.genFileUrl === '/app/app.module.ngsummary.ts');

       expect(genFile.source)
           .toMatch(
               /export function MyModuleNgSummary()[^;]*,\s*MyDirectiveNgSummary,\s*MyPipeNgSummary\s*\]\s*;/);
     }));

  it('should reference imported @NgModule summaries in @NgModule summaries', fakeAsync(() => {
       const appDir = {
         'app.module.ts': `
        import { NgModule } from '@angular/core';

        @NgModule()
        export class MyImportedModule {}

        @NgModule({imports: [MyImportedModule]})
        export class MyModule {}
      `
       };
       const rootDir = {'app': appDir};

       const genFile =
           compileApp(rootDir).genFiles.find(f => f.genFileUrl === '/app/app.module.ngsummary.ts');

       expect(genFile.source)
           .toMatch(
               /export function MyModuleNgSummary()[^;]*,\s*MyImportedModuleNgSummary\s*\]\s*;/);
     }));

  it('should create and use reexports for imported NgModules ' +
         'accross compilation units',
     fakeAsync(() => {
       const lib1In = {
         'lib1': {
           'module.ts': `
          import { NgModule } from '@angular/core';

          @NgModule()
          export class Lib1Module {}
        `,
           'reexport.ts': `
          import { NgModule } from '@angular/core';

          @NgModule()
          export class ReexportModule {}

          export const reexports: any[] = [ ReexportModule ];
        `,
         }
       };
       const {outDir: lib2In, genFiles: lib1Gen} = compileApp(lib1In, {useSummaries: true});

       lib2In['lib2'] = {
         'module.ts': `
          import { NgModule } from '@angular/core';
          import { Lib1Module } from '../lib1/module';

          @NgModule({
            imports: [Lib1Module]
          })
          export class Lib2Module {}
        `,
         'reexport.ts': `
        import { reexports as reexports_lib1 } from '../lib1/reexport';
        export const reexports: any[] = [ reexports_lib1 ];
        `,
       };
       const {outDir: lib3In, genFiles: lib2Gen} = compileApp(lib2In, {useSummaries: true});

       const lib2ModuleNgSummary = lib2Gen.find(f => f.genFileUrl === '/lib2/module.ngsummary.ts');
       const lib2ReexportNgSummary =
           lib2Gen.find(f => f.genFileUrl === '/lib2/reexport.ngsummary.ts');

       // ngsummaries should add reexports for imported NgModules from a direct dependency
       expect(lib2ModuleNgSummary.source)
           .toContain(
               `export {Lib1ModuleNgSummary as Lib1Module_1NgSummary} from '/lib1/module.ngsummary'`);
       // ngsummaries should add reexports for reexported values from a direct dependency
       expect(lib2ReexportNgSummary.source)
           .toContain(
               `export {ReexportModuleNgSummary as ReexportModule_2NgSummary} from '/lib1/reexport.ngsummary'`);

       lib3In['lib3'] = {
         'module.ts': `
          import { NgModule } from '@angular/core';
          import { Lib2Module } from '../lib2/module';
          import { reexports } from '../lib2/reexport';

          @NgModule({
            imports: [Lib2Module, reexports]
          })
          export class Lib3Module {}
        `,
         'reexport.ts': `
        import { reexports as reexports_lib2 } from '../lib2/reexport';
        export const reexports: any[] = [ reexports_lib2 ];
        `,
       };

       const lib3Gen = compileApp(lib3In, {useSummaries: true}).genFiles;
       const lib3ModuleNgSummary = lib3Gen.find(f => f.genFileUrl === '/lib3/module.ngsummary.ts');
       const lib3ReexportNgSummary =
           lib3Gen.find(f => f.genFileUrl === '/lib3/reexport.ngsummary.ts');

       // ngsummary.ts files should use the reexported values from direct and deep deps
       expect(lib3ModuleNgSummary.source)
           .toContain(`import * as import4 from '/lib2/module.ngsummary'`);
       expect(lib3ModuleNgSummary.source)
           .toContain(`import * as import5 from '/lib2/reexport.ngsummary'`);
       expect(lib3ModuleNgSummary.source)
           .toMatch(
               /export function Lib3ModuleNgSummary()[^;]*,\s*import4.Lib1Module_1NgSummary,\s*import4.Lib2ModuleNgSummary,\s*import5.ReexportModule_2NgSummary\s*\]\s*;/);

       // ngsummaries should add reexports for imported NgModules from a deep dependency
       expect(lib3ModuleNgSummary.source)
           .toContain(
               `export {Lib1Module_1NgSummary as Lib1Module_1NgSummary,Lib2ModuleNgSummary as Lib2Module_2NgSummary} from '/lib2/module.ngsummary'`);
       // ngsummaries should add reexports for reexported values from a deep dependency
       expect(lib3ReexportNgSummary.source)
           .toContain(
               `export {ReexportModule_2NgSummary as ReexportModule_3NgSummary} from '/lib2/reexport.ngsummary'`);
     }));
});