mirror of https://github.com/vector-im/riot-web
				
				
				
			
		
			
				
	
	
		
			98 lines
		
	
	
		
			3.3 KiB
		
	
	
	
		
			TypeScript
		
	
	
			
		
		
	
	
			98 lines
		
	
	
		
			3.3 KiB
		
	
	
	
		
			TypeScript
		
	
	
/*
 | 
						|
Copyright 2019, 2022 The Matrix.org Foundation C.I.C.
 | 
						|
 | 
						|
Licensed under the Apache License, Version 2.0 (the "License");
 | 
						|
you may not use this file except in compliance with the License.
 | 
						|
You may obtain a copy of the License at
 | 
						|
 | 
						|
    http://www.apache.org/licenses/LICENSE-2.0
 | 
						|
 | 
						|
Unless required by applicable law or agreed to in writing, software
 | 
						|
distributed under the License is distributed on an "AS IS" BASIS,
 | 
						|
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 | 
						|
See the License for the specific language governing permissions and
 | 
						|
limitations under the License.
 | 
						|
*/
 | 
						|
 | 
						|
import { Room, MatrixClient, RoomMember } from "matrix-js-sdk/src/matrix";
 | 
						|
 | 
						|
import AutocompleteWrapperModel, { UpdateCallback } from "../../src/editor/autocomplete";
 | 
						|
import { Caret } from "../../src/editor/caret";
 | 
						|
import { PillPart, Part, PartCreator } from "../../src/editor/parts";
 | 
						|
import DocumentPosition from "../../src/editor/position";
 | 
						|
 | 
						|
export class MockAutoComplete {
 | 
						|
    public _updateCallback;
 | 
						|
    public _partCreator;
 | 
						|
    public _completions;
 | 
						|
    public _part: Part | null;
 | 
						|
 | 
						|
    constructor(updateCallback: UpdateCallback, partCreator: PartCreator, completions: PillPart[]) {
 | 
						|
        this._updateCallback = updateCallback;
 | 
						|
        this._partCreator = partCreator;
 | 
						|
        this._completions = completions;
 | 
						|
        this._part = null;
 | 
						|
    }
 | 
						|
 | 
						|
    close() {
 | 
						|
        this._updateCallback({ close: true });
 | 
						|
    }
 | 
						|
 | 
						|
    tryComplete(close = true) {
 | 
						|
        const matches = this._completions.filter((o) => {
 | 
						|
            return this._part && o.resourceId.startsWith(this._part.text);
 | 
						|
        });
 | 
						|
        if (matches.length === 1 && this._part && this._part.text.length > 1) {
 | 
						|
            const match = matches[0];
 | 
						|
            let pill: PillPart;
 | 
						|
            if (match.resourceId[0] === "@") {
 | 
						|
                pill = this._partCreator.userPill(match.text, match.resourceId);
 | 
						|
            } else {
 | 
						|
                pill = this._partCreator.roomPill(match.resourceId);
 | 
						|
            }
 | 
						|
            this._updateCallback({ replaceParts: [pill], close });
 | 
						|
        }
 | 
						|
    }
 | 
						|
 | 
						|
    // called by EditorModel when typing into pill-candidate part
 | 
						|
    onPartUpdate(part: Part, pos: DocumentPosition) {
 | 
						|
        this._part = part;
 | 
						|
    }
 | 
						|
}
 | 
						|
 | 
						|
// MockClient & MockRoom are only used for avatars in room and user pills,
 | 
						|
// which is not tested
 | 
						|
class MockRoom {
 | 
						|
    getMember(): RoomMember | null {
 | 
						|
        return null;
 | 
						|
    }
 | 
						|
}
 | 
						|
 | 
						|
export function createPartCreator(completions: PillPart[] = []) {
 | 
						|
    const autoCompleteCreator = (partCreator: PartCreator) => {
 | 
						|
        return (updateCallback: UpdateCallback) =>
 | 
						|
            new MockAutoComplete(updateCallback, partCreator, completions) as unknown as AutocompleteWrapperModel;
 | 
						|
    };
 | 
						|
    const room = new MockRoom() as unknown as Room;
 | 
						|
    const client = {
 | 
						|
        getRooms: jest.fn().mockReturnValue([]),
 | 
						|
        getRoom: jest.fn().mockReturnValue(null),
 | 
						|
    } as unknown as MatrixClient;
 | 
						|
    return new PartCreator(room, client, autoCompleteCreator);
 | 
						|
}
 | 
						|
 | 
						|
export function createRenderer() {
 | 
						|
    const render = (c?: Caret) => {
 | 
						|
        render.caret = c;
 | 
						|
        render.count += 1;
 | 
						|
    };
 | 
						|
    render.count = 0;
 | 
						|
    render.caret = null as unknown as Caret | undefined;
 | 
						|
    return render;
 | 
						|
}
 | 
						|
 | 
						|
// in many tests we need to narrow the caret type
 | 
						|
export function isDocumentPosition(caret: Caret): caret is DocumentPosition {
 | 
						|
    return caret instanceof DocumentPosition;
 | 
						|
}
 |