src/reddit.js


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77

async function fetchRedditPosts() {
  const response = await fetch(redditURL, {
    headers: {
      'User-Agent': 'UmaBot/0.1.0',
    },
  });

  if (!response.ok) {
    let errorText = `Error fetching ${response.url}: ${response.status} ${response.statusText}`;

    try {
      const error = await response.text();

      if (error) errorText = `${errorText} \n\n ${error}`;
    } catch {
      //
    }

    throw new Error(errorText);
  }

  const data = await response.json();

  return data.data.children.map((post) => post.data);
}

function filterPostsByFlair(posts, excludedFlairs = [], includedFlairs = []) {
  return posts.filter((post) => {
    if (post.is_gallery) return false;

    const hasMedia =
      post.media?.reddit_video?.fallback_url ||
      post.secure_media?.reddit_video?.fallback_url ||
      post.url;

    if (!hasMedia) return false;

    if (post.over_18 || post.link_flair_text?.toLowerCase().includes('nsfw'))
      return false;

    const postFlair = post.link_flair_text?.toLowerCase() || '';

    if (includedFlairs.length > 0)
      return includedFlairs.some((flair) =>
        postFlair.includes(flair.toLowerCase()),
      );

    if (excludedFlairs.length > 0)
      return !excludedFlairs.some((flair) =>
        postFlair.includes(flair.toLowerCase()),
      );

    return true;
  });
}

function getRandomPost(posts) {
  const randomIndex = Math.floor(Math.random() * posts.length);

  return posts[randomIndex];
}

export async function getCutePost() {
  const posts = await fetchRedditPosts();
  const filteredPosts = filterPostsByFlair(posts, ['roleplay', 'announcement']);

  return getRandomPost(filteredPosts);
}

export async function getRoleplayPost() {
  const posts = await fetchRedditPosts();
  const filteredPosts = filterPostsByFlair(posts, [], ['roleplay']);

  return getRandomPost(filteredPosts);
}

export const redditURL = 'https://www.reddit.com/r/okbuddyumamusume/hot.json';