start-os/setup-wizard/src/app/pages/embassy/embassy.page.ts

import { Component } from '@angular/core'
import { iosTransitionAnimation, LoadingController, ModalController, NavController } from '@ionic/angular'
import { ApiService, DiskInfo } from 'src/app/services/api/api.service'
import { StateService } from 'src/app/services/state.service'
import { PasswordPage } from '../password/password.page'

@Component({
  selector: 'app-embassy',
  templateUrl: 'embassy.page.html',
  styleUrls: ['embassy.page.scss'],
})
export class EmbassyPage {
  embassyDrives = []
  selectedDrive: DiskInfo = null
  loading = true
  window = window

  constructor(
    private readonly apiService: ApiService,
    private readonly navCtrl: NavController,
    private modalController: ModalController,
    private stateService: StateService,
    private loadingCtrl: LoadingController
  ) {}

  async ngOnInit() {
    const drives = (await this.apiService.getDrives()).filter(d => !d['embassy_os'])
    this.embassyDrives = (await this.apiService.getDrives()).filter(d => !d['embassy_os'])
    this.loading = false
  }

  async chooseDrive(drive: DiskInfo) {    
    const modal = await this.modalController.create({
      component: PasswordPage,
      componentProps: {
        embassyDrive: drive
      }
    })
    modal.onDidDismiss().then(async ret => {
      if (!ret.data || !ret.data.password) return

      const loader = await this.loadingCtrl.create({
        message: 'Setting up your Embassy!'
      })
      
      await loader.present()
  
      this.stateService.embassyDrive = drive
      this.stateService.embassyPassword = ret.data.password
  
      try {
        this.stateService.torAddress = (await this.stateService.setupEmbassy()).torAddress
      } catch (e) {
        console.log(e.message)
      } finally {
        loader.dismiss()
        if(!!this.stateService.recoveryDrive) {
          await this.navCtrl.navigateForward(`/loading`, { animationDirection: 'forward', animation: iosTransitionAnimation })
        } else {
          await this.navCtrl.navigateForward(`/success`, { animationDirection: 'forward', animation: iosTransitionAnimation })
        }
      }
    })
    await modal.present();
  }

  getLabel(drive: DiskInfo) {
    const labels = drive.partitions.map(p => p.label).filter(l => !!l)
    return labels.length ? labels.join(' / ') : 'unnamed'
  }

  getUsage(drive: DiskInfo) {
    let usage = 0
    drive.partitions.forEach(par => {
      if(par.used) {
        usage += par.used
      }
    })
    return usage.toFixed(2)
  }
}