แสดงสถานะ user online ด้วย Firebase (2)

// ดึง uid มาใช้งาน
const uid = firebase.auth().currentUser.uid
// สร้าง RTDB object เอาไว้ใช้งาน
const RTDB = firebase.database().ref()

// node พิเศษ เอามาดูได้ว่า uid ปัจจุบัน connect กับ Firebase อยู่หรือปล่าว
const amOnline = RTDB.child('.info/connected')
// ให้ RTDB ชี้ไปที่ /presence/{uid}
const presenceRef = RTDB.child('presence').child(uid)
// ให้ RTDB ชี้ไปที่ /sessions
const sessionsRef = RTDB.child('sessions')

amOnline.on('value', snapShot => {
  // ถ้ายังไม่ได้ connect กับ Firebase ก็ยังไม่ต้องทำอะไร
  if (!snapShot.val()) return

  // connect ได้แล้วสร้าง presence ได้เลย
  presenceRef.set(true)
  // เราจะไม่ทำ disconnect ในฝั่ง browser แล้ว แต่จะใช้ Cloud Function ลบให้แทน
  // presenceRef.onDisconnect().remove()

  // สร้าง session ขึ้นมาใหม่ โดยเก็บ uid และ startedAt
  const session = sessionsRef.push({
    startedAt: firebase.database.ServerValue.TIMESTAMP,
    uid: uid,
  })

  // ในกรณีที่ disconnect ให้เพิ่ม endedAt ใน session ที่เพิ่งสร้างมาเมื่อกี้
  session.child('endedAt')
    .onDisconnect()
    .set(firebase.database.ServerValue.TIMESTAMP)
})

// ทำ setup env ต่างๆของ Firebase Function
const admin = require('firebase-admin')
const functions = require('firebase-functions')
const DEFAULT_REGION = 'asia-northeast1'
const functionRunner = functions.region(DEFAULT_REGION)

// สร้าง function สำหรับ event listener RTDB
exports.SessionDeleted = functionRunner.database
  // ตรงนี้บอกว่า เราจะ listen เฉพาะที่ /sessions/... เท่านั้นนะ
  .ref('/sessions/{sessionKey}')
  .onUpdate((snapshot, context) => {
    // ดึง data ที่ update มาใช้งาน
    const data = snapshot.data()
    // ไม่มี endedAt แสดงว่าไม่ได้ disconnect ข้าม
    if (!data.endedAt) return

    // ดึง uid มาใช้งาน
    const uid = data.uid

    // สร้าง object ของ RTDB โดยเอา presence และ sessions มาใช้งาน 
    const ref = database.ref()
    const presenceRef = ref.child('presence').child(uid)
    const sessionsRef = ref.child('sessions')

    // ตรงนี้เป็นการดึงข้อมูลใน RTDB sessions มาทำงาน
    sessionsRef.once('value', data => {
      // ดึงข้อมูล sessions มาใช้งาน กรณีที่ไม่มี ให้เอา empty array มาแทน
      const allSessions = data.val() || []
      // ทำ filter หา sessions ที่ยัง active อยู่ คือ มี uid และไม่มี endedAt
      const remainingSessions = Object.keys(allSessions).filter(
        sessionKey => allSessions[sessionKey].uid === uid 
          && !allSessions[sessionKey].endedAt,
      )

      // ถ้าไม่เจอ ก็ลบ presence ทิ้งเลย
      if (remainingSessions.length <= 0) {
        presenceRef.remove()
      }
    })
  })