pollo-mcp
Version:
Pollo AI Model Context Protocol (MCP) Server for video generation
34 lines (33 loc) • 1.56 kB
JavaScript
var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
return new (P || (P = Promise))(function (resolve, reject) {
function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
step((generator = generator.apply(thisArg, _arguments || [])).next());
});
};
import fs from 'node:fs';
import path from 'node:path';
import axios from 'axios';
import * as api from './api.js';
export function downloadFile(directory, url) {
return __awaiter(this, void 0, void 0, function* () {
fs.mkdirSync(directory, { recursive: true });
const response = yield axios.get(url, { responseType: 'stream' });
const saveAs = path.join(directory, path.basename(url));
response.data.pipe(fs.createWriteStream(saveAs));
return saveAs;
});
}
export function uploadImage(filepath) {
return __awaiter(this, void 0, void 0, function* () {
const { sign, accessURL } = yield api.signImage(path.basename(filepath));
yield axios.put(sign, fs.readFileSync(filepath), {
headers: {
'Content-Type': 'image/png',
},
});
return accessURL;
});
}