fix style css in ts

src/utils/image.ts

@@ -28,7 +28,13 @@ function fileToBase64(file: File): Promise<string> {
 	return new Promise((resolve, reject) => {
 		const reader = new FileReader()
 		reader.readAsDataURL(file)
-		reader.onload = () => resolve(reader.result as string)
+		reader.onload = () => {
+		  if (typeof reader.result === 'string') {
+		    resolve(reader.result)
+		  } else {
+		    reject(new Error('Unexpected file reader result type'))
+		  }
+		}
 		reader.onerror = () => reject(new Error('Failed to read file'))
 	})
 }

src/utils/obsidian.ts

@@ -1,4 +1,4 @@
-import { App, Editor, MarkdownView, TFile, TFolder, Vault } from 'obsidian'
+import { App, Editor, MarkdownView, TFile, TFolder, Vault, WorkspaceLeaf } from 'obsidian'
 
 import { MentionableBlockData } from '../types/mentionable'
 
@@ -60,7 +60,11 @@ export function getOpenFiles(app: App): TFile[] {
 	try {
 		const leaves = app.workspace.getLeavesOfType('markdown')
 
-		return leaves.map((v) => (v.view as MarkdownView).file).filter((v) => !!v)
+		return leaves
+			.filter((v): v is WorkspaceLeaf & { view: MarkdownView & { file: TFile } } => 
+				v.view instanceof MarkdownView && !!v.view.file
+			)
+			.map((v) => v.view.file)
 	} catch (e) {
 		return []
 	}
@@ -111,9 +115,8 @@ export function openMarkdownFile(
 	if (existingLeaf) {
 		app.workspace.setActiveLeaf(existingLeaf, { focus: true })
 
-		if (startLine) {
-			const view = existingLeaf.view as MarkdownView
-			view.setEphemeralState({ line: startLine - 1 }) // -1 because line is 0-indexed
+		if (startLine && existingLeaf.view instanceof MarkdownView) {
+			existingLeaf.view.setEphemeralState({ line: startLine - 1 }) // -1 because line is 0-indexed
 		}
 	} else {
 		const leaf = app.workspace.getLeaf('tab')

src/utils/ollama.ts

@@ -1,10 +1,12 @@
 import { requestUrl } from 'obsidian'
 
+type OllamaTagsResponse = {
+	models: { name: string }[]
+}
+
 export async function getOllamaModels(ollamaUrl: string) {
 	try {
-		const response = (await requestUrl(`${ollamaUrl}/api/tags`)).json as {
-			models: { name: string }[]
-		}
+		const response: OllamaTagsResponse = await requestUrl(`${ollamaUrl}/api/tags`).json
 		return response.models.map((model) => model.name)
 	} catch (error) {
 		return []

src/utils/parse-infio-block.test.ts

@@ -1,4 +1,4 @@
-import { ParsedinfioBlock, parseinfioBlocks } from './parse-infio-block'
+import { InfioBlockAction, ParsedInfioBlock, parseinfioBlocks } from './parse-infio-block'
 
 describe('parseinfioBlocks', () => {
 	it('should parse a string with infio_block elements', () => {
@@ -22,7 +22,7 @@ print("Hello, world!")
 </infio_block>
 Some text after`
 
-		const expected: ParsedinfioBlock[] = [
+		const expected: ParsedInfioBlock[] = [
 			{ type: 'string', content: 'Some text before\n' },
 			{
 				type: 'infio_block',
@@ -58,7 +58,7 @@ print("Hello, world!")
       <infio_block language="python"></infio_block>
     `
 
-		const expected: ParsedinfioBlock[] = [
+		const expected: ParsedInfioBlock[] = [
 			{ type: 'string', content: '\n      ' },
 			{
 				type: 'infio_block',
@@ -76,7 +76,7 @@ print("Hello, world!")
 	it('should handle input without infio_block elements', () => {
 		const input = 'Just a regular string without any infio_block elements.'
 
-		const expected: ParsedinfioBlock[] = [{ type: 'string', content: input }]
+		const expected: ParsedInfioBlock[] = [{ type: 'string', content: input }]
 
 		const result = parseinfioBlocks(input)
 		expect(result).toEqual(expected)
@@ -100,7 +100,7 @@ print("Hello, world!")
 </infio_block>
 End`
 
-		const expected: ParsedinfioBlock[] = [
+		const expected: ParsedInfioBlock[] = [
 			{ type: 'string', content: 'Start\n' },
 			{
 				type: 'infio_block',
@@ -139,7 +139,7 @@ print("Hello, world!")
 # Unfinished infio_block
 
 Some text after without closing tag`
-		const expected: ParsedinfioBlock[] = [
+		const expected: ParsedInfioBlock[] = [
 			{ type: 'string', content: 'Start\n' },
 			{
 				type: 'infio_block',
@@ -158,7 +158,7 @@ Some text after without closing tag`,
 
 	it('should handle infio_block with startline and endline attributes', () => {
 		const input = `<infio_block language="markdown" startline="2" endline="5"></infio_block>`
-		const expected: ParsedinfioBlock[] = [
+		const expected: ParsedInfioBlock[] = [
 			{
 				type: 'infio_block',
 				content: '',
@@ -171,4 +171,32 @@ Some text after without closing tag`,
 		const result = parseinfioBlocks(input)
 		expect(result).toEqual(expected)
 	})
+
+	it('should parse infio_block with action attribute', () => {
+		const input = `<infio_block type="edit"></infio_block>`
+		const expected: ParsedInfioBlock[] = [
+			{
+				type: 'infio_block',
+				content: '',
+				action: InfioBlockAction.Edit,
+			},
+		]
+
+		const result = parseinfioBlocks(input)
+		expect(result).toEqual(expected)
+	})
+
+	it('should handle invalid action attribute', () => {
+		const input = `<infio_block type="invalid"></infio_block>`
+		const expected: ParsedInfioBlock[] = [
+			{
+				type: 'infio_block',
+				content: '',
+				action: undefined,
+			},
+		]
+
+		const result = parseinfioBlocks(input)
+		expect(result).toEqual(expected)
+	})
 })

src/utils/parse-infio-block.ts

@@ -1,6 +1,12 @@
 import { parseFragment } from 'parse5'
 
-export type ParsedinfioBlock =
+export enum InfioBlockAction {
+  Edit = 'edit',
+  New = 'new',
+  Reference = 'reference'
+}
+
+export type ParsedInfioBlock =
 	| { type: 'string'; content: string }
 	| {
 		type: 'infio_block'
@@ -9,11 +15,15 @@ export type ParsedinfioBlock =
 		filename?: string
 		startLine?: number
 		endLine?: number
-		action?: 'edit' | 'new' | 'reference'
+		action?: InfioBlockAction
 	}
 
-export function parseinfioBlocks(input: string): ParsedinfioBlock[] {
-	const parsedResult: ParsedinfioBlock[] = []
+function isInfioBlockAction(value: string): value is InfioBlockAction {
+  return Object.values<string>(InfioBlockAction).includes(value)
+}
+
+export function parseinfioBlocks(input: string): ParsedInfioBlock[] {
+	const parsedResult: ParsedInfioBlock[] = []
 	const fragment = parseFragment(input, {
 		sourceCodeLocationInfo: true,
 	})
@@ -36,7 +46,11 @@ export function parseinfioBlocks(input: string): ParsedinfioBlock[] {
 			const filename = node.attrs.find((attr) => attr.name === 'filename')?.value
 			const startLine = node.attrs.find((attr) => attr.name === 'startline')?.value
 			const endLine = node.attrs.find((attr) => attr.name === 'endline')?.value
-			const action = node.attrs.find((attr) => attr.name === 'type')?.value as 'edit' | 'new' | 'reference'
+			const actionValue = node.attrs.find((attr) => attr.name === 'type')?.value
+			const action = actionValue && isInfioBlockAction(actionValue)
+				? actionValue
+				: undefined
+
 
 			const children = node.childNodes
 			if (children.length === 0) {

src/utils/prompt-generator.ts

@@ -449,17 +449,13 @@ When writing out new markdown blocks, remember not to include "line_number|" at
 	 */
 	private async getWebsiteContent(url: string): Promise<string> {
 		if (isYoutubeUrl(url)) {
-			try {
-				// TODO: pass language based on user preferences
-				const { title, transcript } =
-					await YoutubeTranscript.fetchTranscriptAndMetadata(url)
+			// TODO: pass language based on user preferences
+			const { title, transcript } =
+				await YoutubeTranscript.fetchTranscriptAndMetadata(url)
 
-				return `Title: ${title}
+			return `Title: ${title}
 Video Transcript:
 ${transcript.map((t) => `${t.offset}: ${t.text}`).join('\n')}`
-			} catch (error) {
-				console.error('Error fetching YouTube transcript', error)
-			}
 		}
 
 		const response = await requestUrl({ url })