package class6_1_StreamDecorator;

import java.io.FilterWriter;
import java.io.IOException;
import java.io.Writer;

public class EncryptorWriter extends FilterWriter {

    private int key;

    public EncryptorWriter(int key, Writer out) {
        super(out);
        this.key = key;
    }

    @Override
    public void write(String str, int off, int len) throws IOException {
        for(int i = off; i<len; i++) {
            write(str.charAt(i));
        }
    }

    @Override
    public void write(int c) throws IOException {
        int encrypted = (c + key)%255;
        super.write(encrypted); // The bug: I had c on this line instead of encrypted.
    }
}