컴포지트 composite 패턴 - 자바스크립트 예제

// Component: 공통 인터페이스
class FileSystemComponent {
  constructor(name) {
    this.name = name;
  }

  // 모든 컴포넌트가 공통으로 가질 작업
  display(indent = "") {
    throw new Error("display() 메서드는 반드시 구현되어야 합니다.");
  }

  // Composite 객체에서만 의미 있는 메서드들
  add(component) {
    throw new Error("이 객체에는 자식을 추가할 수 없습니다.");
  }

  remove(component) {
    throw new Error("이 객체에서는 자식을 제거할 수 없습니다.");
  }
}

// Leaf: 개별 파일
class File extends FileSystemComponent {
  constructor(name, size) {
    super(name);
    this.size = size;
  }

  display(indent = "") {
    console.log(`${indent}- ${this.name} (${this.size}KB)`);
  }
}

// Composite: 폴더
class Folder extends FileSystemComponent {
  constructor(name) {
    super(name);
    this.children = [];
  }

  add(component) {
    this.children.push(component);
  }

  remove(component) {
    const index = this.children.indexOf(component);
    if (index > -1) {
      this.children.splice(index, 1);
    }
  }

  display(indent = "") {
    console.log(`${indent}+ ${this.name}`);
    // 자식들에게 재귀적으로 display 호출
    this.children.forEach((child) => {
      child.display(indent + "  ");
    });
  }
}

// 클라이언트 코드
const root = new Folder("root");
const documents = new Folder("문서");
const music = new Folder("음악");

const file1 = new File("이력서.docx", 120);
const file2 = new File("프로젝트_계획.pdf", 340);
const song1 = new File("아이유 - 밤편지.mp3", 4500);

documents.add(file1);
documents.add(file2);
music.add(song1);

root.add(documents);
root.add(music);
root.add(new File("중요한_파일.txt", 50));

// 전체 파일 시스템 구조 출력
root.display();

+ root
  + 문서
    - 이력서.docx (120KB)
    - 프로젝트_계획.pdf (340KB)
  + 음악
    - 아이유 - 밤편지.mp3 (4500KB)
  - 중요한_파일.txt (50KB)