winpty/agent/Agent.h

#ifndef AGENT_H
#define AGENT_H

#include <QObject>
#include <QPoint>
#include <windows.h>

class Win32Console;
class QLocalSocket;
class Terminal;
class QTimer;
class ReadBuffer;

const int BUFFER_LINE_COUNT = 3000; // TODO: Use something like 9000.
const int MAX_CONSOLE_WIDTH = 500;

class Agent : public QObject
{
    Q_OBJECT
public:
    explicit Agent(const QString &controlPipeName,
                   const QString &dataPipeName,
                   int initialCols, int initialRows,
                   QObject *parent = 0);
    virtual ~Agent();

private:
    QLocalSocket *makeSocket(const QString &pipeName);
    void resetConsoleTracking(bool sendClear = true);

signals:

private slots:
    void controlSocketReadyRead();
    void handlePacket(ReadBuffer &packet);
    void handleStartProcessPacket(ReadBuffer &packet);
    void handleSetSizePacket(ReadBuffer &packet);
    void dataSocketReadyRead();
    void socketDisconnected();
    void pollTimeout();

private:
    void markEntireWindowDirty();
    void scanForDirtyLines();
    void resizeWindow(int cols, int rows);
    void scrapeOutput();
    void freezeConsole();
    void unfreezeConsole();
    void syncMarkerText(CHAR_INFO *output);
    int findSyncMarker();
    void createSyncMarker(int row);

private:
    Win32Console *m_console;
    QLocalSocket *m_controlSocket;
    QLocalSocket *m_dataSocket;
    Terminal *m_terminal;
    QTimer *m_timer;

    bool m_autoShutDown;

    int m_syncRow;
    int m_syncCounter;

    int m_scrapedLineCount;
    int m_scrolledCount;
    int m_maxBufferedLine;
    CHAR_INFO (*m_bufferData)[MAX_CONSOLE_WIDTH];
    int m_dirtyWindowTop;
    int m_dirtyLineCount;
};

#endif // AGENT_H