Skip to content

Commit

Permalink
feat: Added SendImageStorie and sendTextStorie
Browse files Browse the repository at this point in the history
  • Loading branch information
icleitoncosta committed Jul 19, 2022
1 parent b45c0b5 commit bd17cfd
Show file tree
Hide file tree
Showing 2 changed files with 118 additions and 1 deletion.
3 changes: 2 additions & 1 deletion src/api/layers/profile.layer.ts
Original file line number Diff line number Diff line change
Expand Up @@ -25,8 +25,9 @@ import {
evaluateAndReturn,
} from '../helpers';
import { CreateConfig } from '../../config/create-config';
import { StatusLayer } from './status.layer';

export class ProfileLayer extends HostLayer {
export class ProfileLayer extends StatusLayer {
constructor(public page: Page, session?: string, options?: CreateConfig) {
super(page, session, options);
}
Expand Down
116 changes: 116 additions & 0 deletions src/api/layers/status.layer.ts
Original file line number Diff line number Diff line change
@@ -0,0 +1,116 @@
/*
* This file is part of WPPConnect.
*
* WPPConnect is free software: you can redistribute it and/or modify
* it under the terms of the GNU Lesser General Public License as published by
* the Free Software Foundation, either version 3 of the License, or
* (at your option) any later version.
*
* WPPConnect is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU Lesser General Public License for more details.
*
* You should have received a copy of the GNU Lesser General Public License
* along with WPPConnect. If not, see <https://www.gnu.org/licenses/>.
*/

import { Page } from 'puppeteer';
import { CreateConfig } from '../../config/create-config';
import { BusinessLayer } from './business.layer';
import { HostLayer } from './host.layer';
import {
evaluateAndReturn,
base64MimeType,
fileToBase64,
downloadFileToBase64,
} from '../helpers';

export class StatusLayer extends HostLayer {
constructor(public page: Page, session?: string, options?: CreateConfig) {
super(page, session, options);
}

/**
* Get My Status stories
*/
public async getMyStatus() {
return await evaluateAndReturn(this.page, () => WPP.status.getMyStatus());
}

/**
* Send a image message to status stories
*
* @example
* ```javascript
* client.sendImageStatus('data:image/jpeg;base64,<a long base64 file...>');
* ```
* @param pathOrBase64 Path or base 64 image
*/
public async sendImageStatus(pathOrBase64: string) {
let base64: string = '';
if (pathOrBase64.startsWith('data:')) {
base64 = pathOrBase64;
} else {
let fileContent = await downloadFileToBase64(pathOrBase64, [
'image/gif',
'image/png',
'image/jpg',
'image/jpeg',
'image/webp',
]);
if (!fileContent) {
fileContent = await fileToBase64(pathOrBase64);
}
if (fileContent) {
base64 = fileContent;
}
}

if (!base64) {
const error = new Error('Empty or invalid file or base64');
Object.assign(error, {
code: 'empty_file',
});
throw error;
}

const mimeInfo = base64MimeType(base64);

if (!mimeInfo || !mimeInfo.includes('image')) {
const error = new Error(
'Not an image, allowed formats png, jpeg and webp'
);
Object.assign(error, {
code: 'invalid_image',
});
throw error;
}
return await evaluateAndReturn(
this.page,
({ base64 }) => {
WPP.status.sendImageStatus(base64);
},
{ base64 }
);
}

/**
* Send a text to status stories
*
* @example
* ```javascript
* client.sendTextStatus(`Bootstrap primary color: #0275d8`, { backgroundColor: '#0275d8', font: 2});
* ```
* @param pathOrBase64 Path or base 64 image
*/
public async sendTextStatus(text: string, options: string) {
return await evaluateAndReturn(
this.page,
({ text, options }) => {
WPP.status.sendTextStatus(text, options);
},
{ text, options }
);
}
}

0 comments on commit bd17cfd

Please sign in to comment.