@@ -3,7 +3,10 @@ import {readdirSync, statSync, existsSync, mkdirp, readFileSync, writeFileSync}
3
3
import * as path from 'path' ;
4
4
import * as admin from 'firebase-admin' ;
5
5
import * as firebase from 'firebase' ;
6
- import { openScreenshotsBucket , openFirebaseScreenshotsDatabase , connectFirebaseScreenshots } from '../util/firebase' ;
6
+ import {
7
+ openScreenshotsBucket ,
8
+ openFirebaseScreenshotsDatabase ,
9
+ connectFirebaseScreenshots } from '../util/firebase' ;
7
10
import { setGithubStatus } from '../util/github' ;
8
11
9
12
const imageDiff = require ( 'image-diff' ) ;
@@ -47,7 +50,8 @@ function updateResult(database: firebase.database.Database, prNumber: string, re
47
50
return getPullRequestRef ( database , prNumber ) . child ( 'result' ) . set ( result ) . then ( ( ) => result ) ;
48
51
}
49
52
50
- function getPullRequestRef ( database : firebase . database . Database | admin . database . Database , prNumber : string ) {
53
+ function getPullRequestRef ( database : firebase . database . Database | admin . database . Database ,
54
+ prNumber : string ) {
51
55
return database . ref ( FIREBASE_REPORT ) . child ( prNumber ) . child ( getSecureToken ( ) ) ;
52
56
}
53
57
@@ -65,13 +69,16 @@ function getScreenshotFiles(database: firebase.database.Database) {
65
69
mkdirp ( path . join ( SCREENSHOT_DIR , `golds` ) ) ;
66
70
mkdirp ( path . join ( SCREENSHOT_DIR , `diff` ) ) ;
67
71
68
- return database . ref ( 'screenshot/goldens' ) . once ( 'value' ) . then ( ( snapshot : firebase . database . DataSnapshot ) => {
72
+ return database . ref ( 'screenshot/goldens' ) . once ( 'value' )
73
+ . then ( ( snapshot : firebase . database . DataSnapshot ) => {
69
74
let counter = 0 ;
70
75
snapshot . forEach ( ( childSnapshot : firebase . database . DataSnapshot ) => {
71
76
let key = childSnapshot . key ;
72
77
let binaryData = new Buffer ( childSnapshot . val ( ) , 'base64' ) . toString ( 'binary' ) ;
73
78
writeFileSync ( `${ SCREENSHOT_DIR } /golds/${ key } .screenshot.png` , binaryData , 'binary' ) ;
74
- if ( ++ counter == snapshot . numChildren ( ) ) return true ;
79
+ if ( ++ counter == snapshot . numChildren ( ) ) {
80
+ return true ;
81
+ }
75
82
} ) ;
76
83
} ) . catch ( ( error : any ) => console . log ( error ) ) ;
77
84
}
@@ -101,7 +108,8 @@ function getSecureToken() {
101
108
* If the images are the test results, mode should be 'test'.
102
109
* If the images are the diff images generated, mode should be 'diff'.
103
110
*/
104
- function uploadScreenshotsData ( database : firebase . database . Database , mode : 'test' | 'diff' , prNumber : string ) {
111
+ function uploadScreenshotsData ( database : firebase . database . Database ,
112
+ mode : 'test' | 'diff' , prNumber : string ) {
105
113
let localDir = mode == 'diff' ? path . join ( SCREENSHOT_DIR , 'diff' ) : SCREENSHOT_DIR ;
106
114
let promises : any [ ] = [ ] ;
107
115
getLocalScreenshotFiles ( localDir ) . forEach ( ( file : string ) => {
@@ -192,4 +200,4 @@ function uploadScreenshots() {
192
200
promises . push ( bucket . upload ( fileName , { destination : destination } ) ) ;
193
201
} ) ;
194
202
return Promise . all ( promises ) ;
195
- }
203
+ }
0 commit comments