nniRestServer.ts 2.32 KB
Newer Older
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
/**
 * Copyright (c) Microsoft Corporation
 * All rights reserved.
 *
 * MIT License
 *
 * Permission is hereby granted, free of charge, to any person obtaining a copy of this software and associated
 * documentation files (the "Software"), to deal in the Software without restriction, including without limitation
 * the rights to use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies of the Software, and
 * to permit persons to whom the Software is furnished to do so, subject to the following conditions:
 * The above copyright notice and this permission notice shall be included in all copies or substantial portions of the Software.
 *
 * THE SOFTWARE IS PROVIDED *AS IS*, WITHOUT WARRANTY OF ANY KIND, EXPRESS OR IMPLIED, INCLUDING
 * BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
 * NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM,
 * DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
 */

'use strict';

import * as bodyParser from 'body-parser';
chicm-ms's avatar
chicm-ms committed
23
import * as express from 'express';
goooxu's avatar
goooxu committed
24
import * as path from 'path';
25
26
import * as component from '../common/component';
import { RestServer } from '../common/restServer'
chicm-ms's avatar
chicm-ms committed
27
import { getLogDir } from '../common/utils';
28
29
30
31
32
import { createRestHandler } from './restHandler';

/**
 * NNI Main rest server, provides rest API to support
 * # nnictl CLI tool
goooxu's avatar
goooxu committed
33
 * # NNI WebUI
34
 *
35
36
 */
@component.Singleton
goooxu's avatar
goooxu committed
37
export class NNIRestServer extends RestServer {
38
    private readonly API_ROOT_URL: string = '/api/v1/nni';
chicm-ms's avatar
chicm-ms committed
39
    private readonly LOGS_ROOT_URL: string = '/logs';
40
41
42
43
44
45
46
47
48
49
50
51

    /**
     * constructor to provide NNIRestServer's own rest property, e.g. port
     */
    constructor() {
        super();
    }

    /**
     * NNIRestServer's own router registration
     */
    protected registerRestHandler(): void {
goooxu's avatar
goooxu committed
52
        this.app.use(express.static('static'));
53
54
        this.app.use(bodyParser.json());
        this.app.use(this.API_ROOT_URL, createRestHandler(this));
chicm-ms's avatar
chicm-ms committed
55
        this.app.use(this.LOGS_ROOT_URL, express.static(getLogDir()));
goooxu's avatar
goooxu committed
56
57
58
        this.app.get('*', (req: express.Request, res: express.Response) => {
            res.sendFile(path.resolve('static/index.html'));
        });
59
60
    }
}