rasp-blefi/src/network.ts

Experimental project that allow headless configuration of WIFI network on Rasperry PI using BLE.

import {
  findNetworkBySSID,
  addNetwork,
  setNetworkVariable,
  enableNetwork,
  removeNetwork,
  getConnectedSSID as getCurrentSSID,
} from './utils/cmd';
import { EventEmitter } from 'events';
import { logger } from './utils/logger';

export enum NetworkStatus {
  connected = 'connected',
  disconnected = 'disconnected',
}

/**
 * Represent network
 */
export default class Network extends EventEmitter {
  /**
   * Wifi network interface name
   */
  private ifName: string;

  private ssid: string = '';

  // @ts-ignore
  private status: NetworkStatus;

  constructor(ifName: string) {
    super();
    this.ifName = ifName;

    // Watch for status changing
    setInterval(() => {
      logger.debug('watch for status change');
      const hasChange = this.hasStatusChanged(this.status);
      if (hasChange) {
        this.emit('statusChange', this.status, this.ssid);
      }
    }, 1000);
  }

  /**
   * Get current connected SSID
   */
  async getConnectedSSID(): Promise<string> {
    this.ssid = await getCurrentSSID(this.ifName);
    return this.ssid;
  }

  /**
   * Get current network connection status
   */
  async getStatus(): Promise<NetworkStatus> {
    logger.info('get WIFI network connection status');
    const ssid = await this.getConnectedSSID();
    this.status = ssid ? NetworkStatus.connected : NetworkStatus.disconnected;
    return this.status;
  }

  /**
   * Compare old status with new status
   * @param oldStatus - old status
   */
  async hasStatusChanged(oldStatus: NetworkStatus): Promise<boolean> {
    logger.info('check if network connection status has changed');
    const newStatus = await this.getStatus();
    return newStatus !== oldStatus;
  }

  /**
   * Configure WIFI network
   * @param ssid - ssid
   * @param pwd - password
   */
  async configureNetwork(ssid: string, pwd: string) {
    logger.info('configure WIFI network', { ssid });
    let networkId: number;
    try {
      const network = await findNetworkBySSID(this.ifName, ssid);

      // Create or Update network
      if (network.networkId) {
        networkId = network.networkId;
      } else {
        networkId = await addNetwork(this.ifName);
      }

      // Set credentials
      await setNetworkVariable(this.ifName, networkId, 'ssid', `'"${ssid}"'`);
      await setNetworkVariable(this.ifName, networkId, 'psk', `'"${pwd}"'`);

      await enableNetwork(this.ifName, networkId);
    } catch (err) {
      // @ts-ignore
      if (networkId) {
        await removeNetwork(this.ifName, networkId);
      }
    }
  }
}