import { EncodingType } from "./EncodingType";
import { ZlibEncoder } from "./ZlibEncoder";

export class Encoder {
    public encode(encoding: EncodingType, data: Buffer): Buffer {
        let encoded: Buffer;
        switch (encoding) {
            case EncodingType.Raw:
                encoded = data;
                break;
            case EncodingType.ZlibDeflate:
                encoded = new ZlibEncoder().encode(data);
                break;
            default:
                throw new Error("Unknown encoding type");
        }
        return Buffer.concat([Buffer.from([encoding]), encoded]);
    }

    public decode(buffer: Buffer): Buffer {
        const encoding = buffer.readUInt8(0);
        const raw = buffer.slice(1);
        switch (encoding) {
            case EncodingType.Raw:
                return raw;
            case EncodingType.ZlibDeflate:
                return new ZlibEncoder().decode(raw);
            default:
                throw new Error("Unknown encoding type");
        }
    }
}