mirror of https://github.com/vector-im/riot-web
				
				
				
			
		
			
				
	
	
		
			82 lines
		
	
	
		
			2.7 KiB
		
	
	
	
		
			TypeScript
		
	
	
			
		
		
	
	
			82 lines
		
	
	
		
			2.7 KiB
		
	
	
	
		
			TypeScript
		
	
	
/*
 | 
						|
Copyright 2023 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 http from "http";
 | 
						|
import express from "express";
 | 
						|
import { AddressInfo } from "net";
 | 
						|
 | 
						|
import PluginEvents = Cypress.PluginEvents;
 | 
						|
import PluginConfigOptions = Cypress.PluginConfigOptions;
 | 
						|
 | 
						|
const servers: http.Server[] = [];
 | 
						|
 | 
						|
function startOAuthServer(html: string): number {
 | 
						|
    const app = express();
 | 
						|
 | 
						|
    // static files. This includes the "authorization endpoint".
 | 
						|
    app.use(express.static(__dirname + "/res"));
 | 
						|
 | 
						|
    // token endpoint (see https://openid.net/specs/openid-connect-core-1_0.html#TokenEndpoint)
 | 
						|
    app.use("/oauth/token", express.urlencoded());
 | 
						|
    app.post("/oauth/token", (req, res) => {
 | 
						|
        // if the code is valid, accept it. Otherwise, return an error.
 | 
						|
        const code = req.body.code;
 | 
						|
        if (code === "valid_auth_code") {
 | 
						|
            res.send({
 | 
						|
                access_token: "oauth_access_token",
 | 
						|
                token_type: "Bearer",
 | 
						|
                expires_in: "3600",
 | 
						|
            });
 | 
						|
        } else {
 | 
						|
            res.send({ error: "bad auth code" });
 | 
						|
        }
 | 
						|
    });
 | 
						|
 | 
						|
    // userinfo endpoint (see https://openid.net/specs/openid-connect-core-1_0.html#UserInfo)
 | 
						|
    app.get("/oauth/userinfo", (req, res) => {
 | 
						|
        // TODO: validate that the request carries an auth header which matches the access token we issued above
 | 
						|
 | 
						|
        // return an OAuth2 user info object
 | 
						|
        res.send({
 | 
						|
            sub: "alice",
 | 
						|
            name: "Alice",
 | 
						|
        });
 | 
						|
    });
 | 
						|
 | 
						|
    const server = http.createServer(app);
 | 
						|
    server.listen();
 | 
						|
    servers.push(server);
 | 
						|
    const address = server.address() as AddressInfo;
 | 
						|
    console.log(`Started OAuth server at ${address.address}:${address.port}`);
 | 
						|
    return address.port;
 | 
						|
}
 | 
						|
 | 
						|
function stopOAuthServer(): null {
 | 
						|
    console.log("Stopping OAuth servers");
 | 
						|
    for (const server of servers) {
 | 
						|
        const address = server.address() as AddressInfo;
 | 
						|
        server.close();
 | 
						|
        console.log(`Stopped OAuth server at ${address.address}:${address.port}`);
 | 
						|
    }
 | 
						|
    servers.splice(0, servers.length); // clear
 | 
						|
    return null;
 | 
						|
}
 | 
						|
 | 
						|
export function oAuthServer(on: PluginEvents, config: PluginConfigOptions) {
 | 
						|
    on("task", { startOAuthServer, stopOAuthServer });
 | 
						|
    on("after:run", stopOAuthServer);
 | 
						|
}
 |