| /* |
| * Licensed to the Apache Software Foundation (ASF) under one or more |
| * contributor license agreements. See the NOTICE file distributed with |
| * this work for additional information regarding copyright ownership. |
| * The ASF licenses this file to You under the Apache License, Version 2.0 |
| * (the "License"); you may not use this file except in compliance with |
| * the License. You may obtain a copy of the License at |
| * |
| * http://www.apache.org/licenses/LICENSE-2.0 |
| * |
| * Unless required by applicable law or agreed to in writing, software |
| * distributed under the License is distributed on an "AS IS" BASIS, |
| * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| * See the License for the specific language governing permissions and |
| * limitations under the License. |
| */ |
| |
| import { |
| SpellBase, |
| SpellResult, |
| DefaultDisplayType, |
| } from 'zeppelin-spell'; |
| |
| import 'whatwg-fetch'; |
| |
| export default class TranslatorSpell extends SpellBase { |
| constructor() { |
| super("%translator"); |
| } |
| |
| /** |
| * Consumes text and return `SpellResult`. |
| * |
| * @param paragraphText {string} which doesn't include magic |
| * @param config {Object} |
| * @return {SpellResult} |
| */ |
| interpret(paragraphText, config) { |
| const parsed = this.parseConfig(paragraphText); |
| const auth = config['access-token']; |
| const source = parsed.source; |
| const target = parsed.target; |
| const text = parsed.text; |
| |
| /** |
| * SpellResult.add() |
| * - accepts not only `string` but also `promise` as a parameter |
| * - allows to add multiple output using the `add()` function |
| */ |
| const result = new SpellResult() |
| .add('<h4>Translation Result</h4>', DefaultDisplayType.HTML) |
| // or use display system implicitly like |
| // .add('%html <h4>Translation From English To Korean</h4>') |
| .add(this.translate(source, target, auth, text)); |
| return result; |
| } |
| |
| parseConfig(text) { |
| const pattern = /^\s*(\S+)-(\S+)\s*([\S\s]*)/g; |
| const match = pattern.exec(text); |
| |
| if (!match) { |
| throw new Error(`Failed to parse configuration. See README`); |
| } |
| |
| return { |
| source: match[1], |
| target: match[2], |
| text: match[3], |
| } |
| } |
| |
| translate(source, target, auth, text) { |
| return fetch('https://translation.googleapis.com/language/translate/v2', { |
| method: 'POST', |
| headers: { |
| 'Content-Type': 'application/json', |
| 'Authorization': `Bearer ${auth}`, |
| }, |
| body: JSON.stringify({ |
| 'q': text, |
| 'source': source, |
| 'target': target, |
| 'format': 'text' |
| }) |
| }).then(response => { |
| if (response.status === 200) { |
| return response.json() |
| } |
| throw new Error(`https://translation.googleapis.com/language/translate/v2 ${response.status} (${response.statusText})`); |
| }).then((json) => { |
| const extracted = json.data.translations.map(t => { |
| return t.translatedText; |
| }); |
| return extracted.join('\n'); |
| }); |
| } |
| } |
| |