Created
October 16, 2018 17:43
-
-
Save torresalmonte/ebd0ecd77e9b506053c8e93ea74f4d39 to your computer and use it in GitHub Desktop.
Firebase Cloud Function using Cloud Vision API for safe search filter
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| const functions = require('firebase-functions'); | |
| const mkdirp = require('mkdirp-promise'); | |
| const gcs = require('@google-cloud/storage')(); | |
| const spawn = require('child-process-promise').spawn; | |
| const path = require('path'); | |
| const os = require('os'); | |
| const fs = require('fs'); | |
| /** | |
| * When an image is uploaded we check if it is flagged as Adult or Violence by the Cloud Vision | |
| * API and if it is we blur it using ImageMagick. | |
| */ | |
| exports.blurOffensiveImages = functions.storage.object().onFinalize((object) => { | |
| const vision = require('@google-cloud/vision'); | |
| const client = new vision.ImageAnnotatorClient(); | |
| const bucket = gcs.bucket(object.bucket); | |
| const file = bucket.file(object.name); | |
| // Check the image content using the Cloud Vision API. | |
| return client.safeSearchDetection(`gs://${bucket.name}/${file.name}`) | |
| .then((data) => { | |
| const safeSearch = data[0]; | |
| console.log('SafeSearch results on image', safeSearch); | |
| if (safeSearch.safeSearchAnnotation.violence == 'VERY_LIKELY' || | |
| safeSearch.safeSearchAnnotation.violence == 'POSSIBLE' || | |
| safeSearch.safeSearchAnnotation.adult == 'VERY_LIKELY' || | |
| safeSearch.safeSearchAnnotation.adult == 'POSSIBLE') { | |
| return blurImage(object.name, object.bucket, object.metadata); | |
| } | |
| return null; | |
| }); | |
| }); | |
| /** | |
| * Blurs the given image located in the given bucket using ImageMagick. | |
| */ | |
| function blurImage(filePath, bucketName, metadata) { | |
| const tempLocalFile = path.join(os.tmpdir(), filePath); | |
| const tempLocalDir = path.dirname(tempLocalFile); | |
| const bucket = gcs.bucket(bucketName); | |
| // Create the temp directory where the storage file will be downloaded. | |
| return mkdirp(tempLocalDir).then(() => { | |
| console.log('Temporary directory has been created', tempLocalDir); | |
| // Download file from bucket. | |
| return bucket.file(filePath).download({destination: tempLocalFile}); | |
| }).then(() => { | |
| console.log('The file has been downloaded to', tempLocalFile); | |
| // Blur the image using ImageMagick. | |
| return spawn('convert', [tempLocalFile, '-channel', 'RGBA', '-blur', '0x8', tempLocalFile]); | |
| }).then(() => { | |
| console.log('Blurred image created at', tempLocalFile); | |
| // Uploading the Blurred image. | |
| return bucket.upload(tempLocalFile, { | |
| destination: filePath, | |
| metadata: {metadata: metadata}, // Keeping custom metadata. | |
| }); | |
| }).then(() => { | |
| console.log('Blurred image uploaded to Storage at', filePath); | |
| fs.unlinkSync(tempLocalFile); | |
| return console.log('Deleted local file', filePath); | |
| }); | |
| } |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment