58 lines
1.7 KiB
Go
58 lines
1.7 KiB
Go
|
// SPDX-FileCopyrightText: Amolith <amolith@secluded.site>
|
||
|
|
//
|
||
|
|
// SPDX-License-Identifier: Apache-2.0
|
||
|
|
|
||
|
|
package db
|
||
|
|
|
||
|
|
import (
|
||
|
|
"crypto/sha256"
|
||
|
|
"database/sql"
|
||
|
|
"fmt"
|
||
|
|
)
|
||
|
|
|
||
|
|
// generateAndInsertProjectIDs runs during migration 1, fetches all rows from
|
||
|
|
// projects_tmp, loops through the rows generating a repeatable ID for each
|
||
|
|
// project, and inserting it into the new table along with the data from the old
|
||
|
|
// table.
|
||
|
|
func generateAndInsertProjectIDs(tx *sql.Tx) error {
|
||
|
|
// Loop through projects_tmp, generate a project_id for each, and insert
|
||
|
|
// into projects
|
||
|
|
rows, err := tx.Query("SELECT url, name, forge, version, created_at FROM projects_tmp")
|
||
|
|
if err != nil {
|
||
|
|
return fmt.Errorf("failed to list projects in projects_tmp: %w", err)
|
||
|
|
}
|
||
|
|
defer rows.Close()
|
||
|
|
|
||
|
|
for rows.Next() {
|
||
|
|
var (
|
||
|
|
url string
|
||
|
|
name string
|
||
|
|
forge string
|
||
|
|
version string
|
||
|
|
created_at string
|
||
|
|
)
|
||
|
|
if err := rows.Scan(&url, &name, &forge, &version, &created_at); err != nil {
|
||
|
|
return fmt.Errorf("failed to scan row from projects_tmp: %w", err)
|
||
|
|
}
|
||
|
|
id := fmt.Sprintf("%x", sha256.Sum256([]byte(url+name+forge+created_at)))
|
||
|
|
_, err = tx.Exec(
|
||
|
|
"INSERT INTO projects (id, url, name, forge, version, created_at) VALUES (@id, @url, @name, @forge, @version, @created_at)",
|
||
|
|
sql.Named("id", id),
|
||
|
|
sql.Named("url", url),
|
||
|
|
sql.Named("name", name),
|
||
|
|
sql.Named("forge", forge),
|
||
|
|
sql.Named("version", version),
|
||
|
|
sql.Named("created_at", created_at),
|
||
|
|
)
|
||
|
|
if err != nil {
|
||
|
|
return fmt.Errorf("failed to insert project into projects: %w", err)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
if _, err := tx.Exec("DROP TABLE projects_tmp"); err != nil {
|
||
|
|
return fmt.Errorf("failed to drop projects_tmp: %w", err)
|
||
|
|
}
|
||
|
|
|
||
|
|
return nil
|
||
|
|
}
|