-
Notifications
You must be signed in to change notification settings - Fork 0
/
index.js
53 lines (49 loc) · 1.51 KB
/
index.js
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
const { exec, spawn } = require('child_process')
const GCLOUD_ACCOUNT_KEY_FILE = process.env.npm_package_config_gcloud_service_account_file
const GCLOUD_PROJECT_ID = process.env.npm_package_config_gcloud_project_id
const SLACK_API_TOKEN = process.env.npm_package_config_slack_api_token
const SLACK_CHANNEL = '@shouhei.tai'
const IMAGE_FILE_PATH = './captured.jpeg'
const DETECTION_ENTITY_ID = '/m/027g6wt'
const webcam = spawn('fswebcam', ['-D', '2', '-r', '1280x720', IMAGE_FILE_PATH])
webcam.on('close', code => {
if(code === 0) {
labelDetection()
}
})
function labelDetection() {
const client= require('google-cloud')
const vision = client.vision({
projectId: GCLOUD_PROJECT_ID,
keyFilename: GCLOUD_ACCOUNT_KEY_FILE
})
vision.labelDetection({ source: { filename: IMAGE_FILE_PATH} })
.then((results) => {
const labels = results[0].labelAnnotations
labels.some(label => {
if(label.mid === DETECTION_ENTITY_ID) {
sendSlack()
return true
}
})
})
.catch((err) => {
console.error('ERROR:', err)
})
}
function sendSlack() {
const fs = require('fs')
const Slack = require('node-slack-upload')
const slack = new Slack(SLACK_API_TOKEN)
slack.uploadFile({
file: fs.createReadStream(IMAGE_FILE_PATH),
filetype: 'json',
title: 'サンタクロース',
initialComment: 'サンタクロースの侵入を検知しました',
channels: SLACK_CHANNEL
}, function(err, data) {
if(err) {
console.error(err)
}
})
}