opencode/script/stats.ts at openwork-dev · different-ai/opencode · GitHub

History

executable file

225 lines (186 loc) · 6.45 KB

Raw

100

101

102

103

104

105

106

107

108

109

110

111

112

113

114

115

116

117

118

119

120

121

122

123

124

125

126

127

128

129

130

131

132

133

134

135

136

137

138

139

140

141

142

143

144

145

146

147

148

149

150

151

152

153

154

155

156

157

158

159

160

161

162

163

164

165

166

167

168

169

170

171

172

173

174

175

176

177

178

179

180

181

182

183

184

185

186

187

188

189

190

191

192

193

194

195

196

197

198

199

200

201

202

203

204

205

206

207

208

209

210

211

212

213

214

215

216

217

218

219

220

221

222

223

224

225

#!/usr/bin/env bun

async function sendToPostHog(event: string, properties: Record<string, any>) {

const key = process.env["POSTHOG_KEY"]

if (!key) {

console.warn("POSTHOG_API_KEY not set, skipping PostHog event")

return

}

const response = await fetch("https://us.i.posthog.com/i/v0/e/", {

method: "POST",

headers: {

"Content-Type": "application/json",

body: JSON.stringify({

distinct_id: "download",

api_key: key,

event,

properties: {

...properties,

}),

}).catch(() => null)

if (response && !response.ok) {

console.warn(`PostHog API error: ${response.status}`)

}

interface Asset {

download_count: number

}

interface Release {

tag_name: string

assets: Asset[]

}

interface NpmDownloadsRange {

start: string

end: string

package: string

downloads: Array<{

downloads: number

day: string

}

async function fetchNpmDownloads(packageName: string): Promise<number> {

try {

// Use a range from 2020 to current year + 5 years to ensure it works forever

const currentYear = new Date().getFullYear()

const endYear = currentYear + 5

const response = await fetch(`https://api.npmjs.org/downloads/range/2020-01-01:${endYear}-12-31/${packageName}`)

if (!response.ok) {

console.warn(`Failed to fetch npm downloads for ${packageName}: ${response.status}`)

return 0

}

const data: NpmDownloadsRange = await response.json()

return data.downloads.reduce((total, day) => total + day.downloads, 0)

} catch (error) {

console.warn(`Error fetching npm downloads for ${packageName}:`, error)

return 0

}

async function fetchReleases(): Promise<Release[]> {

const releases: Release[] = []

let page = 1

const per = 100

while (true) {

const url = `https://api.github.com/repos/anomalyco/opencode/releases?page=${page}&per_page=${per}`

const response = await fetch(url)

if (!response.ok) {

throw new Error(`GitHub API error: ${response.status} ${response.statusText}`)

}

const batch: Release[] = await response.json()

if (batch.length === 0) break

releases.push(...batch)

console.log(`Fetched page ${page} with ${batch.length} releases`)

if (batch.length < per) break

page++

await new Promise((resolve) => setTimeout(resolve, 1000))

}

return releases

}

function calculate(releases: Release[]) {

let total = 0

const stats = []

for (const release of releases) {

let downloads = 0

const assets = []

for (const asset of release.assets) {

downloads += asset.download_count

assets.push({

downloads: asset.download_count,

})

}

total += downloads

stats.push({

tag: release.tag_name,

downloads,

assets,

})

}

return { total, stats }

}

async function save(githubTotal: number, npmDownloads: number) {

const file = "STATS.md"

const date = new Date().toISOString().split("T")[0]

const total = githubTotal + npmDownloads

let previousGithub = 0

let previousNpm = 0

let previousTotal = 0

let content = ""

try {

content = await Bun.file(file).text()

const lines = content.trim().split("\n")

for (let i = lines.length - 1; i >= 0; i--) {

const line = lines[i].trim()

if (line.startsWith("|") && !line.includes("Date") && !line.includes("---")) {

const match = line.match(

/\|\s*[\d-]+\s*\|\s*([\d,]+)\s*(?:$[^)]*$)?\s*\|\s*([\d,]+)\s*(?:$[^)]*$)?\s*\|\s*([\d,]+)\s*(?:$[^)]*$)?\s*\|/,

)

if (match) {

previousGithub = parseInt(match[1].replace(/,/g, ""))

previousNpm = parseInt(match[2].replace(/,/g, ""))

previousTotal = parseInt(match[3].replace(/,/g, ""))

break

}

} catch {

content =

"# Download Stats\n\n| Date | GitHub Downloads | npm Downloads | Total |\n|------|------------------|---------------|-------|\n"

}

const githubChange = githubTotal - previousGithub

const npmChange = npmDownloads - previousNpm

const totalChange = total - previousTotal

const githubChangeStr =

githubChange > 0

? ` (+${githubChange.toLocaleString()})`

: githubChange < 0

? ` (${githubChange.toLocaleString()})`

: " (+0)"

const npmChangeStr =

npmChange > 0 ? ` (+${npmChange.toLocaleString()})` : npmChange < 0 ? ` (${npmChange.toLocaleString()})` : " (+0)"

const totalChangeStr =

totalChange > 0

? ` (+${totalChange.toLocaleString()})`

: totalChange < 0

? ` (${totalChange.toLocaleString()})`

: " (+0)"

const line = `| ${date} | ${githubTotal.toLocaleString()}${githubChangeStr} | ${npmDownloads.toLocaleString()}${npmChangeStr} | ${total.toLocaleString()}${totalChangeStr} |\n`

if (!content.includes("# Download Stats")) {

content =

"# Download Stats\n\n| Date | GitHub Downloads | npm Downloads | Total |\n|------|------------------|---------------|-------|\n"

}

await Bun.write(file, content + line)

await Bun.spawn(["bunx", "prettier", "--write", file]).exited

console.log(

`\nAppended stats to ${file}: GitHub ${githubTotal.toLocaleString()}${githubChangeStr}, npm ${npmDownloads.toLocaleString()}${npmChangeStr}, Total ${total.toLocaleString()}${totalChangeStr}`,

)

}

console.log("Fetching GitHub releases for anomalyco/opencode...\n")

const releases = await fetchReleases()

console.log(`\nFetched ${releases.length} releases total\n`)

const { total: githubTotal } = calculate(releases)

console.log("Fetching npm all-time downloads for opencode-ai...\n")

const npmDownloads = await fetchNpmDownloads("opencode-ai")

console.log(`Fetched npm all-time downloads: ${npmDownloads.toLocaleString()}\n`)

await save(githubTotal, npmDownloads)

await sendToPostHog("download", {

source: "github",

})

await sendToPostHog("download", {

source: "npm",

})

const totalDownloads = githubTotal + npmDownloads

console.log("=".repeat(60))

console.log(`TOTAL DOWNLOADS: ${totalDownloads.toLocaleString()}`)

console.log(` GitHub: ${githubTotal.toLocaleString()}`)

console.log(` npm: ${npmDownloads.toLocaleString()}`)

console.log("=".repeat(60))

console.log("-".repeat(60))

console.log(`GitHub Total: ${githubTotal.toLocaleString()} downloads across ${releases.length} releases`)

console.log(`npm Total: ${npmDownloads.toLocaleString()} downloads`)

console.log(`Combined Total: ${totalDownloads.toLocaleString()} downloads`)

opencode/script/stats.ts at openwork-dev · different-ai/opencode · GitHub

Uh oh!

Expand file tree

Breadcrumbs

stats.ts

Latest commit

History

Breadcrumbs

stats.ts

File metadata and controls