Skip to content

Commit a311be4

Browse files
sulcosulco
committed
feat(schematics): ng-add schematics
A possibility to install and configure the package with `ng add angularfire2` command closes angular#1676
1 parent 78a9736 commit a311be4

25 files changed

+2230
-7
lines changed

‎package.json

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -10,7 +10,10 @@
1010
"karma": "karma start",
1111
"test:universal": "npm run build && cp -R dist/packages-dist test/universal-test/node_modules/angularfire2 && cd test/universal-test && npm run prerender",
1212
"delayed_karma": "sleep 10 && karma start",
13-
"build": "rm -rf dist && node tools/build.js",
13+
"build": "rm -rf dist && node tools/build.js && npm run schematics",
14+
"schematics": "npm run schematics:build && npm run schematics:copy",
15+
"schematics:build": "tsc -p ./schematics/tsconfig.json",
16+
"schematics:copy": "ncp schematics/src dist/packages-dist/schematics",
1417
"changelog": "conventional-changelog -p angular -i CHANGELOG.md -s -r 1"
1518
},
1619
"keywords": [
@@ -29,11 +32,14 @@
2932
},
3033
"homepage": "https://github.com/angular/angularfire2#readme",
3134
"dependencies": {
35+
"@angular-devkit/core": "0.7.3",
36+
"@angular-devkit/schematics": "0.7.3",
3237
"@angular/common": "^6.0.0",
3338
"@angular/compiler": "^6.0.0",
3439
"@angular/core": "^6.0.0",
3540
"@angular/platform-browser": "^6.0.0",
3641
"@angular/platform-browser-dynamic": "^6.0.0",
42+
"bufferutil": "^3.0.3",
3743
"firebase": "^5.0.3",
3844
"rxjs": "^6.0.0",
3945
"ws": "^3.3.2",

‎schematics/.editorconfig

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,2 @@
1+
[*]
2+
indent_size = 2

‎schematics/.gitignore

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,3 @@
1+
*.d.ts
2+
*.js
3+
*.js.map

‎schematics/src/collection.json

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,10 @@
1+
{
2+
"$schema": "../node_modules/@angular-devkit/schematics/collection-schema.json",
3+
"schematics": {
4+
"ng-add": {
5+
"description": "Adds AngularFire to an application",
6+
"factory": "./ng-add",
7+
"schema": "./ng-add/schema.json"
8+
}
9+
}
10+
}

‎schematics/src/ng-add/index.ts

Lines changed: 125 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,125 @@
1+
import { chain, Rule, SchematicContext, SchematicsException, Tree } from '@angular-devkit/schematics';
2+
import * as ts from 'typescript';
3+
import { addPackageToPackageJson } from '../utils/package';
4+
import { NodePackageInstallTask } from '@angular-devkit/schematics/tasks';
5+
import { angularfireVersion, firebaseVersion } from '../utils/libs-version';
6+
import { addEnvironmentEntry } from '../utils/environment';
7+
import { getProjectPath } from '../utils/project';
8+
import { Schema } from './schema';
9+
import { parseName } from '../schematics-core/utility/parse-name';
10+
import { buildRelativePath, findModuleFromOptions } from '../schematics-core/utility/find-module';
11+
import { addImportToModule, insertImport } from '../schematics-core/utility/ast-utils';
12+
import { InsertChange } from '../schematics-core/utility/change';
13+
14+
export default function add(options: Schema): Rule {
15+
return chain([
16+
install(),
17+
addConfig(),
18+
addToNgModule(options),
19+
])
20+
}
21+
22+
export function install(): Rule {
23+
return (host: Tree, context: SchematicContext) => {
24+
addPackageToPackageJson(host, 'dependencies', 'firebase', firebaseVersion);
25+
addPackageToPackageJson(host, 'dependencies', '@angular/fire', angularfireVersion);
26+
context.addTask(new NodePackageInstallTask());
27+
return host;
28+
};
29+
}
30+
31+
function addConfig(): Rule {
32+
return (host: Tree) => {
33+
const firebaseConfig = `
34+
firebase: {
35+
apiKey: '<your-key>',
36+
authDomain: '<your-project-authdomain>',
37+
databaseURL: '<your-database-URL>',
38+
projectId: '<your-project-id>',
39+
storageBucket: '<your-storage-bucket>',
40+
messagingSenderId: '<your-messaging-sender-id>'
41+
},`;
42+
addEnvironmentEntry(host, 'environment.ts', firebaseConfig);
43+
return host;
44+
}
45+
}
46+
47+
function addToNgModule(options: Schema): Rule {
48+
return (host: Tree) => {
49+
options.path = parseName(getProjectPath(host, options), '').path;
50+
51+
if (options.module) {
52+
options.module = findModuleFromOptions(host, {
53+
name: '',
54+
module: options.module,
55+
path: options.path,
56+
});
57+
}
58+
59+
const modulePath = options.module;
60+
61+
if (!modulePath) {
62+
return host;
63+
}
64+
65+
if (!host.exists(modulePath)) {
66+
throw new Error(`Specified module path ${modulePath} does not exist`);
67+
}
68+
69+
const text = host.read(modulePath);
70+
if (text === null) {
71+
throw new SchematicsException(`File ${modulePath} does not exist.`);
72+
}
73+
const sourceText = text.toString('utf-8');
74+
75+
const source = ts.createSourceFile(
76+
modulePath,
77+
sourceText,
78+
ts.ScriptTarget.Latest,
79+
true
80+
);
81+
82+
const environmentsPath = buildRelativePath(
83+
modulePath,
84+
`${options.path}/environments/environment`
85+
);
86+
87+
const AngularFireNgModuleImport = addImportToModule(
88+
source,
89+
modulePath,
90+
options.firebaseApp
91+
? `AngularFireModule.initializeApp(environment.firebase, '${options.firebaseApp}')`
92+
: `AngularFireModule.initializeApp(environment.firebase)`,
93+
''
94+
).shift();
95+
96+
const coreImports = [
97+
insertImport(source, modulePath, 'AngularFireModule', '@angular/fire'),
98+
insertImport(source, modulePath, 'environment', environmentsPath),
99+
AngularFireNgModuleImport,
100+
];
101+
102+
const individualImports = options.all
103+
? [
104+
insertImport(source, modulePath, 'AngularFirestoreModule', '@angular/fire/firestore'),
105+
insertImport(source, modulePath, 'AngularFireStorageModule', '@angular/fire/storage'),
106+
insertImport(source, modulePath, 'AngularFireAuthModule', '@angular/fire/auth'),
107+
addImportToModule(source, modulePath, 'AngularFirestoreModule', '').shift(),
108+
addImportToModule(source, modulePath, 'AngularFireStorageModule', '').shift(),
109+
addImportToModule(source, modulePath, 'AngularFireAuthModule', '').shift(),
110+
]
111+
: [];
112+
113+
const changes = [...coreImports, ...individualImports];
114+
115+
const recorder = host.beginUpdate(modulePath);
116+
for (const change of changes) {
117+
if (change instanceof InsertChange) {
118+
recorder.insertLeft(change.pos, change.toAdd);
119+
}
120+
}
121+
host.commitUpdate(recorder);
122+
123+
return host;
124+
}
125+
}

‎schematics/src/ng-add/schema.json

Lines changed: 33 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,33 @@
1+
{
2+
"$schema": "http://json-schema.org/schema",
3+
"id": "SchematicsAngularFire",
4+
"title": "Adds AngularFire to a project",
5+
"type": "object",
6+
"properties": {
7+
"module": {
8+
"type": "string",
9+
"default": "app",
10+
"description": "Allows specification of the declaring module.",
11+
"alias": "m",
12+
"subtype": "filepath"
13+
},
14+
"path": {
15+
"type": "string",
16+
"format": "path",
17+
"description": "The path to the directory that contains the declaring module.",
18+
"visible": false
19+
},
20+
"firebaseApp": {
21+
"type": "string",
22+
"description": "A custom FirebaseApp name.",
23+
"alias": "f"
24+
},
25+
"all": {
26+
"type": "boolean",
27+
"description": "Add all AngularFire modules (auth, db, storage).",
28+
"alias": "a",
29+
"default": false
30+
}
31+
},
32+
"required": []
33+
}

‎schematics/src/ng-add/schema.ts

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,6 @@
1+
export interface Schema {
2+
module?: string;
3+
path?: string;
4+
firebaseApp?: string;
5+
all?: boolean;
6+
}

0 commit comments

Comments
 (0)